@@ -115,13 +115,16 @@ def test_redis_client_creation(self, mocker):
115115 'redisSslCertReqs' : 'some_cert_req' ,
116116 'redisSslCaCerts' : 'some_ca_cert' ,
117117 'redisMaxConnections' : 999 ,
118+ 'flagSetsFilter' : ['set_1' ]
118119 }
119120 factory = get_factory ('some_api_key' , config = config )
120121 assert isinstance (factory ._get_storage ('splits' ), redis .RedisSplitStorage )
121122 assert isinstance (factory ._get_storage ('segments' ), redis .RedisSegmentStorage )
122123 assert isinstance (factory ._get_storage ('impressions' ), redis .RedisImpressionsStorage )
123124 assert isinstance (factory ._get_storage ('events' ), redis .RedisEventsStorage )
124125
126+ assert factory ._get_storage ('splits' ).flag_set_filter .flag_sets == set ([])
127+
125128 adapter = factory ._get_storage ('splits' )._redis
126129 assert adapter == factory ._get_storage ('segments' )._redis
127130 assert adapter == factory ._get_storage ('impressions' )._redis
@@ -569,13 +572,15 @@ def test_pluggable_client_creation(self, mocker):
569572 'labelsEnabled' : False ,
570573 'impressionListener' : 123 ,
571574 'storageType' : 'pluggable' ,
572- 'storageWrapper' : StorageMockAdapter ()
575+ 'storageWrapper' : StorageMockAdapter (),
576+ 'flagSetsFilter' : ['set_1' ]
573577 }
574578 factory = get_factory ('some_api_key' , config = config )
575579 assert isinstance (factory ._get_storage ('splits' ), pluggable .PluggableSplitStorage )
576580 assert isinstance (factory ._get_storage ('segments' ), pluggable .PluggableSegmentStorage )
577581 assert isinstance (factory ._get_storage ('impressions' ), pluggable .PluggableImpressionsStorage )
578582 assert isinstance (factory ._get_storage ('events' ), pluggable .PluggableEventsStorage )
583+ assert factory ._get_storage ('splits' ).flag_set_filter .flag_sets == set ([])
579584
580585 adapter = factory ._get_storage ('splits' )._pluggable_adapter
581586 assert adapter == factory ._get_storage ('segments' )._pluggable_adapter
0 commit comments