TestThreadDump.java

/*
 * Copyright 2003-2012 Yusuke Yamamoto
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package samurai.core;

import junit.framework.TestCase;

import java.io.File;
import java.io.IOException;
import java.io.FileFilter;
import java.io.FileInputStream;
import java.io.PrintWriter;
import java.io.FileOutputStream;
import java.util.Properties;

public class TestThreadDump extends TestCase {

    public TestThreadDump(String name){
        super(name);
    }

    protected void setUp() throws Exception {
        super.setUp();
    }

    protected void tearDown() throws Exception {
        super.tearDown();
    }

    public void testSunThreadDump() throws Exception{
        dig2(new File("testCases"+File.separator+"Sun"));
    }
    public void testBEAThreadDump() throws Exception{
        dig2(new File("testCases"+File.separator+"BEA"));
    }
    public void testAppleThreadDump() throws Exception{
        dig2(new File("testCases"+File.separator+"Apple"));
    }
    public void testHPThreadDump() throws Exception{
        dig2(new File("testCases"+File.separator+"HP"));
    }
    public void testIBMThreadDump() throws Exception{
        dig2(new File("testCases"+File.separator+"IBM"));
    }
    public void dig2(File dir) throws IOException {
        File[] files = dir.listFiles(new FileFilter() {
            public boolean accept(File file) {
                return file.isDirectory() || file.getName().endsWith(".dmp") || file.getName().endsWith(".dump")|| file.getName().endsWith(".txt");
            }
        }
        );
        for (File file : files) {
            if (file.isDirectory()) {
                dig2(file);
            } else {
                examine(file);
            }
        }

    }
    /*package*/
     void examine(File in) throws IOException {
        File out  = new File(in.getAbsolutePath() + ".expected");
        if (!out.exists()) {
            dumpAnalyzed(in);
        }

        ThreadStatistic statistic = new ThreadStatistic();
        ThreadDumpExtractor dumpExtractor = new ThreadDumpExtractor(statistic);
        dumpExtractor.analyze(in);
        Properties props = new Properties();
        props.load(new FileInputStream(out));

        for (int i = 0; i < statistic.getFullThreadDumpCount(); i++) {
            String examining = in.getAbsolutePath() + ":";
            FullThreadDump ftd = statistic.getFullThreadDump(i);
            assertEquals(examining + "ftd" + i + ".deadLockSize", props.getProperty("ftd." + i + ".deadLockSize"), String.valueOf(ftd.getDeadLockSize()));
            assertEquals(examining + "ftd" + i + ".threadCount", props.getProperty("ftd." + i + ".threadCount"), String.valueOf(ftd.getThreadCount()));
            for (int j = 0; j < ftd.getThreadCount(); j++) {
                ThreadDump td = ftd.getThreadDump(j);
                assertEquals(examining + "td." + i + "." + j + ".blockedObjectId", props.getProperty("td." + i + "." + j + ".blockedObjectId"), String.valueOf(td.getBlockedObjectId()));
                assertEquals(examining + "td." + i + "." + j + ".blockerId", props.getProperty("td." + i + "." + j + ".blockerId"), String.valueOf(td.getBlockerId()));
                assertEquals(examining + "td." + i + "." + j + ".condition", props.getProperty("td." + i + "." + j + ".condition"), String.valueOf(td.getCondition()));
                assertEquals(examining + "td." + i + "." + j + ".header", props.getProperty("td." + i + "." + j + ".header"), String.valueOf(td.getHeader()));
                assertEquals(examining + "td." + i + "." + j + ".id", props.getProperty("td." + i + "." + j + ".id"), String.valueOf(td.getId()));
                assertEquals(examining + "td." + i + "." + j + ".name", props.getProperty("td." + i + "." + j + ".name"), String.valueOf(td.getName()));
                assertEquals(examining + "td." + i + "." + j + ".isBlocked", props.getProperty("td." + i + "." + j + ".isBlocked"), String.valueOf(td.isBlocked()));
                assertEquals(examining + "td." + i + "." + j + ".isBlocking", props.getProperty("td." + i + "." + j + ".isBlocking"), String.valueOf(td.isBlocking()));
                assertEquals(examining + "td." + i + "." + j + ".isDaemon", props.getProperty("td." + i + "." + j + ".isDaemon"), String.valueOf(td.isDaemon()));
                assertEquals(examining + "td." + i + "." + j + ".isDeadLocked", props.getProperty("td." + i + "." + j + ".isDeadLocked"), String.valueOf(td.isDeadLocked()));
                assertEquals(examining + "td." + i + "." + j + ".isIdle", props.getProperty("td." + i + "." + j + ".isIdle"), String.valueOf(td.isIdle()));
            }
        }
    }

    /*package*/
    static void dumpAnalyzed(File in) throws IOException {
        dumpAnalyzed(in, new File(in.getAbsolutePath() + ".expected"));
    }

    /*package*/
    static void dumpAnalyzed(File in, File out) throws IOException {
        System.out.println("Analyzing: " + in.getAbsolutePath());
        if (out.exists()) {
            System.out.println("Output exists, skipping: " + out.getAbsolutePath());
        } else {
            ThreadStatistic statistic = new ThreadStatistic();
            ThreadDumpExtractor dumpExtractor = new ThreadDumpExtractor(statistic);
            dumpExtractor.analyze(in);
            PrintWriter pw = new PrintWriter(new FileOutputStream(out));

            for (int i = 0; i < statistic.getFullThreadDumpCount(); i++) {
                FullThreadDump ftd = statistic.getFullThreadDump(i);
                pw.println("ftd." + i + ".deadLockSize=" + ftd.getDeadLockSize());
                pw.println("ftd." + i + ".threadCount=" + ftd.getThreadCount());
                for (int j = 0; j < ftd.getThreadCount(); j++) {
                    ThreadDump td = ftd.getThreadDump(j);
                    pw.println("#" + td.getHeader());
                    for(StackLine line: td.getStackLines()){
                        pw.println("#" + line.toString());
                    }
                    pw.println("td." + i + "." + j + ".blockedObjectId=" + td.getBlockedObjectId());
                    pw.println("td." + i + "." + j + ".blockerId=" + td.getBlockerId());
                    pw.println("td." + i + "." + j + ".condition=" + td.getCondition());
                    pw.println("td." + i + "." + j + ".header=" + td.getHeader());
                    pw.println("td." + i + "." + j + ".id=" + td.getId());
                    pw.println("td." + i + "." + j + ".name=" + td.getName());
                    pw.println("td." + i + "." + j + ".isBlocked=" + td.isBlocked());
                    pw.println("td." + i + "." + j + ".isBlocking=" + td.isBlocking());
                    pw.println("td." + i + "." + j + ".isDaemon=" + td.isDaemon());
                    pw.println("td." + i + "." + j + ".isDeadLocked=" + td.isDeadLocked());
                    pw.println("td." + i + "." + j + ".isIdle=" + td.isIdle());
                    pw.println();
                }
            }
            pw.close();
            System.out.println("Done analyzing: " + out.getAbsolutePath());
            System.out.println("Review and edit it if needed.");
        }
    }
}