🖊️ accommodate case where featureEnabled not provided

optimizely/entities.py

@@ -109,7 +109,10 @@ def __init__(self, id, value, **kwards):
       self.id = id
       self.value = value
 
-  def __init__(self, id, key, featureEnabled=False, variables=None, **kwargs):
+  def __init__(self, id, key, featureEnabled=None, variables=None, **kwargs):
+    if featureEnabled is None:
+      featureEnabled = False
+
     self.id = id
     self.key = key
     self.featureEnabled = featureEnabled

tests/test_config.py

@@ -985,6 +985,25 @@ def test_init__with_v4_datafile(self):
     self.assertEqual(expected_rollout_id_map, project_config.rollout_id_map)
     self.assertEqual(expected_variation_variable_usage_map, project_config.variation_variable_usage_map)
 
+  def test_variation_has_featureEnabled_false_if_prop_undefined(self):
+    """ Test that featureEnabled property by default is set to False, when not given in the data file"""
+    variation = {
+            'key': 'group_exp_1_variation',
+            'id': '28902',
+            'variables': [{
+              'id': '128',
+              'value': 'stage'
+            }, {
+              'id': '129',
+              'value': '112'
+            }, {
+              'id': '130',
+              'value': '1.211'
+            }]}
+
+    variation_entity = entities.Variation(**variation)
+    self.assertFalse(variation_entity.featureEnabled)
+
   def test_get_version(self):
     """ Test that JSON version is retrieved correctly when using get_version. """