1 files changed, 86 insertions, 0 deletions
diff --git a/morphlib/buildcontroller_tests.py b/morphlib/buildcontroller_tests.py
new file mode 100644
index 00000000..40b00213
--- /dev/null
+++ b/morphlib/buildcontroller_tests.py
@@ -0,0 +1,86 @@
+# Copyright (C) 2012  Codethink Limited
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; version 2 of the License.
+# 
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+
+import unittest
+
+import morphlib
+
+
+class DummyApp(object):
+
+    def __init__(self):
+        self.settings = {}
+        self.msg = lambda x: '%s' % x
+
+
+class DummyWorker(object):
+
+    def __init__(self, name, ident):
+        self.name = name
+        self.ident = ident
+
+
+class BuildControllerTests(unittest.TestCase):
+
+    def test_construction_with_app_and_tempdir(self):
+        app = DummyApp()
+        tempdir = '/foo/bar'
+        controller = morphlib.buildcontroller.BuildController(app, tempdir)
+        self.assertEqual(app.settings, controller.settings)
+        self.assertEqual(tempdir, controller.tempdir)
+
+    def test_adding_workers(self):
+        app = DummyApp()
+        tempdir = '/foo/bar'
+        controller = morphlib.buildcontroller.BuildController(app, tempdir)
+
+        worker1 = object()
+        worker2 = object()
+        worker3 = object()
+
+        controller.add_worker(worker1)
+        self.assertTrue(worker1 in controller.workers)
+        self.assertTrue(worker2 not in controller.workers)
+        self.assertTrue(worker3 not in controller.workers)
+
+        controller.add_worker(worker2)
+        self.assertTrue(worker1 in controller.workers)
+        self.assertTrue(worker2 in controller.workers)
+        self.assertTrue(worker3 not in controller.workers)
+
+        controller.add_worker(worker3)
+        self.assertTrue(worker1 in controller.workers)
+        self.assertTrue(worker2 in controller.workers)
+        self.assertTrue(worker3 in controller.workers)
+
+    def test_generation_of_worker_names(self):
+        app = DummyApp()
+        tempdir = '/foo/bar'
+        controller = morphlib.buildcontroller.BuildController(app, tempdir)
+        
+        localname1 = controller.generate_worker_name('local')
+        worker1 = DummyWorker(localname1, 'local')
+        controller.add_worker(worker1)
+        localname2 = controller.generate_worker_name('local')
+        worker2 = DummyWorker(localname1, 'local')
+        controller.add_worker(worker2)
+        localname3 = controller.generate_worker_name('local')
+        worker3 = DummyWorker(localname1, 'local')
+        controller.add_worker(worker3)
+
+        self.assertEqual(localname1, 'local-1')
+        self.assertEqual(localname2, 'local-2')
+        self.assertEqual(localname3, 'local-3')