Java:修改和访问在构造函数中初始化的变量

Question

我有课。
我在该类的构造函数中初始化一个变量。
我调用一个包含 while 循环的方法，并在每次循环时递增变量。
我编写了一个测试来检查调用该方法后变量的值(并执行一次 while 循环)。

public class ThreadGenerator implements Runnable {
  private int requests;
  private int limit;

  public ThreadGenerator() {
    requests = 0;
  }

  public void setRequestLimit(int anyLimit) {
    this.limit = anyLimit;
  }

  public void generateThread() {
    new Thread(this).start();
  }

  public void run() {
      while(requests < limit) {
          try {
              // do some stuff
              requests++;
              // close some stuff
          } catch (IOException e) {
            e.printStackTrace();
          }
      }
  }

  public int getRequests() {
      return requests; // calling this method from my tests always returns 0
  }

在我的测试中，当我创建此类的新实例，然后在该类上调用此方法时，它会正确运行并正确增加请求计数器。我已经尝试了几种打印语句来确保这一点。但是，如果我在测试中对 ThreadGenerator 对象调用 getRequests，它将返回 0，即它初始化时使用的数量。

我的测试代码:

ThreadGenerator threadGenerator = new ThreadGenerator();
threadGenerator.setRequestLimit(1);
threadGenerator.generateThread();
assertEquals(threadGenerator.getRequests(), 1);

如何修改在构造函数中初始化的变量并在测试套件中访问它？

Answer 1

请记住，仅仅因为您要求 Java 使用 Runnable 创建一个新线程，并不意味着将立即调用 run() 方法。很可能 assertEquals 发生在 run 第一次发生之前。

您可能想要返回线程并调用 join在生成的线程上进行测试，这将确保线程运行直到它死亡，可能会有很短的超时。

/* in the system under test */
@VisibleForTesting Thread generateAndReturnThread() {
  Thread thread = new Thread(this);
  thread.start();
  return thread;
}

public void generateThread() {
  generateAndReturnThread();
}

/* in the test */
@Test public void yourTest() {
  ThreadGenerator threadGenerator = new ThreadGenerator();
  threadGenerator.setRequestLimit(1);

  // wait up to a second for thread to complete
  threadGenerator.generateThreadAndReturn().join(1000);

  assertEquals(threadGenerator.getRequests(), 1);
}

旁注:考虑 AtomicInteger对于您的请求类，如果多个线程可能会修改请求。这将有助于防止两个不同的线程同时修改请求并相互覆盖。