Merge pull request #114 from splitio/feature/oneSingleQueue

[SDKS-285]: Issue on DependencyMatcher

Author: mmelograno

splitio/matchers.py

@@ -10,7 +10,7 @@
 from six import string_types
 from splitio.transformers import AsDateHourMinuteTimestampTransformMixin, \
     AsNumberTransformMixin, AsDateTimestampTransformMixin, TransformMixin
-
+from splitio.key import Key
 
 DataType = Enum('DataType', 'DATETIME NUMBER')
 
@@ -714,12 +714,16 @@ def __init__(self, dependency_matcher_data):
     def match(self, key, attributes=None, client=None):
         """
         """
-        treatment = client.get_treatment(
-            key,
+        matching, bucketing = (key.matching_key, key.bucketing_key) \
+            if isinstance(key, Key) else (key, None)
+        treatment = client.evaluate_treatment(
             self._data.get('split'),
+            matching,
+            bucketing,
             attributes
         )
-        return treatment in self._data.get('treatments', [])
+
+        return treatment['treatment'] in self._data.get('treatments', [])
 
 
 class BooleanMatcher(object):

splitio/tests/test_matchers.py

@@ -1058,9 +1058,9 @@ def test_matcher_construction(self):
         '''
         self.assertIsInstance(self._matcher, DependencyMatcher)
 
-    def test_matcher_client_is_created_and_get_treatment_called(self):
+    def test_matcher_client_is_created_and_evaluate_treatment_called(self):
         self._matcher.match('abc', None, self._mock)
-        self._mock.get_treatment.assert_called_once_with('abc', 'someSplit', None)
+        self._mock.evaluate_treatment.assert_called_once_with('someSplit', 'abc', None, None)
         self.assertTrue(True)