Fix decision test

Author: esrakartalOpt

core-api/src/test/java/com/optimizely/ab/notification/DecisionNotificationTest.java

@@ -37,13 +37,15 @@ public class DecisionNotificationTest {
 
     private static final Boolean FEATURE_ENABLED = Boolean.FALSE;
     private static final String EXPERIMENT_KEY = "experimentKey";
+    private static final String EXPERIMENT_ID = "1234567";
     private static final String FEATURE_KEY = "featureKey";
     private static final String FEATURE_VARIABLE_KEY = "featureVariableKey";
     private static final String FEATURE_TEST = "featureTest";
     private static final String FEATURE_TEST_VARIATION = "featureTestVariation";
     private static final String USER_ID = "userID";
     private static final Map<String, String> USER_ATTRIBUTES = Collections.singletonMap("user", "attr");
     private static final Variation VARIATION = mock(Variation.class);
+    private static final String VARIATION_ID = "1234567";
 
     private FeatureTestSourceInfo featureTestSourceInfo;
     private RolloutSourceInfo rolloutSourceInfo;
@@ -58,6 +60,8 @@ public void setUp() {
             .withAttributes(USER_ATTRIBUTES)
             .withExperimentKey(EXPERIMENT_KEY)
             .withVariation(VARIATION)
+            .withExperimentId(EXPERIMENT_ID)
+            .withVariationId(VARIATION_ID)
             .withType(NotificationCenter.DecisionNotificationType.AB_TEST.toString())
             .build();
         featureTestSourceInfo = new FeatureTestSourceInfo(FEATURE_TEST, FEATURE_TEST_VARIATION);
@@ -107,6 +111,8 @@ public void testGetDecisionInfo() {
         HashMap<String, String> expectedExperimentDecisionInfo = new HashMap<>();
         expectedExperimentDecisionInfo.put(DecisionNotification.ExperimentDecisionNotificationBuilder.EXPERIMENT_KEY, EXPERIMENT_KEY);
         expectedExperimentDecisionInfo.put(DecisionNotification.ExperimentDecisionNotificationBuilder.VARIATION_KEY, VARIATION.getKey());
+        expectedExperimentDecisionInfo.put(DecisionNotification.ExperimentDecisionNotificationBuilder.EXPERIMENT_ID, EXPERIMENT_ID);
+        expectedExperimentDecisionInfo.put(DecisionNotification.ExperimentDecisionNotificationBuilder.VARIATION_ID, VARIATION_ID);
         assertEquals(expectedExperimentDecisionInfo, experimentDecisionNotification.getDecisionInfo());
 
         // Assert for Feature's DecisionInfo
@@ -128,7 +134,7 @@ public void testGetDecisionInfo() {
 
     @Test
     public void testToString() {
-        assertEquals("DecisionNotification{type='ab-test', userId='userID', attributes={user=attr}, decisionInfo={experimentKey=experimentKey, variationKey=null}}", experimentDecisionNotification.toString());
+        assertEquals("DecisionNotification{type='ab-test', userId='userID', attributes={user=attr}, decisionInfo={experimentKey=experimentKey, variationKey=null, experimentId='1234567', variationId='1234567'}}", experimentDecisionNotification.toString());
         assertEquals("DecisionNotification{type='feature', userId='userID', attributes={user=attr}, decisionInfo={featureEnabled=false, sourceInfo={experimentKey=featureTest, variationKey=featureTestVariation}, source=feature-test, featureKey=featureKey}}", featureDecisionNotification.toString());
         assertEquals("DecisionNotification{type='feature-variable', userId='userID', attributes={user=attr}, decisionInfo={variableType=string, featureEnabled=true, sourceInfo={}, variableValue=null, variableKey=featureVariableKey, source=rollout, featureKey=featureKey}}", featureVariableDecisionNotification.toString());
     }