@@ -21,7 +21,7 @@ def test_get_split(self, mocker):
2121 adapter = mocker .Mock (spec = RedisAdapter )
2222 adapter .get .return_value = '{"name": "some_split"}'
2323 from_raw = mocker .Mock ()
24- mocker .patch ('splitio.models .splits.from_raw' , new = from_raw )
24+ mocker .patch ('splitio.storage.redis .splits.from_raw' , new = from_raw )
2525
2626 storage = RedisSplitStorage (adapter )
2727 storage .get ('some_split' )
@@ -43,7 +43,7 @@ def test_get_split_with_cache(self, mocker):
4343 adapter = mocker .Mock (spec = RedisAdapter )
4444 adapter .get .return_value = '{"name": "some_split"}'
4545 from_raw = mocker .Mock ()
46- mocker .patch ('splitio.models .splits.from_raw' , new = from_raw )
46+ mocker .patch ('splitio.storage.redis .splits.from_raw' , new = from_raw )
4747
4848 storage = RedisSplitStorage (adapter , True , 1 )
4949 storage .get ('some_split' )
@@ -62,6 +62,7 @@ def test_get_split_with_cache(self, mocker):
6262 from_raw .reset_mock ()
6363 adapter .get .return_value = None
6464
65+ # Still cached
6566 result = storage .get ('some_split' )
6667 assert result is not None
6768 time .sleep (1 ) # wait for expiration
@@ -75,7 +76,7 @@ def test_get_splits_with_cache(self, mocker):
7576 adapter = mocker .Mock (spec = RedisAdapter )
7677 storage = RedisSplitStorage (adapter )
7778 from_raw = mocker .Mock ()
78- mocker .patch ('splitio.models .splits.from_raw' , new = from_raw )
79+ mocker .patch ('splitio.storage.redis .splits.from_raw' , new = from_raw )
7980
8081 adapter .mget .return_value = ['{"name": "split1"}' , '{"name": "split2"}' , None ]
8182
@@ -102,7 +103,7 @@ def test_get_all_splits(self, mocker):
102103 adapter = mocker .Mock (spec = RedisAdapter )
103104 storage = RedisSplitStorage (adapter )
104105 from_raw = mocker .Mock ()
105- mocker .patch ('splitio.models .splits.from_raw' , new = from_raw )
106+ mocker .patch ('splitio.storage.redis .splits.from_raw' , new = from_raw )
106107
107108 adapter .keys .return_value = [
108109 'SPLITIO.split.split1' ,
@@ -177,7 +178,7 @@ def test_fetch_segment(self, mocker):
177178 adapter .smembers .return_value = set (["key1" , "key2" , "key3" ])
178179 adapter .get .return_value = '100'
179180 from_raw = mocker .Mock ()
180- mocker .patch ('splitio.models .segments.from_raw' , new = from_raw )
181+ mocker .patch ('splitio.storage.redis .segments.from_raw' , new = from_raw )
181182
182183 storage = RedisSegmentStorage (adapter )
183184 result = storage .get ('some_segment' )
0 commit comments