|
3 | 3 | import asyncio |
4 | 4 | from typing import Any |
5 | 5 |
|
6 | | -from deebot_client.util import cancel, create_task |
| 6 | +from deebot_client.util import OnChangedDict, cancel, create_task |
7 | 7 |
|
8 | 8 |
|
9 | 9 | async def test_create_task_and_cancel() -> None: |
@@ -35,3 +35,39 @@ async def sleep(delay: float) -> None: |
35 | 35 | for task in _tasks: |
36 | 36 | assert task.cancelled() |
37 | 37 | assert task.done() |
| 38 | + |
| 39 | + |
| 40 | +def test_on_changed_dict() -> None: |
| 41 | + callback_count = 0 |
| 42 | + |
| 43 | + def increase_counter() -> None: |
| 44 | + nonlocal callback_count |
| 45 | + callback_count += 1 |
| 46 | + |
| 47 | + sut: OnChangedDict[str, int] = OnChangedDict(increase_counter) |
| 48 | + |
| 49 | + sut["test"] = 1001 |
| 50 | + assert callback_count == 1 |
| 51 | + assert sut["test"] == 1001 |
| 52 | + |
| 53 | + sut.update({"test": 1002, "test2": 2001, "test3": 3001, "test4": 4001}) |
| 54 | + assert sut["test"] == 1002 |
| 55 | + assert callback_count == 2 |
| 56 | + |
| 57 | + del sut["test"] |
| 58 | + assert "test" not in sut |
| 59 | + assert callback_count == 3 |
| 60 | + |
| 61 | + assert sut.pop("test2") == 2001 |
| 62 | + assert "test2" not in sut |
| 63 | + assert callback_count == 4 |
| 64 | + |
| 65 | + (popped_key, popped_value) = sut.popitem() |
| 66 | + assert popped_key == "test4" |
| 67 | + assert popped_value == 4001 |
| 68 | + assert "test4" not in sut |
| 69 | + assert callback_count == 5 |
| 70 | + |
| 71 | + sut.clear() |
| 72 | + assert sut == {} |
| 73 | + assert callback_count == 6 |
0 commit comments