diff --git a/splitio/storage/inmemmory.py b/splitio/storage/inmemmory.py index 00dbb16b..3967bdf8 100644 --- a/splitio/storage/inmemmory.py +++ b/splitio/storage/inmemmory.py @@ -499,6 +499,10 @@ def record_ready_time(self, ready_time): """Record ready time.""" self._tel_config.record_ready_time(ready_time) + def record_flag_sets(self, flag_sets): + """Record flag sets.""" + self._tel_config.record_flag_sets(flag_sets) + def add_tag(self, tag): """Record tag string.""" with self._lock: @@ -567,6 +571,10 @@ def record_update_from_sse(self, event): """Record update from sse.""" self._counters.record_update_from_sse(event) + def get_flag_sets(self): + """Get flag sets.""" + self._tel_config.get_flag_sets() + def get_bur_time_outs(self): """Get block until ready timeout.""" return self._tel_config.get_bur_time_outs() diff --git a/tests/storage/test_inmemory_storage.py b/tests/storage/test_inmemory_storage.py index 7319548d..dea130fd 100644 --- a/tests/storage/test_inmemory_storage.py +++ b/tests/storage/test_inmemory_storage.py @@ -467,7 +467,8 @@ def test_resets(self): 'iL': False, 'hp': None, 'aF': 0, - 'rF': 0 + 'rF': 0, + 'fS': 0 }) assert(storage._streaming_events.pop_streaming_events() == {'streamingEvents': []}) assert(storage._tags == []) @@ -492,6 +493,7 @@ def test_record_config(self): } storage.record_config(config, {}) storage.record_active_and_redundant_factories(1, 0) + storage.record_flag_sets(2) assert(storage._tel_config.get_stats() == {'oM': 0, 'sT': storage._tel_config._get_storage_type(config['operationMode'], config['storageType']), 'sE': config['streamingEnabled'], @@ -506,7 +508,8 @@ def test_record_config(self): 'tR': 0, 'nR': 0, 'aF': 1, - 'rF': 0} + 'rF': 0, + 'fS': 2} ) def test_record_counters(self):