polishing

Author: chillaq

splitio/storage/pluggable.py

@@ -448,7 +448,7 @@ def get(self, segment_name):
         :rtype: splitio.models.segments.Segment
         """
         try:
-            return segments.from_raw({'name': segment_name, 'added': list(self._pluggable_adapter.get(self._prefix.format(segment_name=segment_name))), 'removed': [], 'till': self._pluggable_adapter.get(self._segment_till_prefix.format(segment_name=segment_name))})
+            return segments.from_raw({'name': segment_name, 'added': self._pluggable_adapter.get_items(self._prefix.format(segment_name=segment_name)), 'removed': [], 'till': self._pluggable_adapter.get(self._segment_till_prefix.format(segment_name=segment_name))})
         except Exception:
             _LOGGER.error('Error getting segment')
             _LOGGER.debug('Error: ', exc_info=True)

tests/storage/test_pluggable.py

@@ -16,6 +16,11 @@ def get(self, key):
             return None
         return self._keys[key]
 
+    def get_items(self, key):
+        if key not in self._keys:
+            return None
+        return list(self._keys[key])
+
     def get_many(self, keys):
         return [self.get(key) for key in keys]