diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_stash.py | 21 |
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() |
