Fix: Rollout experiment mapping issue

Author: rashidsp

lib/optimizely/project_config.rb

@@ -55,6 +55,7 @@ class ProjectConfig
     attr_reader :group_key_map
     attr_reader :rollout_id_map
     attr_reader :rollout_experiment_id_map
+    attr_reader :rollout_experiment_key_map
     attr_reader :variation_id_map
     attr_reader :variation_id_to_variable_usage_map
     attr_reader :variation_key_map
@@ -120,13 +121,15 @@ def initialize(datafile, logger, error_handler)
         end
       end
       @rollout_id_map = generate_key_map(@rollouts, 'id')
-      # split out the experiment id map for rollouts
+      # split out the experiment id and key map for rollouts
       @rollout_experiment_id_map = {}
+      @rollout_experiment_key_map = {}
       @rollout_id_map.each_value do |rollout|
         exps = rollout.fetch('experiments')
         @rollout_experiment_id_map = @rollout_experiment_id_map.merge(generate_key_map(exps, 'id'))
+        @rollout_experiment_key_map = @rollout_experiment_key_map.merge(generate_key_map(exps, 'key'))
       end
-      @all_experiments = @experiment_key_map.merge(@rollout_experiment_id_map)
+      @all_experiments = @experiment_key_map.merge(@rollout_experiment_key_map)
       @all_experiments.each do |key, exp|
         variations = exp.fetch('variations')
         variations.each do |variation|

spec/project_config_spec.rb

@@ -264,6 +264,14 @@
             feature_enabled => true,
             'variables' => []
           }
+        },
+        'rollout_exp_with_diff_id_and_key' => {
+          '177781' => {
+            'id' => '177781',
+            'key' => 'rollout_var_with_diff_id_and_key',
+            feature_enabled => true,
+            'variables' => []
+          }
         }
       }
 
@@ -448,6 +456,14 @@
             feature_enabled => true,
             'variables' => []
           }
+        },
+        'rollout_exp_with_diff_id_and_key' => {
+          'rollout_var_with_diff_id_and_key' => {
+            'id' => '177781',
+            'key' => 'rollout_var_with_diff_id_and_key',
+            feature_enabled => true,
+            'variables' => []
+          }
         }
       }
 
@@ -641,7 +657,8 @@
             'value' => 'false'
           }
         },
-        '177780' => {}
+        '177780' => {},
+        '177781' => {}
       }
 
       expected_rollout_id_map = {
@@ -654,7 +671,17 @@
         '177772' => config_body['rollouts'][0]['experiments'][1],
         '177776' => config_body['rollouts'][0]['experiments'][2],
         '177774' => config_body['rollouts'][1]['experiments'][0],
-        '177779' => config_body['rollouts'][1]['experiments'][1]
+        '177779' => config_body['rollouts'][1]['experiments'][1],
+        '177780' => config_body['rollouts'][1]['experiments'][2]
+      }
+
+      expected_rollout_experiment_key_map = {
+        '177770' => config_body['rollouts'][0]['experiments'][0],
+        '177772' => config_body['rollouts'][0]['experiments'][1],
+        '177776' => config_body['rollouts'][0]['experiments'][2],
+        '177774' => config_body['rollouts'][1]['experiments'][0],
+        '177779' => config_body['rollouts'][1]['experiments'][1],
+        'rollout_exp_with_diff_id_and_key' => config_body['rollouts'][1]['experiments'][2]
       }
 
       expect(project_config.attribute_key_map).to eq(expected_attribute_key_map)
@@ -669,6 +696,7 @@
       expect(project_config.variation_id_to_variable_usage_map).to eq(expected_variation_id_to_variable_usage_map)
       expect(project_config.rollout_id_map).to eq(expected_rollout_id_map)
       expect(project_config.rollout_experiment_id_map).to eq(expected_rollout_experiment_id_map)
+      expect(project_config.rollout_experiment_key_map).to eq(expected_rollout_experiment_key_map)
     end
 
     it 'should initialize properties correctly upon creating project with typed audience dict' do

spec/spec_params.rb

@@ -659,6 +659,22 @@ module OptimizelySpec
           'entityId' => '177780',
           'endOfRange' => 1500
         }]
+      }, {
+        'id' => '177780',
+        'key' => 'rollout_exp_with_diff_id_and_key',
+        'status' => 'Running',
+        'layerId' => '166661',
+        'audienceIds' => [],
+        'variations' => [{
+          'id' => '177781',
+          'key' => 'rollout_var_with_diff_id_and_key',
+          'featureEnabled' => true,
+          'variables' => []
+        }],
+        'trafficAllocation' => [{
+          'entityId' => '177781',
+          'endOfRange' => 1500
+        }]
       }]
     }]
   }.freeze