Add tests

DataDog · Jul 21, 2020 · 041ea03 · 041ea03
1 parent 1b10a40
commit 041ea03
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 0 deletions.
diff --git a/ddtrace/tracer/option.go b/ddtrace/tracer/option.go
@@ -143,6 +143,8 @@ func newConfig(opts ...StartOption) *config {
 	c.logStartup = internal.BoolEnv("DD_TRACE_STARTUP_LOGS", true)
 	if internal.BoolEnv("DD_TRACE_ANALYTICS_ENABLED", false) {
 		globalconfig.SetAnalyticsRate(1.0)
+	} else {
+		globalconfig.SetAnalyticsRate(math.NaN())
 	}
 	c.runtimeMetrics = internal.BoolEnv("DD_RUNTIME_METRICS_ENABLED", false)
 	c.debug = internal.BoolEnv("DD_TRACE_DEBUG", false)

diff --git a/ddtrace/tracer/option_test.go b/ddtrace/tracer/option_test.go
@@ -42,6 +42,7 @@ func TestTracerOptionsDefaults(t *testing.T) {
 
 	t.Run("analytics", func(t *testing.T) {
 		assert := assert.New(t)
+		defer globalconfig.SetAnalyticsRate(math.NaN())
 		assert.True(math.IsNaN(globalconfig.AnalyticsRate()))
 		newTracer(WithAnalyticsRate(0.5))
 		assert.Equal(0.5, globalconfig.AnalyticsRate())
@@ -51,6 +52,27 @@ func TestTracerOptionsDefaults(t *testing.T) {
 		assert.Equal(1., globalconfig.AnalyticsRate())
 	})
 
+	t.Run("analyticsEnv", func(t *testing.T) {
+		assert := assert.New(t)
+		assert.True(math.IsNaN(globalconfig.AnalyticsRate()))
+		os.Setenv("DD_TRACE_ANALYTICS_ENABLED", "true")
+		defer os.Unsetenv("DD_TRACE_ANALYTICS_ENABLED")
+		newTracer()
+		assert.Equal(1.0, globalconfig.AnalyticsRate())
+		os.Setenv("DD_TRACE_ANALYTICS_ENABLED", "false")
+		newTracer()
+		assert.True(math.IsNaN(globalconfig.AnalyticsRate()))
+		os.Setenv("DD_TRACE_ANALYTICS_ENABLED", "true")
+		newTracer(WithAnalytics(false))
+		assert.True(math.IsNaN(globalconfig.AnalyticsRate()))
+		os.Setenv("DD_TRACE_ANALYTICS_ENABLED", "false")
+		newTracer(WithAnalytics(true))
+		assert.Equal(1.0, globalconfig.AnalyticsRate())
+		os.Setenv("DD_TRACE_ANALYTICS_ENABLED", "false")
+		newTracer(WithAnalyticsRate(0.5))
+		assert.Equal(0.5, globalconfig.AnalyticsRate())
+	})
+
 	t.Run("dogstatsd", func(t *testing.T) {
 		t.Run("default", func(t *testing.T) {
 			tracer := newTracer()

diff --git a/ddtrace/tracer/tracer_test.go b/ddtrace/tracer/tracer_test.go
@@ -226,6 +226,24 @@ func TestTracerRuntimeMetrics(t *testing.T) {
 		assert.Contains(t, tp.Lines()[0], "DEBUG: Runtime metrics enabled")
 	})
 
+	t.Run("env", func(t *testing.T) {
+		os.Setenv("DD_RUNTIME_METRICS_ENABLED", "true")
+		defer os.Unsetenv("DD_RUNTIME_METRICS_ENABLED")
+		tp := new(testLogger)
+		tracer := newTracer(WithLogger(tp), WithDebugMode(true))
+		defer tracer.Stop()
+		assert.Contains(t, tp.Lines()[0], "DEBUG: Runtime metrics enabled")
+	})
+
+	t.Run("overrideEnv", func(t *testing.T) {
+		os.Setenv("DD_RUNTIME_METRICS_ENABLED", "false")
+		defer os.Unsetenv("DD_RUNTIME_METRICS_ENABLED")
+		tp := new(testLogger)
+		tracer := newTracer(WithRuntimeMetrics(), WithLogger(tp), WithDebugMode(true))
+		defer tracer.Stop()
+		assert.Contains(t, tp.Lines()[0], "DEBUG: Runtime metrics enabled")
+	})
+
 	t.Run("off", func(t *testing.T) {
 		tp := new(testLogger)
 		tracer := newTracer(WithLogger(tp), WithDebugMode(true))