dd-trace-java/dd-java-agent/agent-tooling/src/test/java/datadog/trace/agent/tooling/InstrumenterModuleTest.java at bb5d6cf7589f7a2e495fa178317684c9a6f7b49c · DataDog/dd-trace-java · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package datadog.trace.agent.tooling;

import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertTrue;

import datadog.trace.agent.tooling.InstrumenterModule.TargetSystem;
import datadog.trace.test.junit.utils.config.WithConfig;
import datadog.trace.test.junit.utils.config.WithConfigExtension;
import java.util.HashSet;
import java.util.Set;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;

@ExtendWith(WithConfigExtension.class)
class InstrumenterModuleTest {

  @Test
  void testDataStreamsIsApplicableWithTracing() {
    Set<TargetSystem> enabledSystems = new HashSet<>();
    enabledSystems.add(TargetSystem.TRACING);

    InstrumenterModule.DataStreams module = new InstrumenterModule.DataStreams("test-module") {};

    assertTrue(module.isApplicable(enabledSystems));
  }

  @Test
  void testDataStreamsIsApplicableWithDataStreams() {
    Set<TargetSystem> enabledSystems = new HashSet<>();
    enabledSystems.add(TargetSystem.DATA_STREAMS);

    InstrumenterModule.DataStreams module = new InstrumenterModule.DataStreams("test-module") {};

    assertTrue(module.isApplicable(enabledSystems));
  }

  @Test
  void testDataStreamsIsApplicableWithBoth() {
    Set<TargetSystem> enabledSystems = new HashSet<>();
    enabledSystems.add(TargetSystem.TRACING);
    enabledSystems.add(TargetSystem.DATA_STREAMS);

    InstrumenterModule.DataStreams module = new InstrumenterModule.DataStreams("test-module") {};

    assertTrue(module.isApplicable(enabledSystems));
  }

  @Test
  void testDataStreamsIsApplicableWithNeither() {
    Set<TargetSystem> enabledSystems = new HashSet<>();
    enabledSystems.add(TargetSystem.APPSEC);

    InstrumenterModule.DataStreams module = new InstrumenterModule.DataStreams("test-module") {};

    assertFalse(module.isApplicable(enabledSystems));
  }

  @Test
  @WithConfig(key = "trace.test-kafka-module.enabled", value = "false")
  @WithConfig(key = "data.streams.enabled", value = "true")
  void testDataStreamsIsEnabledWhenDataStreamsEnabledOverridesFalse() {
    // When tracing for this integration is disabled but DSM is explicitly enabled,
    // isEnabled() should still return true.
    InstrumenterModule.DataStreams module =
        new InstrumenterModule.DataStreams("test-kafka-module") {};

    assertTrue(module.isEnabled());
  }

  @Test
  @WithConfig(key = "trace.test-kafka-module.enabled", value = "true")
  @WithConfig(key = "data.streams.enabled", value = "false")
  void testDataStreamsIsEnabledWhenSuperEnabledIsTrue() {
    // When super.isEnabled() is true, isEnabled() should return true regardless of DSM state.
    InstrumenterModule.DataStreams module =
        new InstrumenterModule.DataStreams("test-kafka-module") {};

    assertTrue(module.isEnabled());
  }

  @Test
  @WithConfig(key = "trace.test-kafka-module.enabled", value = "true")
  @WithConfig(key = "data.streams.enabled", value = "true")
  void testDataStreamsIsEnabledWhenBothEnabled() {
    // When both super.isEnabled() and DSM are enabled, isEnabled() should return true.
    InstrumenterModule.DataStreams module =
        new InstrumenterModule.DataStreams("test-kafka-module") {};

    assertTrue(module.isEnabled());
  }

  @Test
  @WithConfig(key = "trace.test-kafka-module.enabled", value = "false")
  @WithConfig(key = "data.streams.enabled", value = "false")
  void testDataStreamsIsEnabledWhenBothDisabled() {
    // When both super.isEnabled() and DSM are disabled, isEnabled() should return false.
    InstrumenterModule.DataStreams module =
        new InstrumenterModule.DataStreams("test-kafka-module") {};

    assertFalse(module.isEnabled());
  }
}