Merge "Bugfix: Correct the path in unit test."

1 files changed, 3 insertions, 3 deletions

diff --git a/tests/cli_test.py b/tests/cli_test.py
index cc8c7630..86dd6b98 100644
--- a/tests/cli_test.py
+++ b/tests/cli_test.py
@@ -1,7 +1,7 @@
 import pytest
 import mock
 import os
-from func.cli import Cli
+from qtip.utils.cli import Cli
 from os.path import expanduser
 
 
@@ -10,7 +10,7 @@ class TestClass:
         (['-l',
           'zte',
           '-f',
-          'compute'], "You have specified a lab that is not present under test_plan"),
+          'compute'], "You have specified a lab that is not present under benchmarks/test_plan"),
         (['-l',
           'default',
           '-f',
@@ -33,7 +33,7 @@ class TestClass:
           'storage'], [('fuel', '/home', 'benchmarks/test_plan/default/storage/fio_bm.yaml'),
                        ('fuel', '/home', 'benchmarks/test_plan/default/storage/fio_vm.yaml')])
     ])
-    @mock.patch('func.cli.args_handler.prepare_and_run_benchmark')
+    @mock.patch('qtip.utils.cli.args_handler.prepare_and_run_benchmark')
     def test_cli_successful(self, mock_args_handler, test_input, expected):
         k = mock.patch.dict(os.environ, {'INSTALLER_TYPE': 'fuel', 'PWD': '/home'})
         k.start()