mirror of
https://github.com/senju1337/senju.git
synced 2025-12-24 07:39:29 +00:00
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
# do not remove this import. This is needed for
|
|
# pytest fixtures to work
|
|
from __future__ import annotations
|
|
|
|
from _pytest.mark.structures import store_mark
|
|
import pytest # noqa: F401
|
|
|
|
from senju.haiku import DEFAULT_HAIKU, Haiku
|
|
from senju.store_manager import StoreManager # noqa: F401
|
|
|
|
|
|
def test_save_and_load_any(store_manager: StoreManager):
|
|
thing = {
|
|
"color": "blue",
|
|
"number": 19,
|
|
"inner": {
|
|
"no": "yes"
|
|
}
|
|
}
|
|
thing_id = store_manager._save(thing)
|
|
thing_loaded = store_manager._load(thing_id)
|
|
|
|
if thing_loaded is None:
|
|
assert False, "store manager load did not return anything but \
|
|
should have"
|
|
for key in thing.keys():
|
|
assert thing[key] == thing_loaded[key]
|
|
|
|
|
|
def test_save_and_load_haiku(store_manager: StoreManager):
|
|
h = Haiku(lines=["foobar", "qux"])
|
|
hid = store_manager.save_haiku(h)
|
|
h_loaded = store_manager.load_haiku(hid)
|
|
|
|
if h_loaded is None:
|
|
assert False, "store manager load_haiku did not return anything \
|
|
but should have"
|
|
|
|
assert h == h_loaded
|
|
|
|
|
|
def test_load_latest_with_empty_store(temp_data_dir):
|
|
store = StoreManager(temp_data_dir / "empty_store.json")
|
|
h = store.get_id_of_latest_haiku()
|
|
assert h is None
|
|
|
|
|
|
def test_load_latest_or_default_with_empty(temp_data_dir):
|
|
store = StoreManager(temp_data_dir / "load_or_default_empty.json")
|
|
haiku = store.get_latest_haiku_or_default()
|
|
assert haiku == DEFAULT_HAIKU
|
|
|
|
|
|
def test_load_latest_or_default_with_non_empty(temp_data_dir):
|
|
store = StoreManager(temp_data_dir / "load_or_default_not_empty.json")
|
|
nonsense_test_haiku = Haiku(["nonsense", "test", "haiku"])
|
|
store.save_haiku(nonsense_test_haiku)
|
|
haiku = store.get_latest_haiku_or_default()
|
|
assert haiku != DEFAULT_HAIKU
|
|
assert haiku == nonsense_test_haiku
|