Update cli_mock to stub out authorization_headers() in its new location.
Signed-off-by: Steve Howard <showard@google.com>
git-svn-id: http://test.kernel.org/svn/autotest/trunk@4233 592f7852-d20e-0410-864c-8624ca9c26a4
diff --git a/cli/cli_mock.py b/cli/cli_mock.py
index d5ee672..7d010f4 100644
--- a/cli/cli_mock.py
+++ b/cli/cli_mock.py
@@ -7,6 +7,7 @@
import common
from autotest_lib.cli import atest, topic_common, rpc
+from autotest_lib.frontend.afe import rpc_client_lib
from autotest_lib.frontend.afe.json_rpc import proxy
from autotest_lib.client.common_lib.test_utils import mock
from autotest_lib.client.common_lib import autotemp
@@ -34,7 +35,7 @@
def stub_authorization_headers(*args, **kwargs):
return {}
- self.god.stub_with(rpc, 'authorization_headers',
+ self.god.stub_with(rpc_client_lib, 'authorization_headers',
stub_authorization_headers)