summaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/utilities_tests/data_handler_test.py7
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')
-