From 54c0e5c1682f38a3e805b29b9e8219d7c6c675f1 Mon Sep 17 00:00:00 2001
From: gengchc2 <geng.changcai2@zte.com.cn>
Date: Fri, 9 Nov 2018 01:19:01 -0800
Subject: [PATCH] Add unit test for job table in sqlalchemy

Change-Id: I5de2151f2e88da0fe3dcfcae1b55b8b6d4b53e52
---
 freezer_api/db/base.py                        |  1 +
 freezer_api/tests/unit/sqlalchemy/test_job.py | 61 +++++++++++++++++++
 2 files changed, 62 insertions(+)
 create mode 100644 freezer_api/tests/unit/sqlalchemy/test_job.py

diff --git a/freezer_api/db/base.py b/freezer_api/db/base.py
index 6d0919e1..81069433 100644
--- a/freezer_api/db/base.py
+++ b/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"),
diff --git a/freezer_api/tests/unit/sqlalchemy/test_job.py b/freezer_api/tests/unit/sqlalchemy/test_job.py
new file mode 100644
index 00000000..e07554d5
--- /dev/null
+++ b/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'))