simplify test_results.py with eq() and templates/

Change-Id: Ibc5066f721b6b3343c2f8cb4ec0b9234960e9cb6
Signed-off-by: SerenaFeng <feng.xiaowei@zte.com.cn>

3 files changed, 79 insertions, 149 deletions

diff --git a/testapi/opnfv_testapi/models/result_models.py b/testapi/opnfv_testapi/models/result_models.py
index 97fda08..1dbe729 100644
--- a/testapi/opnfv_testapi/models/result_models.py
+++ b/testapi/opnfv_testapi/models/result_models.py
@@ -31,6 +31,14 @@ class TI(base_models.ModelBase):
         self.current = current
         self.histories = list()
 
+    def __eq__(self, other):
+        return (self.current == other.current and self._histories_eq(other))
+
+    def _histories_eq(self, other):
+        hs_equal = all(self.histories[i] == other.histories[i]
+                       for i in range(len(self.histories)))
+        return len(self.histories) == len(other.histories) and hs_equal
+
     @staticmethod
     def attr_parser():
         return {'histories': TIHistory}
@@ -72,6 +80,16 @@ class ResultCreateRequest(base_models.ModelBase):
         self.public = public
         self.trust_indicator = trust_indicator if trust_indicator else TI(0)
 
+    def __eq__(self, other):
+        simple_equal = all(getattr(self, k) == getattr(other, k)
+                           for k in self.format().keys()
+                           if k not in ['_id', 'trust_indicator'])
+        return simple_equal and self.trust_indicator == other.trust_indicator
+
+    @staticmethod
+    def attr_parser():
+        return {'trust_indicator': TI}
+
 
 @swagger.model()
 class ResultUpdateRequest(base_models.ModelBase):
@@ -84,35 +102,14 @@ class ResultUpdateRequest(base_models.ModelBase):
 
 
 @swagger.model()
-class TestResult(base_models.ModelBase):
+class TestResult(ResultCreateRequest):
     """
         @property trust_indicator: used for long duration test case
         @ptype trust_indicator: L{TI}
     """
-    def __init__(self, _id=None, case_name=None, project_name=None,
-                 pod_name=None, installer=None, version=None,
-                 start_date=None, stop_date=None, details=None,
-                 build_tag=None, scenario=None, criteria=None,
-                 user=None, public="true", trust_indicator=None):
+    def __init__(self, _id=None, **kwargs):
+        super(TestResult, self).__init__(**kwargs)
         self._id = _id
-        self.case_name = case_name
-        self.project_name = project_name
-        self.pod_name = pod_name
-        self.installer = installer
-        self.version = version
-        self.start_date = start_date
-        self.stop_date = stop_date
-        self.details = details
-        self.build_tag = build_tag
-        self.scenario = scenario
-        self.criteria = criteria
-        self.user = user
-        self.public = public
-        self.trust_indicator = trust_indicator
-
-    @staticmethod
-    def attr_parser():
-        return {'trust_indicator': TI}
 
 
 @swagger.model()
diff --git a/testapi/opnfv_testapi/tests/unit/handlers/test_result.py b/testapi/opnfv_testapi/tests/unit/handlers/test_result.py
index ac74741..f1f055e 100644
--- a/testapi/opnfv_testapi/tests/unit/handlers/test_result.py
+++ b/testapi/opnfv_testapi/tests/unit/handlers/test_result.py
@@ -12,116 +12,35 @@ from datetime import timedelta
 import httplib
 import json
 import urllib
-import unittest
 
 from opnfv_testapi.common import message
-from opnfv_testapi.models import result_models
-from opnfv_testapi.models import testcase_models
+from opnfv_testapi.models import result_models as rm
 from opnfv_testapi.tests.unit import executor
 from opnfv_testapi.tests.unit import fake_pymongo
 from opnfv_testapi.tests.unit.handlers import test_base as base
 
 
-class Details(object):
-    def __init__(self, timestart=None, duration=None, status=None):
-        self.timestart = timestart
-        self.duration = duration
-        self.status = status
-        self.items = [{'item1': 1}, {'item2': 2}]
-
-    def format(self):
-        return {
-            "timestart": self.timestart,
-            "duration": self.duration,
-            "status": self.status,
-            'items': [{'item1': 1}, {'item2': 2}]
-        }
-
-    @staticmethod
-    def from_dict(a_dict):
-
-        if a_dict is None:
-            return None
-
-        t = Details()
-        t.timestart = a_dict.get('timestart')
-        t.duration = a_dict.get('duration')
-        t.status = a_dict.get('status')
-        t.items = a_dict.get('items')
-        return t
-
-
 class TestResultBase(base.TestBase):
     @executor.mock_valid_lfid()
     def setUp(self):
         super(TestResultBase, self).setUp()
-        self.pod = self.pod_d.name
-        self.project = 'functest'
-        self.case = 'vPing'
-        self.installer = 'fuel'
-        self.version = 'C'
-        self.build_tag = 'v3.0'
-        self.scenario = 'odl-l2'
-        self.criteria = 'PASS'
-        self.trust_indicator = result_models.TI(0.7)
-        self.start_date = str(datetime.now())
-        self.stop_date = str(datetime.now() + timedelta(minutes=1))
-        self.update_date = str(datetime.now() + timedelta(days=1))
-        self.update_step = -0.05
-        self.details = Details(timestart='0', duration='9s', status='OK')
-        self.req_d = result_models.ResultCreateRequest(
-            pod_name=self.pod,
-            project_name=self.project,
-            case_name=self.case,
-            installer=self.installer,
-            version=self.version,
-            start_date=self.start_date,
-            stop_date=self.stop_date,
-            details=self.details.format(),
-            build_tag=self.build_tag,
-            scenario=self.scenario,
-            criteria=self.criteria,
-            trust_indicator=self.trust_indicator)
-        self.get_res = result_models.TestResult
-        self.list_res = result_models.TestResults
-        self.update_res = result_models.TestResult
+        self.req_d = rm.ResultCreateRequest.from_dict(
+            self.load_json('test_result'))
+        self.req_d.start_date = str(datetime.now())
+        self.req_d.stop_date = str(datetime.now() + timedelta(minutes=1))
+        self.get_res = rm.TestResult
+        self.list_res = rm.TestResults
+        self.update_res = rm.TestResult
         self.basePath = '/api/v1/results'
-        fake_pymongo.projects.insert(self.project_e.format())
-        self.req_testcase = testcase_models.TestcaseCreateRequest(
-            self.case,
-            '/cases/vping',
-            'vping-ssh test')
-        fake_pymongo.pods.insert(self.pod_d.format())
-        self.create_help('/api/v1/projects/%s/cases',
-                         self.req_testcase,
-                         self.project)
+        fake_pymongo.pods.insert({'name': self.req_d.pod_name})
+        fake_pymongo.projects.insert({'name': self.req_d.project_name})
+        fake_pymongo.testcases.insert({'name': self.req_d.case_name,
+                                       'project_name': self.req_d.project_name})
 
     def assert_res(self, result, req=None):
         if req is None:
             req = self.req_d
-        self.assertEqual(result.pod_name, req.pod_name)
-        self.assertEqual(result.project_name, req.project_name)
-        self.assertEqual(result.case_name, req.case_name)
-        self.assertEqual(result.installer, req.installer)
-        self.assertEqual(result.version, req.version)
-        details_req = Details.from_dict(req.details)
-        details_res = Details.from_dict(result.details)
-        self.assertEqual(details_res.duration, details_req.duration)
-        self.assertEqual(details_res.timestart, details_req.timestart)
-        self.assertEqual(details_res.status, details_req.status)
-        self.assertEqual(details_res.items, details_req.items)
-        self.assertEqual(result.build_tag, req.build_tag)
-        self.assertEqual(result.scenario, req.scenario)
-        self.assertEqual(result.criteria, req.criteria)
-        self.assertEqual(result.start_date, req.start_date)
-        self.assertEqual(result.stop_date, req.stop_date)
-        self.assertIsNotNone(result._id)
-        ti = result.trust_indicator
-        self.assertEqual(ti.current, req.trust_indicator.current)
-        if ti.histories:
-            history = ti.histories[0]
-            self.assertEqual(history.date, self.update_date)
-            self.assertEqual(history.step, self.update_step)
+        self.assertEqual(result, req)
 
     def _create_d(self):
         _, res = self.create_d()
@@ -204,17 +123,8 @@ class TestResultCreate(TestResultBase):
 
     @executor.create(httplib.OK, '_assert_no_ti')
     def test_no_ti(self):
-        req = result_models.ResultCreateRequest(pod_name=self.pod,
-                                                project_name=self.project,
-                                                case_name=self.case,
-                                                installer=self.installer,
-                                                version=self.version,
-                                                start_date=self.start_date,
-                                                stop_date=self.stop_date,
-                                                details=self.details.format(),
-                                                build_tag=self.build_tag,
-                                                scenario=self.scenario,
-                                                criteria=self.criteria)
+        req = copy.deepcopy(self.req_d)
+        req.trust_indicator = rm.TI(0)
         self.actual_req = req
         return req
 
@@ -370,8 +280,13 @@ class TestResultGet(TestResultBase):
 
     def _set_query(self, *args, **kwargs):
         def get_value(arg):
-            return self.__getattribute__(arg) \
-                if arg != 'trust_indicator' else self.trust_indicator.current
+            if arg in ['pod', 'project', 'case']:
+                return getattr(self.req_d, arg + '_name')
+            elif arg == 'trust_indicator':
+                return self.req_d.trust_indicator.current
+            else:
+                return getattr(self.req_d, arg)
+
         query = []
         for arg in args:
             query.append((arg, get_value(arg)))
@@ -387,24 +302,15 @@ class TestResultUpdate(TestResultBase):
 
     @executor.update(httplib.OK, '_assert_update_ti')
     def test_success(self):
-        new_ti = copy.deepcopy(self.trust_indicator)
-        new_ti.current += self.update_step
-        new_ti.histories.append(
-            result_models.TIHistory(self.update_date, self.update_step))
-        new_data = copy.deepcopy(self.req_d)
-        new_data.trust_indicator = new_ti
-        update = result_models.ResultUpdateRequest(trust_indicator=new_ti)
-        self.update_req = new_data
+        update_date = str(datetime.now() + timedelta(days=1))
+        update_step = -0.05
+        self.after_update = copy.deepcopy(self.req_d)
+        self.after_update.trust_indicator.current += update_step
+        self.after_update.trust_indicator.histories.append(
+            rm.TIHistory(update_date, update_step))
+        update = rm.ResultUpdateRequest(
+            trust_indicator=self.after_update.trust_indicator)
         return update, self.req_d_id
 
     def _assert_update_ti(self, request, body):
-        ti = body.trust_indicator
-        self.assertEqual(ti.current, request.trust_indicator.current)
-        if ti.histories:
-            history = ti.histories[0]
-            self.assertEqual(history.date, self.update_date)
-            self.assertEqual(history.step, self.update_step)
-
-
-if __name__ == '__main__':
-    unittest.main()
+        self.assert_res(body, self.after_update)
diff --git a/testapi/opnfv_testapi/tests/unit/templates/test_result.json b/testapi/opnfv_testapi/tests/unit/templates/test_result.json
new file mode 100644
index 0000000..b7cb910
--- /dev/null
+++ b/testapi/opnfv_testapi/tests/unit/templates/test_result.json
@@ -0,0 +1,27 @@
+{
+  "project_name": "functest",
+  "scenario": "odl-l2",
+  "trust_indicator": {
+    "current": 0.7,
+    "histories": []
+  },
+  "case_name": "vPing",
+  "build_tag": "v3.0",
+  "public": "true",
+  "version": "C",
+  "details": {
+    "timestart": "0",
+    "duration": "9s",
+    "status": "OK",
+    "items": [
+      {"item1": 1},
+      {"item2": 2}
+    ]
+  },
+  "criteria": "PASS",
+  "installer": "fuel",
+  "pod_name": "zte-pod1",
+  "start_date": "",
+  "stop_date": "",
+  "user": ""
+}
\ No newline at end of file