From 9bc09d1d18d4777438515393a9427fc0733e6f09 Mon Sep 17 00:00:00 2001 From: dongwenjuan Date: Wed, 1 Aug 2018 11:25:28 +0800 Subject: add ssh_keyfile for connect to installer in Apex Change-Id: I8809a24e75c0260f67d3e128b8bbd1becf924529 Signed-off-by: dongwenjuan --- doctor_tests/installer/__init__.py | 4 ++++ doctor_tests/installer/apex.py | 1 + doctor_tests/installer/mcp.py | 3 ++- 3 files changed, 7 insertions(+), 1 deletion(-) (limited to 'doctor_tests') diff --git a/doctor_tests/installer/__init__.py b/doctor_tests/installer/__init__.py index 31fce754..ee44018c 100644 --- a/doctor_tests/installer/__init__.py +++ b/doctor_tests/installer/__init__.py @@ -24,6 +24,10 @@ OPTS = [ default='root', help='the user name for login installer server', required=True), + cfg.StrOpt('key_file', + default=os.environ.get('SSH_KEY', None), + help='the key for user to login installer server', + required=False), ] diff --git a/doctor_tests/installer/apex.py b/doctor_tests/installer/apex.py index 1ce3eb65..c851b2ac 100644 --- a/doctor_tests/installer/apex.py +++ b/doctor_tests/installer/apex.py @@ -19,6 +19,7 @@ class ApexInstaller(BaseInstaller): super(ApexInstaller, self).__init__(conf, log) self.client = SSHClient(self.conf.installer.ip, self.conf.installer.username, + key_filename=self.conf.installer.key_file, look_for_keys=True) self.key_file = None self.controllers = list() diff --git a/doctor_tests/installer/mcp.py b/doctor_tests/installer/mcp.py index 8ba9f000..e7e41dbe 100644 --- a/doctor_tests/installer/mcp.py +++ b/doctor_tests/installer/mcp.py @@ -22,7 +22,8 @@ class McpInstaller(BaseInstaller): self.key_file = self.get_ssh_key_from_installer() self.client = SSHClient(self.conf.installer.ip, self.node_user_name, - key_filename=self.key_file) + key_filename=self.key_file, + look_for_keys=True) self.controllers = list() self.controller_clients = list() -- cgit 1.2.3-korg