diff --git a/components/orchestration/test_orchestrator.py b/components/orchestration/test_orchestrator.py index 4cb7a2e..928a46a 100644 --- a/components/orchestration/test_orchestrator.py +++ b/components/orchestration/test_orchestrator.py @@ -1,3 +1,4 @@ +import datetime import unittest from unittest.mock import patch @@ -8,8 +9,16 @@ from orchestrator import Orchestrator class TestOrchestrator(unittest.TestCase): def setUp(self) -> None: + self.timestamp = datetime.datetime.now() with patch('requests.sessions.Session.get', MyResponse): self.orc = Orchestrator() def test(self): - ... \ No newline at end of file + self.orc.tls = {'1': {'color': 'RED', 'switching_time': 1, 'last_switch': self.timestamp}} + tl_geo = {'cursor': [ + {'id': '1', 'calculatedRange': 100} + ]} + current_vel = 130.0 + + target_vel = self.orc._compute_velocity(tl_geo, current_vel) + self.assertEqual(current_vel, target_vel)