48 lines
1.1 KiB
Python
48 lines
1.1 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_save_empty():
|
|
"""Test that save() works when the database is empty."""
|
|
delete_file()
|
|
start()
|
|
save()
|
|
global database
|
|
database = []
|
|
load()
|
|
assert get_database() == []
|
|
|
|
|
|
def test_save_one():
|
|
"""Test that save() works when the database has one entry."""
|
|
delete_file()
|
|
start()
|
|
add({"name": "John", "age": 30})
|
|
save()
|
|
global database
|
|
database = []
|
|
load()
|
|
assert get_database() == [{"id": 1, "rev": 1, "data": {"name": "John", "age": 30}}]
|
|
|
|
|
|
def test_save_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()
|
|
global database
|
|
database = []
|
|
load()
|
|
assert get_database() == [
|
|
{"id": 1, "rev": 1, "data": {"name": "John", "age": 30}},
|
|
{"id": 2, "rev": 1, "data": {"name": "Jane", "age": 28}},
|
|
]
|