Merge "Add unit test for job table in sqlalchemy"

freezer_api/db/base.py

@@ -27,6 +27,7 @@ class DBDriver(object):
     _OPTS = [
         cfg.StrOpt('host',
                    required=True,
+                   default='0.0.0.0',
                    help="Database host"),
         cfg.StrOpt("username",
                    help="Database username"),

freezer_api/tests/unit/sqlalchemy/test_job.py

@@ -0,0 +1,61 @@
+# -*- encoding: utf-8 -*-
+# (c) Copyright 2018 ZTE Corporation.
+#
+# 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.
+
+
+"""Tests for manipulating job via the DB API"""
+
+import copy
+
+from freezer_api.tests.unit import common
+from freezer_api.tests.unit.sqlalchemy import base
+
+
+class DbJobTestCase(base.DbTestCase):
+
+    def setUp(self):
+        super(DbJobTestCase, self).setUp()
+        self.fake_job_0 = common.get_fake_job_0()
+        self.fake_job_0.pop('job_id')
+
+    def test_add_and_get_job(self):
+        job_doc = copy.deepcopy(self.fake_job_0)
+        job_id = self.dbapi.add_job(user_id=self.fake_job_0.get('user_id'),
+                                    doc=job_doc,
+                                    project_id="myproject")
+        self.assertIsNotNone(job_id)
+        result = self.dbapi.get_job(project_id="myproject",
+                                    user_id=self.fake_job_0.get('user_id'),
+                                    job_id=job_id)
+        self.assertIsNotNone(result)
+        self.assertEqual(result.get('client_id'),
+                         self.fake_job_0.get('client_id'))
+        self.assertEqual(result.get('description'),
+                         self.fake_job_0.get('description'))
+        self.assertEqual(result.get('job_actions'),
+                         self.fake_job_0.get('job_actions'))
+        self.assertEqual(result.get('user_id'),
+                         self.fake_job_0.get('user_id'))
+        self.assertEqual(result.get('job_schedule').get('status'),
+                         self.fake_job_0.get('job_schedule').get('status'))
+        self.assertEqual(result.get('job_schedule').get('result'),
+                         self.fake_job_0.get('job_schedule').get('result'))
+        self.assertEqual(result.get('job_schedule').get('schedule_date'),
+                         self.fake_job_0.get('job_schedule').
+                         get('schedule_date'))
+        self.assertEqual(result.get('job_schedule').
+                         get('schedule_interval'),
+                         self.fake_job_0.get('job_schedule').
+                         get('schedule_interval'))