diff options
Diffstat (limited to 'tests/unit/api/utils/test_influx.py')
| -rw-r--r-- | tests/unit/api/utils/test_influx.py | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/tests/unit/api/utils/test_influx.py b/tests/unit/api/utils/test_influx.py index 5f1e2c36f..0852da2dd 100644 --- a/tests/unit/api/utils/test_influx.py +++ b/tests/unit/api/utils/test_influx.py @@ -1,3 +1,11 @@ +############################################################################## +# Copyright (c) 2016 Huawei Technologies Co.,Ltd and others. +# +# 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 unittest import mock import uuid @@ -13,7 +21,7 @@ class GetDataDbClientTestCase(unittest.TestCase): mock_parser.ConfigParser().get.return_value = 'file' try: influx.get_data_db_client() - except Exception, e: + except Exception as e: self.assertIsInstance(e, RuntimeError) @@ -54,6 +62,18 @@ class WriteDataTasklistTestCase(unittest.TestCase): mock_write_data.assert_called_with('tasklist', field, timestamp, tags) +class QueryTestCase(unittest.TestCase): + + @mock.patch('api.utils.influx.ConfigParser') + def test_query_dispatcher_not_influxdb(self, mock_parser): + mock_parser.ConfigParser().get.return_value = 'file' + try: + sql = 'select * form tasklist' + influx.query(sql) + except Exception as e: + self.assertIsInstance(e, RuntimeError) + + def main(): unittest.main() |