More tests for worker manager.

aimmo-game-creator/setup.py

@@ -12,6 +12,7 @@
     ],
     tests_require=[
         'httmock',
+        'mock',
     ],
     test_suite='tests',
     zip_safe=False,

aimmo-game-creator/tests/test_worker_manager.py

@@ -1,11 +1,16 @@
 from __future__ import absolute_import
 
+import cPickle as pickle
 import unittest
+
 from json import dumps, loads
 
 from httmock import HTTMock
+import mock
 
 from worker_manager import WorkerManager
+from worker_manager import LocalWorkerManager
+
 
 class ConcreteWorkerManager(WorkerManager):
     def __init__(self, *args, **kwargs):
@@ -93,3 +98,44 @@ def test_added_workers_given_correct_url(self):
                 'http://test/{}/'.format(i)
             )
             self.assertEqual(self.worker_manager.added_workers[str(i)]['name'], 'Game %s' % i)
+
+class TestLocalWorkerManager(unittest.TestCase):
+
+    def test_create_worker(self):
+        with mock.patch('subprocess.Popen') as mocked_popen:
+            localWorkerManager = LocalWorkerManager("")
+
+            game_id = 1
+            game_data = {
+                "test" : "test"
+            }
+
+            localWorkerManager.create_worker(game_id, game_data)
+            call_args = mocked_popen.call_args
+
+            argument_dictionary = call_args[1]
+            self.assertTrue("aimmo-game" in argument_dictionary["cwd"])
+            self.assertEqual(argument_dictionary["env"]["test"], "test")
+
+    def test_remove_worker(self):
+        self.killed = False
+        class KillableWorker():
+            def __init__(self, binder):
+                self.binder = binder
+                self.binder.killed = False
+
+            def kill(self):
+                self.binder.killed = True
+
+        localWorkerManger = LocalWorkerManager("")
+        localWorkerManger.workers = {
+            1 : KillableWorker(self)
+        }
+
+        self.assertFalse(self.killed)
+        self.assertTrue(1 in localWorkerManger.workers)
+
+        localWorkerManger.remove_worker(1)
+
+        self.assertTrue(self.killed)
+        self.assertTrue(1 not in localWorkerManger.workers)

aimmo-game-worker/tests/simulation/test_location.py

@@ -24,8 +24,7 @@ def test_repr(self):
         self.assertTrue("Location(1, 2)" == str(dummy_location))
 
 
-
-
-
-
-
+    def test_not_equal(self):
+        dummy_location_1 = Location(1, 1)
+        dummy_location_2 = Location(2, 2)
+        self.assertTrue(dummy_location_1 != dummy_location_2)