feat(test): added failing test for RemoveUnusedLocalVariables

openrewrite · Oct 20, 2024 · 8b1b29b · 8b1b29b
1 parent b030670
commit 8b1b29b
Showing 1 changed file with 40 additions and 0 deletions.
diff --git a/src/test/java/org/openrewrite/featureflags/launchdarkly/RemoveBoolVariationTest.java b/src/test/java/org/openrewrite/featureflags/launchdarkly/RemoveBoolVariationTest.java
@@ -178,6 +178,46 @@ void bar() {
         );
     }
 
+    @Test
+    void removeUnusedLDContextWithBuilder() {
+        rewriteRun(
+          // language=java
+          java(
+            """
+              import com.launchdarkly.sdk.*;
+              import com.launchdarkly.sdk.server.*;
+              class Foo {
+                  LDClient client = new LDClient("sdk-key-123abc");
+                  void bar() {
+                      LDContext context = LDContext.builder("context-key-123abc")
+                        .name("Sandy")
+                        .build();
+                      if (client.boolVariation("flag-key-123abc", context, false)) {
+                          // Application code to show the feature
+                          System.out.println("Feature is on");
+                      }
+                      else {
+                        // The code to run if the feature is off
+                          System.out.println("Feature is off");
+                      }
+                  }
+              }
+              """,
+            """
+              import com.launchdarkly.sdk.*;
+              import com.launchdarkly.sdk.server.*;
+              class Foo {
+                  LDClient client = new LDClient("sdk-key-123abc");
+                  void bar() {
+                      // Application code to show the feature
+                      System.out.println("Feature is on");
+                  }
+              }
+              """
+          )
+        );
+    }
+
     @Test
     void enablePermanentlyWithParameters() {
         rewriteRun(