use tests

DataDog · Apr 11, 2024 · d7c2086 · d7c2086
1 parent 8473aab
commit d7c2086
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/test/new-e2e/tests/apm/docker_test.go b/test/new-e2e/tests/apm/docker_test.go
@@ -113,6 +113,18 @@ func (s *DockerFakeintakeSuite) TestAutoVersionStats() {
 	}, 2*time.Minute, 10*time.Second, "Failed finding version tags")
 }
 
+func (s *DockerFakeintakeSuite) TestIsTraceRootTag() {
+	err := s.Env().FakeIntake.Client().FlushServerAndResetAggregators()
+	s.Require().NoError(err)
+
+	service := fmt.Sprintf("tracegen-auto-is-trace-root-tag-%s", s.transport)
+	defer waitTracegenShutdown(&s.Suite, s.Env().FakeIntake)
+	defer runTracegenDocker(s.Env().RemoteHost, service, tracegenCfg{transport: s.transport})()
+	s.EventuallyWithTf(func(c *assert.CollectT) {
+		testIsTraceRootTag(s.T(), c, s.Env().FakeIntake)
+	}, 2*time.Minute, 10*time.Second, "Failed finding is_trace_root tag")
+}
+
 func (s *DockerFakeintakeSuite) TestStatsForService() {
 	err := s.Env().FakeIntake.Client().FlushServerAndResetAggregators()
 	s.Require().NoError(err)

diff --git a/test/new-e2e/tests/apm/vm_test.go b/test/new-e2e/tests/apm/vm_test.go
@@ -220,6 +220,29 @@ func (s *VMFakeintakeSuite) TestAutoVersionStats() {
 	}, 3*time.Minute, 10*time.Second, "Failed finding stats")
 }
 
+func (s *VMFakeintakeSuite) TestIsTraceRootTag() {
+	err := s.Env().FakeIntake.Client().FlushServerAndResetAggregators()
+	s.Require().NoError(err)
+
+	service := fmt.Sprintf("tracegen-stats-%s", s.transport)
+
+	// Wait for agent to be live
+	s.T().Log("Waiting for Trace Agent to be live.")
+	s.Require().NoError(waitRemotePort(s, 8126))
+
+	// Run Trace Generator
+	s.T().Log("Starting Trace Generator.")
+	defer waitTracegenShutdown(&s.Suite, s.Env().FakeIntake)
+	shutdown := runTracegenDocker(s.Env().RemoteHost, service, tracegenCfg{transport: s.transport})
+	defer shutdown()
+
+	s.EventuallyWithTf(func(c *assert.CollectT) {
+		s.logStatus()
+		testIsTraceRootTag(s.T(), c, s.Env().FakeIntake)
+		s.logJournal()
+	}, 3*time.Minute, 10*time.Second, "Failed finding stats")
+}
+
 func (s *VMFakeintakeSuite) TestBasicTrace() {
 	err := s.Env().FakeIntake.Client().FlushServerAndResetAggregators()
 	s.Require().NoError(err)