Initial autokey functionality

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()