1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
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)
|