Test for invalid paths

1 files changed, 13 insertions, 6 deletions

diff --git a/tests/test_stash.py b/tests/test_stash.py
index 6b9e47f..88cc466 100644
--- a/tests/test_stash.py
+++ b/tests/test_stash.py
@@ -5,6 +5,8 @@ from ymlstash import YmlStash
 from dataclasses import dataclass
 from typing import ClassVar
 
+TEST_STASH_PATH = "/tmp"
+
 
 @dataclass
 class User:
@@ -13,14 +15,19 @@ class User:
 
 
 def test_stash_path():
-    stash = YmlStash(User, "foo")
-    assert stash.path == Path("foo")
-    stash = YmlStash(User, Path("foo"))
-    assert stash.path == Path("foo")
+    stash = YmlStash(User, ".")
+    assert stash.path == Path(".")
+    stash = YmlStash(User, Path("."))
+    assert stash.path == Path(".")
+
+
+def test_invalid_path():
+    with pytest.raises(Exception):
+        YmlStash(User, "/tmp/does/not/exist")
 
 
 def test_stash():
-    stash = YmlStash(User, "/tmp/")
+    stash = YmlStash(User, TEST_STASH_PATH)
     yuval = User(name="yuval", age=42)
     stash.save(yuval, "foo")
     assert stash.list_keys() == ["foo"]
@@ -43,7 +50,7 @@ def test_key_field():
         name: str
         key: ClassVar[str] = "name"
 
-    stash = YmlStash(Dog, "/tmp/")
+    stash = YmlStash(Dog, TEST_STASH_PATH)
     terra = Dog(name="terra")
     stash.save(terra)
     assert stash.list_keys() == ["terra"]