Skip to content

Commit 8493ce0

Browse files
committed
fixed manager tests
1 parent 8d50b81 commit 8493ce0

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/client/test_manager.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,7 @@ def test_manager_calls(self, mocker):
2929
manager = SplitManager(factory)
3030
split1 = splits.from_raw(splits_json["splitChange1_1"]["splits"][0])
3131
split2 = splits.from_raw(splits_json["splitChange1_3"]["splits"][0])
32-
storage.put(split1)
33-
storage.put(split2)
32+
storage.update([split1, split2], [], -1)
3433
manager._storage = storage
3534

3635
assert manager.split_names() == ['SPLIT_2', 'SPLIT_1']
@@ -102,8 +101,7 @@ async def test_manager_calls(self, mocker):
102101
manager = SplitManagerAsync(factory)
103102
split1 = splits.from_raw(splits_json["splitChange1_1"]["splits"][0])
104103
split2 = splits.from_raw(splits_json["splitChange1_3"]["splits"][0])
105-
await storage.put(split1)
106-
await storage.put(split2)
104+
await storage.update([split1, split2], [], -1)
107105
manager._storage = storage
108106

109107
assert await manager.split_names() == ['SPLIT_2', 'SPLIT_1']

0 commit comments

Comments
 (0)