summaryrefslogtreecommitdiffstats
path: root/tests/driver_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/driver_test.py')
-rw-r--r--tests/driver_test.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/driver_test.py b/tests/driver_test.py
index 41a7dee4..d3025bd2 100644
--- a/tests/driver_test.py
+++ b/tests/driver_test.py
@@ -1,6 +1,6 @@
import pytest
import mock
-from func.driver import Driver
+from utils.driver import Driver
from os.path import expanduser
HOME_DIR = expanduser('~')
@@ -65,8 +65,8 @@ class TestClass:
'bandwidthGbps': 0,
"role": "2-host"}])
])
- @mock.patch('func.driver.AnsibleApi.execute_playbook')
- @mock.patch('func.driver.AnsibleApi.get_detail_playbook_stats')
+ @mock.patch('utils.driver.AnsibleApi.execute_playbook')
+ @mock.patch('utils.driver.AnsibleApi.get_detail_playbook_stats')
def test_driver_success(self, mock_stats, mock_ansible, test_input, expected):
mock_ansible.return_value = True
mock_stats.return_value = [(u'10.20.6.14', {'unreachable': 0,