test: record an exposure for the control arm under local evaluation

Author: gagantrivedi

src/test/java/com/flagsmith/FlagsmithClientTest.java

@@ -57,6 +57,7 @@
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.CsvSource;
 import org.junit.jupiter.params.provider.MethodSource;
 import org.mockito.ArgumentCaptor;
 import org.mockito.Mockito;
@@ -1202,21 +1203,26 @@ private static FlagsmithClient localExperimentClient(EventProcessor processor) {
         return client;
     }
 
-    @Test
-    public void testGetExperimentFlagRecordsAnExposureWithLocalEvaluationWhenEnrolled()
+    @ParameterizedTest
+    @CsvSource({
+        "user-1, new checkout, treatment",
+        "user-3, control, control"
+    })
+    public void testGetExperimentFlagRecordsAnExposureWithLocalEvaluationWhenEnrolled(
+            String identifier, String expectedValue, String expectedVariant)
             throws FlagsmithClientError {
         EventProcessor processor = mock(EventProcessor.class);
         FlagsmithClient client = localExperimentClient(processor);
         Map<String, Object> traits = Collections.singletonMap("plan", "premium");
 
-        Flag flag = (Flag) client.getExperimentFlag("checkout", "user-1", traits);
+        Flag flag = (Flag) client.getExperimentFlag("checkout", identifier, traits);
 
-        assertEquals("new checkout", flag.getValue());
-        assertEquals("treatment", flag.getVariant());
+        assertEquals(expectedValue, flag.getValue());
+        assertEquals(expectedVariant, flag.getVariant());
         assertEquals(42, flag.getExperiment().getId());
         assertEquals("Checkout experiment", flag.getExperiment().getName());
         assertEquals(Boolean.TRUE, flag.getExperiment().getInExperiment());
-        verify(processor).trackExposureEvent("checkout", "user-1", "treatment", traits,
+        verify(processor).trackExposureEvent("checkout", identifier, expectedVariant, traits,
                 Collections.singletonMap("experiment_id", 42));
     }
 

src/test/java/com/flagsmith/FlagsmithTestHelper.java

@@ -456,7 +456,7 @@ public static String getIdentitiesFlagsWithExperiment() {
 
   /**
    * An environment document running an experiment on {@code checkout}. Its rollout segment is
-   * identities on the premium plan, who are served the {@code treatment} variant.
+   * identities on the premium plan, split evenly between {@code control} and {@code treatment}.
    * {@code user_overridden} has an identity override for {@code checkout}, and {@code plain} has no
    * experiment.
    */
@@ -473,7 +473,7 @@ public static String getEnvironmentDocumentWithExperiment() {
         + "   \"feature_states\": [{\"id\": 2, \"featurestate_uuid\": \"c2c2c2c2-c2c2-c2c2-c2c2-c2c2c2c2c2c2\",\n"
         + "    \"feature\": " + checkout + ", \"feature_state_value\": \"control\", \"enabled\": true,\n"
         + "    \"feature_segment\": {\"priority\": 0},\n"
-        + "    \"multivariate_feature_state_values\": [{\"id\": 3, \"percentage_allocation\": 100,\n"
+        + "    \"multivariate_feature_state_values\": [{\"id\": 3, \"percentage_allocation\": 50,\n"
         + "     \"mv_fs_value_uuid\": \"3e3e3e3e-3e3e-3e3e-3e3e-3e3e3e3e3e3e\",\n"
         + "     \"multivariate_feature_option\": {\"id\": 3, \"value\": \"new checkout\", \"key\": \"treatment\"}}],\n"
         + "    \"metadata\": {\"experiment\": " + String.format(experiment, "true") + "}}]}]},\n"