groovy - 在类的所有实例方法中隐式使用 Groovy Category

Question

我有简单的 Groovy 类别类，它向 String 实例添加方法:

final class SampleCategory {

    static String withBraces(String self) {
        "($self)"
    }

}

我想在我的单元测试中使用这个类别(例如)。它看起来像这样:

class MyTest {
    @Test
    void shouldDoThis() {
        use (SampleCategory) {
            assert 'this'.withBraces() == '(this)'
        }
    }

    @Test
    void shouldDoThat() {
        use (SampleCategory) {
            assert 'that'.withBraces() == '(that)'
        }
    }
}

然而，我想要实现的是能够指定该类别 SampleCategory用于 MyTest 的每个实例方法的范围内所以我不必指定 use(SampleCategory) { ... }在每种方法中。

是否可以？

Answer 1

您可以使用 mixin 将类别直接应用于 String 的元类。为元类分配 null 以将其重置为常规默认值。例如:

@Before void setUp() { 
    String.mixin(SampleCategory)
}

@After void tearDown() {
    String.metaClass = null
}

@Test
void shouldDoThat() {
    assert 'that'.withBraces() == '(that)'
}