Skip to content

Commit 929d1c2

Browse files
committed
simplifying extension
1 parent 6405ab7 commit 929d1c2

1 file changed

Lines changed: 3 additions & 13 deletions

File tree

utils/config-utils/src/testFixtures/java/datadog/trace/test/ConfigInversionExtension.java

Lines changed: 3 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -17,28 +17,18 @@
1717
*/
1818
public class ConfigInversionExtension implements BeforeAllCallback, AfterAllCallback {
1919

20-
private static final ExtensionContext.Namespace NS =
21-
ExtensionContext.Namespace.create("dd", "config-inversion");
22-
23-
private static final String PREVIOUS_POLICY = "previousPolicy";
20+
private StrictnessPolicy previousPolicy;
2421

2522
@Override
2623
public void beforeAll(ExtensionContext ctx) {
27-
StrictnessPolicy previous = ConfigHelper.get().configInversionStrictFlag();
28-
ctx.getStore(NS).put(PREVIOUS_POLICY, previous);
24+
previousPolicy = ConfigHelper.get().configInversionStrictFlag();
2925
ConfigHelper.get().setConfigInversionStrict(StrictnessPolicy.STRICT_TEST);
3026
}
3127

3228
@Override
3329
public void afterAll(ExtensionContext ctx) {
3430
List<String> unsupported = ConfigHelper.get().drainUnsupportedConfigs();
35-
36-
StrictnessPolicy previous = ctx.getStore(NS).get(PREVIOUS_POLICY, StrictnessPolicy.class);
37-
if (previous != null) {
38-
ConfigHelper.get().setConfigInversionStrict(previous);
39-
} else {
40-
ConfigHelper.get().setConfigInversionStrict(StrictnessPolicy.WARNING);
41-
}
31+
ConfigHelper.get().setConfigInversionStrict(previousPolicy);
4232

4333
if (!unsupported.isEmpty()) {
4434
throw new AssertionError(

0 commit comments

Comments
 (0)