Skip to content

Commit 595ab35

Browse files
committed
Flush lists in BaseListsStorage's tearDown method
1 parent 9003491 commit 595ab35

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

stream_framework/tests/storage/base_lists_storage.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
def implementation(meth):
77
def wrapped_test(self, *args, **kwargs):
8-
if self.lists_storage_class in (BaseListsStorage,):
8+
if self.lists_storage_class == BaseListsStorage:
99
raise unittest.SkipTest('only test this on actual implementations')
1010
return meth(self, *args, **kwargs)
1111
return wrapped_test
@@ -21,6 +21,10 @@ def setUp(self):
2121
self.lists_storage = self.lists_storage_class(key=self.key,
2222
max_length = self.max_length)
2323

24+
def tearDown(self):
25+
if self.lists_storage_class != BaseListsStorage:
26+
self.lists_storage.flush('whenever', 'everyday', 'whatever')
27+
2428
@implementation
2529
def test_add_empty_values(self):
2630
self.lists_storage.add(whenever=[])

stream_framework/tests/storage/redis/lists_storage.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,3 @@ class TestStorage(RedisListsStorage):
1010
class TestRedisListsStorage(TestBaseListsStorage):
1111

1212
lists_storage_class = TestStorage
13-
14-
def tearDown(self):
15-
self.lists_storage.redis.flushall()

0 commit comments

Comments
 (0)