diff --git a/.coverage b/.coverage deleted file mode 100644 index 9b42c3f..0000000 Binary files a/.coverage and /dev/null differ diff --git a/.gitignore b/.gitignore index 54ef793..618391f 100644 --- a/.gitignore +++ b/.gitignore @@ -2,3 +2,5 @@ *.pyc local_settings.py +.coverage +worker.log diff --git a/run_tests.sh b/run_tests.sh index 54904d1..d5e661f 100755 --- a/run_tests.sh +++ b/run_tests.sh @@ -1,2 +1,2 @@ #!/bin/bash -nosetests tests --exclude-dir=stacktach --with-coverage --cover-package=stacktach +nosetests tests --exclude-dir=stacktach --with-coverage --cover-package=stacktach,worker diff --git a/tests/unit/test_worker.py b/tests/unit/test_worker.py new file mode 100644 index 0000000..cfb534b --- /dev/null +++ b/tests/unit/test_worker.py @@ -0,0 +1,36 @@ +import unittest + +import kombu +import kombu.entity +import mox + +import worker.worker as worker + +class NovaConsumerTestCase(unittest.TestCase): + def setUp(self): + self.mox = mox.Mox() + + def test_get_consumer(self): + created_queues = None + created_callback = None + created_consumers = [] + def Consumer(queues=None, callbacks=None): + created_queues = queues + created_callback = callbacks + consumer = self.mox.CreateMockAnything() + created_consumers.append(consumer) + return consumer + self.mox.StubOutClassWithMocks(kombu.entity, 'Exchange') + self.mox.StubOutClassWithMocks(kombu, 'Queue') + kombu.entity.Exchange('nova', type='topic', exclusive=False, + durable=True, auto_delete=False) + kombu.Queue('monitor.info', kombu.entity.Exchange, auto_delete=False, + durable=True, exclusive=False, routing_key='monitor.info') + kombu.Queue('monitor.error', kombu.entity.Exchange, auto_delete=False, + durable=True, exclusive=False, routing_key='monitor.error') + consumer = worker.NovaConsumer('test', None, None, True) + self.mox.ReplayAll() + consumers = consumer.get_consumers(Consumer, None) + self.assertEqual(len(consumers), len(created_consumers)) + self.assertEqual(consumers[0], created_consumers[0]) + self.mox.VerifyAll() \ No newline at end of file diff --git a/worker/__init__.py b/worker/__init__.py new file mode 100644 index 0000000..e69de29