summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_stash.py15
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()