def test__get_last_value_time(self):
influxdb_helper = vcmts_vnf.InfluxDBHelper("localhost", 8086)
self.assertEqual(influxdb_helper._get_last_value_time('cpu_value'),
vcmts_vnf.InfluxDBHelper.INITIAL_VALUE)
influxdb_helper._last_values_time['cpu_value'] = "RANDOM"
self.assertEqual(influxdb_helper._get_last_value_time('cpu_value'),
"RANDOM")
def test__query_measurement(self):
influxdb_helper = vcmts_vnf.InfluxDBHelper("localhost", 8086)
influxdb_helper._read_client = mock.MagicMock()
resulted_generator = mock.MagicMock()
resulted_generator.keys.return_value = []
influxdb_helper._read_client.query.return_value = resulted_generator
query_result = influxdb_helper._query_measurement('cpu_value')
self.assertIsNone(query_result)
resulted_generator = mock.MagicMock()
resulted_generator.keys.return_value = ["", ""]
resulted_generator.get_points.return_value = ResultSet({"": ""})
influxdb_helper._read_client.query.return_value = resulted_generator
query_result = influxdb_helper._query_measurement('cpu_value')
self.assertIsNotNone(query_result)
def test__rw_measurment(self):
influxdb_helper = vcmts_vnf.InfluxDBHelper("localhost", 8086)
influxdb_helper._query_measurement = mock.MagicMock()
influxdb_helper._query_measurement.return_value = None
influxdb_helper._rw_measurment('cpu_value', [])
self.assertEqual(len(influxdb_helper._last_values_time), 0)
entry = {
"type": "type",
"host": "host",
"time": "time",
"id": "1",
"value": "1.0"
}
influxdb_helper._query_measurement.return_value = [entry]
influxdb_helper._write_client = mock.MagicMock()
influxdb_helper._rw_measurment('cpu_value', ["id", "value"])
self.assertEqual(len(influxdb_helper._last_values_time), 1)
influxdb_helper._write_client.write_points.assert_called_once()
def test_copy_kpi(self):
influxdb_helper = vcmts_vnf.InfluxDBHelper("localhost", 8086)
influxdb_helper._rw_measurment = mock.MagicMock()
influxdb_helper.copy_kpi()
influxdb_helper._rw_measurment.assert_called()
def test__set_last_value_time(self):
influxdb_helper = vcmts_vnf.InfluxDBHelper("localhost", 8086)
influxdb_helper._set_last_value_time('cpu_value', '00:00')
self.assertEqual(influxdb_helper._last_values_time['cpu_value'],
"'00:00'")
def test_start(self):
influxdb_helper = vcmts_vnf.InfluxDBHelper("localhost", 8086)
influxdb_helper.start()
self.assertIsNotNone(influxdb_helper._read_client)
self.assertIsNotNone(influxdb_helper._write_client)
def test___init__(self):
influxdb_helper = vcmts_vnf.InfluxDBHelper("localhost", 8086)
self.assertEqual(influxdb_helper._vcmts_influxdb_ip, "localhost")
self.assertEqual(influxdb_helper._vcmts_influxdb_port, 8086)
self.assertIsNotNone(influxdb_helper._last_upstream_rx)
self.assertIsNotNone(influxdb_helper._last_values_time)