diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_stash.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/test_stash.py b/tests/test_stash.py index defdd7e..e7e573b 100644 --- a/tests/test_stash.py +++ b/tests/test_stash.py @@ -1,6 +1,7 @@ from pathlib import Path from ymlstash import YmlStash from dataclasses import dataclass +from typing import ClassVar @dataclass @@ -25,3 +26,17 @@ def test_stash(): assert obj == yuval stash.drop() assert stash.list_all_keys() == [] + + +@dataclass +class Dog: + name: str + key: ClassVar[str] = "name" + + +def test_auto_key(): + stash = YmlStash(Dog, "/tmp/") + terra = Dog(name="terra") + stash.save(None, terra) + assert stash.list_all_keys() == ["terra"] + stash.drop() |
