From 39fead94f97d2f2ed63f0b057b9e0373bf42519a Mon Sep 17 00:00:00 2001
From: Marco Zeisler <marco.zeisler@kapsch.net>
Date: Thu, 17 Jun 2021 23:50:54 +0200
Subject: [PATCH] update tests

---
 components/orchestration/test_orchestrator.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/components/orchestration/test_orchestrator.py b/components/orchestration/test_orchestrator.py
index 85a9807..7f90374 100644
--- a/components/orchestration/test_orchestrator.py
+++ b/components/orchestration/test_orchestrator.py
@@ -3,6 +3,7 @@ import unittest
 from unittest.mock import patch
 
 from dse_shared_libs.mock.response import MyResponse
+from dse_shared_libs.traffic_light_color import TrafficLightColor
 
 from orchestrator import Orchestrator
 
@@ -13,22 +14,31 @@ class TestOrchestrator(unittest.TestCase):
         with patch('requests.sessions.Session.get', MyResponse):
             self.orc = Orchestrator()
 
+    def test_full_speed_if_nothing_in_range(self):
+        self.orc.tls = {'1': {'color': TrafficLightColor.RED, 'switching_time': 100, 'last_switch': self.timestamp}}
+        tl_geo = {'cursor': []}
+        current_vel = 55.0
+
+        target_vel = self.orc._compute_velocity(tl_geo, current_vel)
+        self.assertEqual(130, target_vel)
+
     def test_full_speed_if_far_away(self):
-        self.orc.tls = {'1': {'color': 'RED', 'switching_time': 1, 'last_switch': self.timestamp}}
+        self.orc.tls = {'1': {'color': TrafficLightColor.GREEN, 'switching_time': 1000, 'last_switch': self.timestamp}}
         tl_geo = {'cursor': [
             {'id': '1', 'calculatedRange': 1000}
         ]}
         current_vel = 55.0
 
         target_vel = self.orc._compute_velocity(tl_geo, current_vel)
-        self.assertEqual(current_vel, target_vel)
+        self.assertEqual(130, target_vel)
 
     def test_slow_down_if_passing_not_possible(self):
-        self.orc.tls = {'1': {'color': 'RED', 'switching_time': 100000, 'last_switch': self.timestamp}}
+        self.orc.tls = {'1': {'color': TrafficLightColor.RED, 'switching_time': 100000, 'last_switch': self.timestamp}}
         tl_geo = {'cursor': [
-            {'id': '1', 'calculatedRange': 1000}
+            {'id': '1', 'calculatedRange': 1}
         ]}
         current_vel = 130.0
 
         target_vel = self.orc._compute_velocity(tl_geo, current_vel)
         self.assertNotEqual(current_vel, target_vel)
+        self.assertEqual(0, target_vel)