Test fixes

tests/fakes.py

@@ -15,8 +15,11 @@
 # under the License.
 
 import gear
+import json
 import threading
 import os
+import re
+import time
 
 from turbo_hipster.worker_manager import GearmanManager
 from turbo_hipster.task_plugins.gate_real_db_upgrade.task import Runner\
@@ -31,7 +34,7 @@ class FakeGearmanManager(GearmanManager):
     def setup_gearman(self):
         hostname = os.uname()[1]
         self.gearman_worker = FakeWorker('turbo-hipster-manager-%s'
-                                          % hostname, self.test)
+                                         % hostname, self.test)
         self.gearman_worker.addServer(
             self.config['zuul_server']['gearman_host'],
             self.config['zuul_server']['gearman_port']
@@ -166,6 +169,15 @@ class FakeRealDbUpgradeRunner(RealDbUpgradeRunner):
         self.register_functions()
 
 
+class BuildHistory(object):
+    def __init__(self, **kw):
+        self.__dict__.update(kw)
+
+    def __repr__(self):
+        return ("<Completed build, result: %s name: %s #%s changes: %s>" %
+                (self.result, self.name, self.number, self.changes))
+
+
 class FakeBuild(threading.Thread):
     def __init__(self, worker, job, number, node):
         threading.Thread.__init__(self)
@@ -295,4 +307,3 @@ class FakeGearmanServer(gear.Server):
         qlen = (len(self.high_queue) + len(self.normal_queue) +
                 len(self.low_queue))
         self.log.debug("done releasing queued jobs %s (%s)" % (regex, qlen))
-

tests/test_worker_manager.py

@@ -19,7 +19,8 @@ import json
 import os
 import testtools
 import time
-from fakes import FakeGearmanManager, FakeGearmanServer, FakeRealDbUpgradeRunner
+from fakes import FakeGearmanManager, FakeGearmanServer,\
+    FakeRealDbUpgradeRunner
 
 CONFIG_DIR = os.path.join(os.path.dirname(__file__), 'etc')
 with open(os.path.join(CONFIG_DIR, 'config.json'), 'r') as config_stream:
@@ -65,6 +66,3 @@ class TestGearmanManager(testtools.TestCase):
         """ Check that the manager successfully stops a task when requested
         """
         pass
-
-if __name__ == '__main__':
-    unittest.main()