Improve tests

DataDog · Apr 19, 2021 · c43d1d3 · c43d1d3
1 parent 1aa0f15
commit c43d1d3
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/profiler/profiler_test.go b/profiler/profiler_test.go
@@ -58,13 +58,21 @@ func TestStart(t *testing.T) {
 		mu.Unlock()
 	})
 
-	t.Run("options/GoodAPIKey", func(t *testing.T) {
+	t.Run("options/GoodAPIKey/Agent", func(t *testing.T) {
+		err := Start(WithAPIKey("12345678901234567890123456789012"))
+		defer Stop()
+		assert.Nil(t, err)
+		assert.Equal(t, activeProfiler.cfg.agentURL, activeProfiler.cfg.targetURL)
+	})
+
+	t.Run("options/GoodAPIKey/Agentless", func(t *testing.T) {
 		err := Start(
 			WithAPIKey("12345678901234567890123456789012"),
 			WithAgentlessUpload(),
 		)
 		defer Stop()
 		assert.Nil(t, err)
+		assert.Equal(t, activeProfiler.cfg.apiURL, activeProfiler.cfg.targetURL)
 	})
 
 	t.Run("options/BadAPIKey", func(t *testing.T) {