在套件中使用 @BeforeClass @Before 编写 JUnit 测试

Question

我正在使用 JUnit 4 来测试带有内存数据库的后端系统。我正在使用 @BeforeClass @Before @After 和 @AfterClass。

到目前为止，它在班级级别上运行良好。

@BeforeClass 包含缓慢的数据库设置，但每个测试会话只需执行一次。

@Before 只是为下一次测试擦干净。速度挺快的。

我的测试看起来像这样：

class CompanyTest  {

    @BeforeClass
    public static void beforeAll() throws Exception {
        BeforeAll.run(); //Setup In Memory Database!!! Very time intensive!!!
    }
    @Before
    public void beforeTest() throws Exception {
        BeforeTest.run(); //Setup data in database. Relatively quick
    }


    @Test
    public void testCreateCompany() throws Exception {
        ///...
    }
    @Test
    public void testDeleteCompany() throws Exception {
        ///...
    }
    @Test
    public void testAdminCompany() throws Exception {
        ///...
    }


    @After
    public void afterTest() {
        AfterTest.run(); //clear data
    }
    @AfterClass
    public static void afterAll() throws Exception {
        AfterAll.run(); //tear down database
    }
}

到目前为止，它在班级级别上运行良好。

我可以右键单击（在 Eclipse 中）单个测试，它将运行 @BeforeClass，然后运行 ​​@Before。

我也可以（在 Eclipse 中）单击类本身，它只会运行一次 @BeforeClass，然后在每次测试之前运行 @Before。

....但是这个原则如何扩展到套件级别？

我想在我的套件中的所有课程之前运行@BeforeClass。如果我这样写我的套件：

@Suite.SuiteClasses({ CompanyTest.class, CustomerTest.class, SomeOtherTest.class, })

public class AllTests {

    @BeforeClass
    public static void beforeAll() throws Exception {
        BeforeAll.run();
    }

    @AfterClass
    public static void afterAll() throws Exception {
        AfterAll.run();
    }
}

...我需要从我的所有测试类中删除 @BeforeClass。这很烦人，因为我有很多测试类，我不想删除我的 @BeforeClass 因为我想单独测试它们。

我的基本意思是：

是否有一种简单的方法（在 IDE 中单击鼠标）在 (a) 方法级别、(b) 类级别和 (c) 套件级别测试 JUnit 测试，同时保持会话级别设置和拆卸过程？

Answer 1

您需要的是通过不同的入口点管理全局状态。在您的 BeforeAll 或 AfterAll 类中，您可以保留一个静态引用，即 AtomicBoolean 来跟踪您是否已经执行它。

现在的问题 - beforeAll 和 afterAll 有两个单独的类 - 哪个类应该负责那个标志。

因此，我建议您定义一个 Rule，其中包含所有设置和拆卸逻辑（基本上是 BeforeAll 和 AfterAll 的所有代码）并管理用于跟踪初始化的全局状态。

基本上是这样的

class MyRule implements TestRule {
    static final AtomicBoolean INITIALIZED = new AtomicBoolean();

    @Override
    public Statement apply(final Statement base, final Description description) {

        return new Statement() {

            @Override
            public void evaluate() throws Throwable {
                boolean needsTearDown = false;
                try {
                    if (INITIALIZED.compareAndSet(false, true)) {
                        BeforeAll.run();
                        needsTearDown = true;
                    }
                    base.evaluate();
                } finally {
                    if (needsTearDown) {
                        AfterAll.run();
                        INITIALIZED.set(false);
                    }
                }
            }
        };
    }
}

在您的测试和套件中添加

class YourTestOrSuite {

  @ClassRule
  public static MyRule rule = new MyRule();
}

Answer 2

所以我有一个类似的问题（数据库初始化、连接池、缓存数据等），需要在运行任何和所有测试之前完成一次。这通常在应用程序服务器启动时完成。对于需要这个的测试，我添加了一个基类：

public class BaseInitializedTest {
    private boolean isInitialized;

    @BeforeClass
    public static void init() {
        doOnetimeInit();

        // Do other per unique class initialization here
    }

    private static synchronized void doOnetimeInit() {
        if (!isInitialized) {
            // Do you one time init here

            isInitialized = true;
        }
    }

    @AfterClass
    public static void deinit() {
        // Cleanup
    }
}

然后是我需要初始化的测试：

public class SomethingTest extends BaseInitializedTest {
    @Test
    public testSomething() {
        // Your test here
    }
}

并非所有测试都需要初始化堆栈，但那些需要初始化的测试将从这个基类继承并正确处理 init。希望对您有所帮助。