LongRunningTestMonitor.java

/**
 *  BlueCove - Java library for Bluetooth
 *  Copyright (C) 2008-2009 Vlad Skarzhevskyy
 *
 *  Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF licenses this file
 *  to you 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.
 *
 *  @author vlads
 *  @version $Id$
 */
package net.sf.bluecove;

/**
 * 
 */
public class LongRunningTestMonitor extends Thread {

    private boolean testFinished = false;

    private Thread testThread;

    private ThreadGroup secondaryTestThreadGroup;

    private int gracePeriod = 0;

    private String testName;

    LongRunningTestMonitor(int gracePeriod, String testName) {
        super("TestMonitor");
        this.gracePeriod = gracePeriod;
        this.testName = testName;
        testThread = Thread.currentThread();
    }

    public void setSecondaryTestThreadGroup(ThreadGroup secondaryTestThreadGroup) {
        this.secondaryTestThreadGroup = secondaryTestThreadGroup;
    }

    public void run() {
        try {
            sleep(gracePeriod);
        } catch (InterruptedException e) {
            return;
        }

        int count = 0;
        while (!testFinished) {

            try {
                sleep(4 * 1000);
            } catch (InterruptedException e) {
                return;
            }

            if (!testFinished) {
                System.out.println("Long running test " + testName + " detected in thread:" + testThread.getName());
                StackTraceElement[] ste = testThread.getStackTrace();
                StringBuffer buf = new StringBuffer();
                buf.append("stack trace:\n");
                for (int i = 0; i < ste.length; i++) {
                    buf.append("\t").append(ste[i].toString()).append('\n');
                    if (ste[i].getClassName().startsWith("junit.framework")) {
                        break;
                    }
                }
                if (secondaryTestThreadGroup != null) {
                    int secondaryTestThreadCount = secondaryTestThreadGroup.activeCount();
                    if (secondaryTestThreadCount > 0) {
                        buf.append("---------------\n");
                        buf.append("stack trace for secondary " + secondaryTestThreadCount + " threads:\n");
                        Thread[] activeThreads = new Thread[secondaryTestThreadCount];
                        secondaryTestThreadCount = secondaryTestThreadGroup.enumerate(activeThreads);
                        for (int i = 0; i < secondaryTestThreadCount; i++) {
                            StackTraceElement[] sste = activeThreads[i].getStackTrace();
                            for (int k = 0; k < sste.length; k++) {
                                buf.append("\t").append(sste[k].toString()).append('\n');
                                if (sste[k].getClassName().startsWith("junit.framework")) {
                                    break;
                                }
                            }
                            buf.append("---------------\n");
                        }
                    }
                }

                System.out.println(buf.toString());
                count++;
                if (count > 4) {
                    System.out.println("Sending ThreadDeath");
                    testThread.stop();
                    break;
                } else {
                    System.out.println("Sending InterruptedException");
                    testThread.interrupt();
                }
            }
        }
    }

    public void finish() {
        testFinished = true;
        interrupt();
    }

}