From a280ae6b7fe8da892a7261175ecc061608bdc373 Mon Sep 17 00:00:00 2001
From: Tyler Benson <tyler.benson@datadoghq.com>
Date: Fri, 17 Jan 2020 11:05:44 -0800
Subject: [PATCH] Check for both regular and lowercase names with decorator
 enabled flags.

---
 dd-trace-api/src/main/java/datadog/trace/api/Config.java | 3 ++-
 .../opentracing/decorators/SpanDecoratorTest.groovy      | 9 +++++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/dd-trace-api/src/main/java/datadog/trace/api/Config.java b/dd-trace-api/src/main/java/datadog/trace/api/Config.java
index c50d4026fb..cbaca72e7e 100644
--- a/dd-trace-api/src/main/java/datadog/trace/api/Config.java
+++ b/dd-trace-api/src/main/java/datadog/trace/api/Config.java
@@ -591,7 +591,8 @@ public class Config {
   }
 
   public boolean isDecoratorEnabled(final String name) {
-    return getBooleanSettingFromEnvironment("trace." + name.toLowerCase() + ".enabled", true);
+    return getBooleanSettingFromEnvironment("trace." + name + ".enabled", true)
+        && getBooleanSettingFromEnvironment("trace." + name.toLowerCase() + ".enabled", true);
   }
 
   /**
diff --git a/dd-trace-ot/src/test/groovy/datadog/opentracing/decorators/SpanDecoratorTest.groovy b/dd-trace-ot/src/test/groovy/datadog/opentracing/decorators/SpanDecoratorTest.groovy
index ceb265e0ff..87744428b3 100644
--- a/dd-trace-ot/src/test/groovy/datadog/opentracing/decorators/SpanDecoratorTest.groovy
+++ b/dd-trace-ot/src/test/groovy/datadog/opentracing/decorators/SpanDecoratorTest.groovy
@@ -457,7 +457,7 @@ class SpanDecoratorTest extends DDSpecification {
   def "disable decorator via config"() {
     setup:
     ConfigUtils.updateConfig {
-      System.setProperty("dd.trace." + PeerServiceDecorator.getSimpleName().toLowerCase() + ".enabled", "false")
+      System.setProperty("dd.trace.${decorator}.enabled", "false")
     }
 
     tracer = DDTracer.builder()
@@ -475,8 +475,13 @@ class SpanDecoratorTest extends DDSpecification {
 
     cleanup:
     ConfigUtils.updateConfig {
-      System.clearProperty("dd.trace." + PeerServiceDecorator.getSimpleName().toLowerCase() + ".enabled")
+      System.clearProperty("dd.trace.${decorator}.enabled")
     }
+
+    where:
+    decorator                                          | _
+    PeerServiceDecorator.getSimpleName().toLowerCase() | _
+    PeerServiceDecorator.getSimpleName()               | _
   }
 
   def "disabling service decorator does not disable split by tags"() {