Added support for FF to push.parser class

Author: chillaq

splitio/push/parser.py

@@ -7,7 +7,6 @@
 from splitio.util.time import utctime_ms
 from splitio.push.sse import SSE_EVENT_ERROR, SSE_EVENT_MESSAGE
 
-
 class EventType(Enum):
     """Event type enumeration."""
 
@@ -326,9 +325,12 @@ def change_number(self):
 class SplitChangeUpdate(BaseUpdate):
     """Split Change notification."""
 
-    def __init__(self, channel, timestamp, change_number):
+    def __init__(self, channel, timestamp, change_number, previous_change_number, split_definition, compression):
         """Class constructor."""
         BaseUpdate.__init__(self, channel, timestamp, change_number)
+        self._previous_change_number = previous_change_number
+        self._split_definition = split_definition
+        self._compression = compression
 
     @property
     def update_type(self):  # pylint:disable=no-self-use
@@ -340,6 +342,36 @@ def update_type(self):  # pylint:disable=no-self-use
         """
         return UpdateType.SPLIT_UPDATE
 
+    @property
+    def previous_change_number(self):  # pylint:disable=no-self-use
+        """
+        Return previous change number
+
+        :returns: The previous change number
+        :rtype: int
+        """
+        return self._previous_change_number
+
+    @property
+    def split_definition(self):  # pylint:disable=no-self-use
+        """
+        Return split definition
+
+        :returns: The new split definition
+        :rtype: str
+        """
+        return self._split_definition
+
+    @property
+    def compression(self):  # pylint:disable=no-self-use
+        """
+        Return previous compression type
+
+        :returns: The compression type
+        :rtype: int
+        """
+        return self._compression
+
     def __str__(self):
         """Return string representation."""
         return "SplitChange - changeNumber=%d" % (self.change_number)
@@ -472,7 +504,7 @@ def _parse_update(channel, timestamp, data):
     update_type = UpdateType(data['type'])
     change_number = data['changeNumber']
     if update_type == UpdateType.SPLIT_UPDATE:
-        return SplitChangeUpdate(channel, timestamp, change_number)
+        return SplitChangeUpdate(channel, timestamp, change_number, data.get('pcn'), data.get('d'), data.get('c'))
     elif update_type == UpdateType.SPLIT_KILL:
         return SplitKillUpdate(channel, timestamp, change_number,
                                data['splitName'], data['defaultTreatment'])

tests/push/test_parser.py

@@ -57,13 +57,28 @@ def test_event_parsing(self):
         assert parsed0.change_number == 1591996754396
         assert parsed0.split_name == 'test'
 
+        e1 = make_message(
+            'NDA5ODc2MTAyNg==_MzAyODY0NDkyOA==_splits',
+            {'type':'SPLIT_UPDATE','changeNumber':1591996685190, 'pcn': 12, 'c': 2, 'd': 'eJzEUtFu2kAQ/BU0z4d0hw2Be0MFRVGJIx'},
+        )
+        parsed1 = parse_incoming_event(e1)
+        assert isinstance(parsed1, SplitChangeUpdate)
+        assert parsed1.change_number == 1591996685190
+        assert parsed1.previous_change_number == 12
+        assert parsed1.compression == 2
+        assert parsed1.split_definition == 'eJzEUtFu2kAQ/BU0z4d0hw2Be0MFRVGJIx'
+
         e1 = make_message(
             'NDA5ODc2MTAyNg==_MzAyODY0NDkyOA==_splits',
             {'type':'SPLIT_UPDATE','changeNumber':1591996685190},
         )
         parsed1 = parse_incoming_event(e1)
         assert isinstance(parsed1, SplitChangeUpdate)
         assert parsed1.change_number == 1591996685190
+        assert parsed1.previous_change_number == None
+        assert parsed1.compression == None
+        assert parsed1.split_definition == None
+
 
         e2 = make_message(
             'NDA5ODc2MTAyNg==_MzAyODY0NDkyOA==_segments',