Still refactoring and unit tests

This commit is contained in:
2025-08-30 22:37:17 +00:00
parent 1d423c5ea2
commit c76a77cb0c
16 changed files with 78 additions and 73 deletions

View File

@@ -1,7 +1,7 @@
import unittest
from models import Gate, Status
from repository import GatesRepository
from src.models import Gate, Status
from src.repository import GatesRepository
class TestGatesRepository(unittest.TestCase):
@@ -15,7 +15,7 @@ class TestGatesRepository(unittest.TestCase):
result = self.repo.get_by_key(gate_key)
self.assertEqual(expected_gate.id, result.id)
self.assertEqual(expected_gate.gid, result.gid)
self.assertEqual(expected_gate.name, result.name)
self.assertEqual(expected_gate.status, result.status)
@@ -33,7 +33,7 @@ class TestGatesRepository(unittest.TestCase):
self.assertEqual(len(enabled_gate_ids), len(result))
self.assertEqual(
sorted(enabled_gate_ids), sorted([gate.id for gate in result])
sorted(enabled_gate_ids), sorted([gate.gid for gate in result])
)
for gate in result: