BaseThreadTest.java

package test.thread;

import org.testng.Assert;
import org.testng.collections.Lists;
import org.testng.collections.Maps;
import org.testng.collections.Sets;

import test.SimpleBaseTest;

import java.util.List;
import java.util.Map;
import java.util.Set;

public class BaseThreadTest extends SimpleBaseTest {
  static private Set<Long> m_threadIds;
  static private Map<String, Long> m_suitesMap;
  static private List<String> m_strings;

  static void initThreadLog() {
    m_threadIds = Sets.newHashSet();
    m_suitesMap = Maps.newHashMap();
    m_strings = Lists.newArrayList();
  }

  protected void logString(String s) {
    synchronized(m_strings) {
      log("BaseThreadTest", "Logging string:" + s);
      m_strings.add(s);
    }
  }

  public static List<String> getStrings() {
    return m_strings;
  }

  protected void logCurrentThread() {
    logThread(Thread.currentThread().getId());
  }

  protected void logThread(long threadId) {
    synchronized(m_threadIds) {
      log("BaseThreadTest", "Logging thread:" + threadId);
      m_threadIds.add(threadId);
    }
  }

  protected void logSuite(String suiteName, long time) {
    synchronized(m_suitesMap) {
      m_suitesMap.put(suiteName, time);
    }
  }

  static int getThreadCount() {
    synchronized(m_threadIds) {
      return m_threadIds.size();
    }
  }

  static Map<String, Long> getSuitesMap() {
    return m_suitesMap;
  }

  protected void log(String cls, String s) {
    if (false) {
      System.out.println("[" + cls + "] thread:" + Thread.currentThread().getId()
          + " hash:" + hashCode() + " " + s);
    }
  }

  protected void verifyThreads(int expected) {
    Assert.assertEquals(getThreadCount(), expected,
        "Ran on " + getThreadCount() + " threads instead of " + expected);
  }
}