Implement delete() and exists()

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/test_stash.py b/tests/test_stash.py
index 88cc466..0e84eae 100644
--- a/tests/test_stash.py
+++ b/tests/test_stash.py
@@ -28,11 +28,26 @@ def test_invalid_path():
 
 def test_stash():
     stash = YmlStash(User, TEST_STASH_PATH)
+
     yuval = User(name="yuval", age=42)
     stash.save(yuval, "foo")
+    assert stash.exists("foo")
+    assert not stash.exists("goo")
     assert stash.list_keys() == ["foo"]
+
     obj = stash.load("foo")
     assert obj == yuval
+
+    goo = User(name="goo", age=10)
+    stash.save(goo, "goo")
+    assert stash.list_keys() == ["foo", "goo"]
+
+    stash.delete("foo")
+    assert stash.list_keys() == ["goo"]
+
+    with pytest.raises(Exception):
+        stash.delete("foo")
+
     stash.drop()
     assert stash.list_keys() == []