Merge lp:~psivaa/adt-cloud-worker/unittest-process into lp:adt-cloud-worker

[Para Siva]

Unittest to test adt-cloud-worker.process.

Unittest to test adt-cloud-worker.process.

Unmerged revisions

=== added directory 'tests'
=== added file 'tests/test_cloud_worker.py'
--- tests/test_cloud_worker.py	1970-01-01 00:00:00 +0000
+++ tests/test_cloud_worker.py	2015-03-03 19:41:37 +0000
@@ -0,0 +1,36 @@
+import unittest
+import os
+import imp
+import kombu
+import mock
+from mock import MagicMock
+
+script = imp.load_source(os.path.realpath(__file__), '../adt-cloud-worker.py')
+
+
+class message(object):
+    def ack():
+        return 0
+
+
+class TestCloudWorker(unittest.TestCase):
+    def setUp(self):
+        super(TestCloudWorker, self).setUp()
+        self.name = 'TestWorker'
+        self.connection = kombu.Connection
+        self.body = {
+            'request_id': 'fake_uuid',
+            'arch': 'amd64',
+            'platform': 'nova',
+            'package_name': 'libpng',
+            'nova_flavor': 'm1.medium',
+            'nova_image': 'fake.img',
+            }
+        self.worker = script.AdtNovaWorker
+        self.run_adt = MagicMock(name='run_adt')
+
+    @mock.patch('kombu.Connection.SimpleQueue')
+    def test_process(self, f_queue):
+        self.run_adt.return_value = 0
+        self.worker.process(self, self.body, message)
+        self.assertEqual(0, self.body['exit_code'])