56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
from models.model import Model
|
|
import os
|
|
|
|
def test_model():
|
|
"""Test base Model class functionality."""
|
|
# Create a test model class
|
|
class TestModel(Model):
|
|
pass
|
|
|
|
# Clean up any existing test file (model uses db/<classname>.json)
|
|
db_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "db")
|
|
testmodel_file = os.path.join(db_dir, "testmodel.json")
|
|
if os.path.exists(testmodel_file):
|
|
os.remove(testmodel_file)
|
|
|
|
model = TestModel()
|
|
|
|
print("Testing get_next_id with empty model")
|
|
next_id = model.get_next_id()
|
|
assert next_id == "1"
|
|
print(f"Next ID: {next_id}")
|
|
|
|
print("\nTesting get_next_id with existing entries")
|
|
model["1"] = {"data": "test1"}
|
|
model["2"] = {"data": "test2"}
|
|
model["5"] = {"data": "test5"}
|
|
next_id = model.get_next_id()
|
|
assert next_id == "6"
|
|
print(f"Next ID: {next_id}")
|
|
|
|
print("\nTesting save and load")
|
|
model.save()
|
|
|
|
# Create new instance to test load
|
|
model2 = TestModel()
|
|
assert "1" in model2
|
|
assert "2" in model2
|
|
assert "5" in model2
|
|
assert model2["1"]["data"] == "test1"
|
|
|
|
print("\nTesting set_defaults")
|
|
model2.set_defaults()
|
|
# Note: set_defaults currently doesn't work as expected (self = {} doesn't modify the dict)
|
|
# But we can test that the method exists and doesn't crash
|
|
assert hasattr(model2, 'set_defaults')
|
|
|
|
# Clean up
|
|
if os.path.exists(testmodel_file):
|
|
os.remove(testmodel_file)
|
|
|
|
print("\nAll model base class tests passed!")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
test_model()
|