fiableDB/test/test_save_and_load.py
2022-11-28 16:55:39 +01:00

42 lines
1.0 KiB
Python

import os
from fiable_db import start, add, save, get_database, load
filename = "fiabledb.json"
def delete_file():
"""Delete the database file."""
if os.path.exists(filename):
os.remove(filename)
def test_empty():
"""Test that save() works when the database is empty."""
delete_file()
start()
save()
load()
assert get_database() == []
def test_one():
"""Test that save() works when the database has one entry."""
delete_file()
start()
add({"name": "John", "age": 30})
save()
load()
assert get_database() == [{"id": 1, "rev": 1, "data": {"name": "John", "age": 30}}]
def test_two():
"""Test that save() works when the database has two entries."""
delete_file()
start()
add({"name": "John", "age": 30})
add({"name": "Jane", "age": 28})
save()
load()
assert get_database() == [
{"id": 1, "rev": 1, "data": {"name": "John", "age": 30}},
{"id": 2, "rev": 1, "data": {"name": "Jane", "age": 28}},
]