Add unit test for guestagent.service.py and query.py

Bug # 1090139

Change-Id: Ib6ddf1f1823ccec348cf552daaee0c127940aa29

reddwarf/tests/guestagent/test_query.py

@@ -0,0 +1,76 @@
+#    Copyright 2012 OpenStack LLC
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License
+
+import testtools
+from reddwarf.guestagent import query
+from proboscis import test
+
+
+@test(groups=["dbaas.guestagent.dbaas"])
+class QueryTest(testtools.TestCase):
+    def setUp(self):
+        super(QueryTest, self).setUp()
+
+    def tearDown(self):
+        super(QueryTest, self).tearDown()
+
+    def test_columns(self):
+        myQuery = query.Query(columns=None)
+        self.assertEqual("SELECT *", myQuery._columns)
+
+    def test_columns_2(self):
+        columns = ["col_A", "col_B"]
+        myQuery = query.Query(columns=columns)
+        self.assertEqual("SELECT col_A, col_B", myQuery._columns)
+
+    def test_tables(self):
+        tables = ['table_A', 'table_B']
+        myQuery = query.Query(tables=tables)
+        self.assertEqual("FROM table_A, table_B", myQuery._tables)
+
+    def test_where(self):
+        myQuery = query.Query(where=None)
+        self.assertEqual("", myQuery._where)
+
+    def test_where_2(self):
+        conditions = ['cond_A', 'cond_B']
+        myQuery = query.Query(where=conditions)
+        self.assertEqual("WHERE cond_A AND cond_B", myQuery._where)
+
+    def test_order(self):
+        myQuery = query.Query(order=None)
+        self.assertEqual('', myQuery._order)
+
+    def test_order_2(self):
+        orders = ['deleted_at', 'updated_at']
+        myQuery = query.Query(order=orders)
+        self.assertEqual('ORDER BY deleted_at, updated_at', myQuery._order)
+
+    def test_group_by(self):
+        myQuery = query.Query(group=None)
+        self.assertEqual('', myQuery._group_by)
+
+    def test_group_by_2(self):
+        groups = ['deleted=1']
+        myQuery = query.Query(group=groups)
+        self.assertEqual('GROUP BY deleted=1', myQuery._group_by)
+
+    def test_limit(self):
+        myQuery = query.Query(limit=None)
+        self.assertEqual('', myQuery._limit)
+
+    def test_limit_2(self):
+        limit_count = 20
+        myQuery = query.Query(limit=limit_count)
+        self.assertEqual('LIMIT 20', myQuery._limit)

reddwarf/tests/guestagent/test_service.py

@@ -0,0 +1,32 @@
+#    Copyright 2012 OpenStack LLC
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License
+
+import testtools
+from mock import Mock, MagicMock
+from reddwarf.guestagent import service
+from proboscis import test
+
+
+@test(groups=["dbaas.guestagent.dbaas"])
+class ServiceTest(testtools.TestCase):
+    def setUp(self):
+        super(ServiceTest, self).setUp()
+
+    def tearDown(self):
+        super(ServiceTest, self).tearDown()
+
+    def test_app_factory(self):
+        service.API._instance_router = MagicMock()
+        service.app_factory(Mock)
+        self.assertEqual(1, service.API._instance_router.call_count)

run_tests.py

@@ -125,5 +125,7 @@ if __name__=="__main__":
     from reddwarf.tests.api.mgmt import storage
     from reddwarf.tests.guestagent import test_dbaas
     from reddwarf.tests.guestagent import test_manager
+    from reddwarf.tests.guestagent import test_service
+    from reddwarf.tests.guestagent import test_query
 
     proboscis.TestProgram().run_and_exit()