Better error handling and message when the (site) authentication didn't work.
Signed-off-by: Jean-Marc Eurin <jmeurin@google.com>
git-svn-id: http://test.kernel.org/svn/autotest/trunk@4299 592f7852-d20e-0410-864c-8624ca9c26a4
diff --git a/cli/rpc.py b/cli/rpc.py
index 1b50ea7..1cff0c1 100644
--- a/cli/rpc.py
+++ b/cli/rpc.py
@@ -13,6 +13,10 @@
TKO_RPC_PATH = '/new_tko/server/rpc/'
+class AuthError(Exception):
+ pass
+
+
def get_autotest_server(web_server=None):
if not web_server:
if 'AUTOTEST_WEB' in os.environ:
@@ -34,7 +38,10 @@
def __init__(self, web_server, rpc_path, username):
self.username = username
self.web_server = get_autotest_server(web_server)
- self.proxy = self._connect(rpc_path)
+ try:
+ self.proxy = self._connect(rpc_path)
+ except rpc_client_lib.AuthError, s:
+ raise AuthError(s)
def _connect(self, rpc_path):