diff options
Diffstat (limited to 'tests/utilities_tests/data_handler_test.py')
| -rw-r--r-- | tests/utilities_tests/data_handler_test.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/utilities_tests/data_handler_test.py b/tests/utilities_tests/data_handler_test.py index 7963c9f..90df0f6 100644 --- a/tests/utilities_tests/data_handler_test.py +++ b/tests/utilities_tests/data_handler_test.py @@ -8,11 +8,12 @@ ############################################################################## import os -from storperf.utilities.data_handler import DataHandler import unittest import mock +from storperf.utilities.data_handler import DataHandler + class MockGraphiteDB(object): @@ -54,7 +55,8 @@ class DataHandlerTest(unittest.TestCase): def push_results_to_db(self, *args): self.pushed = True self.db_results = args - pass + results = {"href": "http://localhost/api/result/uuid-that-is-long"} + return results def terminate(self): self._terminated = True @@ -283,4 +285,3 @@ class DataHandlerTest(unittest.TestCase): 'Start time') self.assertEqual('2017-09-04 21:20:00', self.db_results[4], 'End time') - |