diff options
Diffstat (limited to 'tests/driver_test.py')
| -rw-r--r-- | tests/driver_test.py | 6 |
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, |