Adding unit test

sarrubia · sarrubia · commit 5c0640a3de62 · 2017-05-25T14:11:59.000-03:00
diff --git a/splitio/tests/test_splits.py b/splitio/tests/test_splits.py
@@ -1063,6 +1063,12 @@ def setUp(self):
         raw_split['trafficAllocationSeed'] = -1
         self._splitObjects['rollout2'] = split_parser.parse(raw_split, True)
 
+        raw_split['name'] = 'test4'
+        raw_split['trafficAllocation'] = None #must be mapped as 100
+        raw_split['trafficAllocationSeed'] = -1
+        self._splitObjects['rollout3'] = split_parser.parse(raw_split, True)
+
+
     def testTrafficAllocation(self):
         '''
         '''
@@ -1080,3 +1086,8 @@ def testTrafficAllocation(self):
             self._splitObjects['rollout2'], 'testKey', None
         )
         self.assertEqual(treatment3, 'default')
+
+        treatment4, label1 = self._client._get_treatment_for_split(
+            self._splitObjects['rollout3'], 'testKey', None
+        )
+        self.assertEqual(treatment4, 'on')
