1 files changed, 114 insertions, 0 deletions
diff --git a/yardstick/tests/functional/benchmark/core/test_report.py b/yardstick/tests/functional/benchmark/core/test_report.py
new file mode 100644
index 000000000..5f060dd1e
--- /dev/null
+++ b/yardstick/tests/functional/benchmark/core/test_report.py
@@ -0,0 +1,114 @@
+##############################################################################
+# Copyright (c) 2018 Intel Corporation.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+import ast
+import tempfile
+import unittest
+
+import mock
+from six.moves import configparser
+
+from yardstick.benchmark import core
+from yardstick.benchmark.core import report
+from yardstick.cmd.commands import change_osloobj_to_paras
+
+
+GOOD_YAML_NAME = 'fake_name'
+GOOD_TASK_ID = "9cbe74b6-df09-4535-8bdc-dc3a43b8a4e2"
+GOOD_DB_FIELDKEYS = [
+    {u'fieldKey': u'metric1', u'fieldType': u'integer'},
+    {u'fieldKey': u'metric4', u'fieldType': u'integer'},
+    {u'fieldKey': u'metric2', u'fieldType': u'integer'},
+    {u'fieldKey': u'metric3', u'fieldType': u'integer'},
+]
+GOOD_DB_METRICS = [
+    {u'time': u'2018-08-20T16:49:26.372662016Z',
+     u'metric1': 1, u'metric2': 0, u'metric3': 8, u'metric4': 5},
+    {u'time': u'2018-08-20T16:49:27.374208000Z',
+     u'metric1': 1, u'metric2': 1, u'metric3': 5, u'metric4': 4},
+    {u'time': u'2018-08-20T16:49:28.375742976Z',
+     u'metric1': 2, u'metric2': 2, u'metric3': 3, u'metric4': 3},
+    {u'time': u'2018-08-20T16:49:29.377299968Z',
+     u'metric1': 3, u'metric2': 3, u'metric3': 2, u'metric4': 2},
+    {u'time': u'2018-08-20T16:49:30.378252032Z',
+     u'metric1': 5, u'metric2': 4, u'metric3': 1, u'metric4': 1},
+    {u'time': u'2018-08-20T16:49:30.379359421Z',
+     u'metric1': 8, u'metric2': 5, u'metric3': 1, u'metric4': 0},
+]
+
+yardstick_config = """
+[DEFAULT]
+dispatcher = influxdb
+"""
+
+
+def my_query(query_sql):
+    get_fieldkeys_cmd = 'show field keys'
+    get_metrics_cmd = 'select * from'
+
+    if get_fieldkeys_cmd in query_sql:
+        return GOOD_DB_FIELDKEYS
+    elif get_metrics_cmd in query_sql:
+        return GOOD_DB_METRICS
+    return []
+
+
+class ReportTestCase(unittest.TestCase):
+
+    @mock.patch.object(report.influx, 'query', new=my_query)
+    @mock.patch.object(configparser.ConfigParser,
+        'read', side_effect=mock.mock_open(read_data=yardstick_config))
+    def test_report_generate_nsb_simple(self, *args):
+        tmpfile = tempfile.NamedTemporaryFile(delete=True)
+
+        args = core.Param({"task_id": [GOOD_TASK_ID], "yaml_name": [GOOD_YAML_NAME]})
+        params = change_osloobj_to_paras(args)
+
+        with mock.patch.object(report.consts, 'DEFAULT_HTML_FILE', tmpfile.name):
+            report.Report().generate_nsb(params)
+
+        data_act = None
+        time_act = None
+        keys_act = None
+        tree_act = None
+        with open(tmpfile.name) as f:
+            for l in f.readlines():
+                 if "var report_data = {" in l:
+                     data_act = ast.literal_eval(l.strip()[18:-1])
+                 elif "var report_time = [" in l:
+                     time_act = ast.literal_eval(l.strip()[18:-1])
+                 elif "var report_keys = [" in l:
+                     keys_act = ast.literal_eval(l.strip()[18:-1])
+                 elif "var report_tree = [" in l:
+                     tree_act = ast.literal_eval(l.strip()[18:-1])
+
+        data_exp = {
+            'metric1': [1, 1, 2, 3, 5, 8],
+            'metric2': [0, 1, 2, 3, 4, 5],
+            'metric3': [8, 5, 3, 2, 1, 1],
+            'metric4': [5, 4, 3, 2, 1, 0],
+        }
+        time_exp = [
+            '16:49:26.372662', '16:49:27.374208', '16:49:28.375742',
+            '16:49:29.377299', '16:49:30.378252', '16:49:30.379359',
+        ]
+        keys_exp = [
+            'metric1', 'metric2', 'metric3', 'metric4',
+        ]
+        tree_exp = [
+            {'parent': '#', 'text': 'metric1', 'id': 'metric1'},
+            {'parent': '#', 'text': 'metric2', 'id': 'metric2'},
+            {'parent': '#', 'text': 'metric3', 'id': 'metric3'},
+            {'parent': '#', 'text': 'metric4', 'id': 'metric4'},
+        ]
+
+        self.assertEqual(data_exp, data_act)
+        self.assertEqual(time_exp, time_act)
+        self.assertEqual(keys_exp, keys_act)
+        self.assertEqual(tree_exp, tree_act)