From 81458cda4950628ac11af77349f788e97c5dabe8 Mon Sep 17 00:00:00 2001
From: Marco Zeisler <marco.zeisler@kapsch.net>
Date: Thu, 17 Jun 2021 22:44:11 +0200
Subject: [PATCH] setup orchestrator test framework;

---
 components/orchestration/test_orchestrator.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

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)