From 050f60ee7c3747587cc8e43ff9782e5d06cb7d30 Mon Sep 17 00:00:00 2001
From: Thibaud <thibaud.gasser@gmx.com>
Date: Fri, 7 Dec 2018 16:45:35 +0100
Subject: [PATCH] Create unittests

---
 tests/configurator_tests.py | 48 +++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/tests/configurator_tests.py b/tests/configurator_tests.py
index e69de29..fcc9036 100644
--- a/tests/configurator_tests.py
+++ b/tests/configurator_tests.py
@@ -0,0 +1,48 @@
+import os
+import shutil
+import tempfile
+import unittest
+from configurator import configurator
+
+class ConfiguratorTests(unittest.TestCase):
+
+    def setUp(self):
+        self.sim_path = tempfile.mkdtemp()
+        self.sim_name = 'test_simulation'
+
+    def tearDown(self):
+        shutil.rmtree(self.sim_path)
+
+    def test_generate_scenario(self):
+        configurator.generate_scenario(self.sim_path, self.sim_name)
+        self.assert_is_dir(self.sim_path)
+        self.assert_is_dir(os.path.join(self.sim_path, 'log'))
+        generated_files = [
+            f'{self.sim_name}.sumocfg',
+            f'{self.sim_name}.poly.xml',
+            f'{self.sim_name}.net.xml'
+        ]
+        for f in generated_files:
+            self.assert_is_file(os.path.join(self.sim_path, f))
+
+    def test_generate_mobility(self):
+        # The scenario must be generated before the mobility
+        configurator.generate_scenario(self.sim_path, self.sim_name)
+        configurator.generate_mobility(self.sim_path, self.sim_name)
+        trips_file = os.path.join(self.sim_path, f'{self.sim_name}.trips.xml')
+        self.assert_is_file(trips_file)
+
+    def assert_exists(self, path):
+        self.assertTrue(os.path.exists(path), msg=f'{path} does not exist')
+
+    def assert_is_file(self, path):
+        self.assert_exists(path)
+        self.assertTrue(os.path.isfile(path), msg=f'{path} is not a file')
+
+    def assert_is_dir(self, path):
+        self.assert_exists(path)
+        self.assertTrue(os.path.isdir(path), msg=f'{path} is not a directory')
+
+
+if __name__ == '__main__':
+    unittest.main()