Add support for filtering null values

1 files changed, 20 insertions, 1 deletions

diff --git a/tests/test_stash.py b/tests/test_stash.py
index d6f8ab9..3ab86a3 100644
--- a/tests/test_stash.py
+++ b/tests/test_stash.py
@@ -3,7 +3,7 @@ import pytest
 from pathlib import Path
 from ymlstash import YmlStash
 from dataclasses import dataclass
-from typing import ClassVar
+from typing import ClassVar, Optional
 
 TEST_STASH_PATH = Path("/tmp")
 
@@ -101,3 +101,22 @@ def test_dump_order():
         assert keys == "ordeaz"
 
     stash.drop()
+
+
+def test_null_values():
+    @dataclass
+    class Nullable:
+        name: str
+        nullval: Optional[str] = None
+        key: ClassVar[str] = "name"
+
+    stash = YmlStash(Nullable, TEST_STASH_PATH, filter_none=True)
+    goo = Nullable("goo")
+    stash.save(goo)
+
+    with open(TEST_STASH_PATH / "goo.yml", "r") as f:
+        lines = f.readlines()
+        for line in lines:
+            assert "nullval" not in line
+
+    stash.drop()