如何开发单元测试

单元测试目标
如何编写单元测试
惯例
应避免的情况
管理全局状态

单元测试目标

单元测试用于示例如何使用测试中的代码。
单元测试在对象失去功能时检测。
当对象被重构但仍满足相同功能时，单元测试不会打断它。

如何编写单元测试

1. 如果创建类的实例需要一些工作，创建一个@Before方法来执行常用的设置。该@Before方法在每个单元测试之前被自动运行。只有适用于所有测试才做通用设置。特定的测试设置应该在本地需要它的测试中完成。在这个例子中，我们测试BlockMaster，它依赖于一个日志，时钟和executor service。我们提供的executor service和日志是真正的实现，TestClock是可以由单元测试来控制的伪时钟。

@Before
public void before() throws Exception {
  Journal blockJournal = new ReadWriteJournal(mTestFolder.newFolder().getAbsolutePath());
  mClock = new TestClock();
  mExecutorService =
      Executors.newFixedThreadPool(2, ThreadFactoryUtils.build("TestBlockMaster-%d", true));
  mMaster = new BlockMaster(blockJournal, mClock, mExecutorService);
  mMaster.start(true);
}

2. 如果任何在@Before中创建的东西需要进行清理（例如一个BlockMaster），创建一个@After方法来做清理。此方法在每次测试后被自动调用。

@After
public void after() throws Exception {
  mMaster.stop();
}

3. 确定一个功能性元素来测试。你决定要测试的功能应该是公共API的一部分，而不应该与实现的细节相关。测试应专注于只测试一项功能。

4. 给你的测试起一个名字，来描述它测试的功能。被测试的功能最好足够简单，以便用一个名字就能表示，例如removeNonexistentBlockThrowsException, mkdirCreatesDirectory, 或者cannotMkdirExistingFile。

@Test
public void detectLostWorker() throws Exception {

5. 设置要测试的场景。这里我们注册一个worker，然后模拟一小时。HeartbeatScheduler部分确保丢失worker的heartbeat至少执行一次。

  // Register a worker.
  long worker1 = mMaster.getWorkerId(NET_ADDRESS_1);
  mMaster.workerRegister(worker1,
      ImmutableList.of("MEM"),
      ImmutableMap.of("MEM", 100L),
      ImmutableMap.of("MEM", 10L),
      NO_BLOCKS_ON_TIERS);

  // Advance the block master's clock by an hour so that the worker appears lost.
  mClock.setTimeMs(System.currentTimeMillis() + Constants.HOUR_MS);

  // Run the lost worker detector.
  HeartbeatScheduler.await(HeartbeatContext.MASTER_LOST_WORKER_DETECTION, 1, TimeUnit.SECONDS);
  HeartbeatScheduler.schedule(HeartbeatContext.MASTER_LOST_WORKER_DETECTION);
  HeartbeatScheduler.await(HeartbeatContext.MASTER_LOST_WORKER_DETECTION, 1, TimeUnit.SECONDS);

6. 检查类的执行是否正确：

  // Make sure the worker is detected as lost.
  Set<WorkerInfo> info = mMaster.getLostWorkersInfo();
  Assert.assertEquals(worker1, Iterables.getOnlyElement(info).getId());
}

7. 循环回到步骤＃3，直到类的整个公共API都已经被测试。

惯例

对src/main/java/ClassName.java的测试应该运行为src/test/java/ClassNameTest.java。
测试不需要处理或记录特定的检查异常，更倾向于简单地添加throws Exception到方法声明中。
目标是保持测试简明扼要而不需要注释帮助理解。

应避免的情况

避免随机性。边缘检测应被明确处理。
避免通过调用Thread.sleep()来等待。这会导致单元测试变慢，如果时间不够长，可能导致时间片的失败。
避免使用白盒测试，这会混乱测试对象的内部状态。如果你需要模拟一个依赖，改变对象，将依赖作为其构造函数的参数（见依赖注入）
避免低效测试。模拟花销较大的依赖关系，将各个测试时间控制在100ms以下。

管理全局状态

一个模块中的所有测试在同一个JVM上运行，所以妥善管理全局状态是非常重要的，这样测试不会互相干扰。全局状态包括系统性能，Alluxio配置，以及任何静态字段。我们对管理全局状态的解决方案是使用JUnit对@Rules的支持。

在测试过程中更改Alluxio配置

一些单元测试需要测试不同配置下的Alluxio。这需要修改全局Configuration对象。当在一个套件中的所有测试都需要配置参数时，设置某一个方式，用ConfigurationRule对它们进行设置。

@Rule
public ConfigurationRule mConfigurationRule = new ConfigurationRule(ImmutableMap.of(
    PropertyKey.key1, "value1",
    PropertyKey.key2, "value2"));

对于一个单独测试需要的配置更改，请使用ConfigurationRule#set(key, value), 这个方法造成的变化局限在调用方法的范围内：

@Rule
public ConfigurationRule mConfigurationRule = new ConfigurationRule(ImmutableMap.of(
    PropertyKey.key1, "value1",
    PropertyKey.key2, "value2"));

@Test
public void testSomething() {
  mConfigurationRule.set(PropertyKey.key1, "value3");
  // Now PropertyKey.key1 = "value3"
  ...
}

@Test
public void testAnotherThing() {
  // Now PropertyKey.key1 = "value1"
}

在测试过程中更改系统属性

如果你在测试套件期间需要更改一个系统属性，请使用SystemPropertyRule。

@Rule
public SystemPropertyRule mSystemPropertyRule = new SystemPropertyRule("propertyName", "value");

在一个特定的测试中设置系统属性，请在try-catch语句中使用SetAndRestoreSystemProperty：

@Test
public void test() {
  try (SetAndRestorySystemProperty p = new SetAndRestorySystemProperty("propertyKey", "propertyValue")) {
     // Test something with propertyKey set to propertyValue.
  }
}

其他全局状态

如果测试需要修改其它类型的全局状态，创建一个新的@Rule用于管理状态，这样就可以在测试中共享。这样的一个例子是TtlIntervalRule。

Categories