34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
|
import os
|
|
import unittest
|
|
from pathlib import Path
|
|
import project_manager
|
|
from models import Track, Ticket
|
|
|
|
class TestMMAPersistence(unittest.TestCase):
|
|
def test_default_project_has_mma(self):
|
|
proj = project_manager.default_project("test")
|
|
self.assertIn("mma", proj)
|
|
self.assertEqual(proj["mma"], {"epic": "", "tracks": []})
|
|
|
|
def test_save_load_mma(self):
|
|
proj = project_manager.default_project("test")
|
|
proj["mma"] = {"epic": "Test Epic", "tracks": [{"id": "track_1"}]}
|
|
|
|
test_file = Path("test_mma_proj.toml")
|
|
try:
|
|
project_manager.save_project(proj, test_file)
|
|
loaded = project_manager.load_project(test_file)
|
|
self.assertIn("mma", loaded)
|
|
self.assertEqual(loaded["mma"]["epic"], "Test Epic")
|
|
self.assertEqual(len(loaded["mma"]["tracks"]), 1)
|
|
finally:
|
|
if test_file.exists():
|
|
test_file.unlink()
|
|
hist_file = Path("test_mma_proj_history.toml")
|
|
if hist_file.exists():
|
|
hist_file.unlink()
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|