73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
# Copyright (c) 2019 Red Hat, Inc.
|
|
#
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
from __future__ import absolute_import
|
|
|
|
import io
|
|
import os
|
|
|
|
import mock
|
|
import paramiko
|
|
|
|
import tobiko
|
|
from tobiko import config
|
|
from tobiko.shell import ssh
|
|
from tobiko.tests import unit
|
|
|
|
|
|
CONF = config.CONF
|
|
|
|
|
|
class SSHClientFixtureTest(unit.TobikoUnitTest):
|
|
|
|
fixture = tobiko.required_fixture(ssh.SSHClientFixture)
|
|
|
|
expected_host = None
|
|
expected_proxy_client = None
|
|
|
|
def setUp(self):
|
|
super(SSHClientFixtureTest, self).setUp()
|
|
tobiko.cleanup_fixture(self.fixture)
|
|
self.ssh_client = mock.MagicMock(specs=paramiko.SSHClient)
|
|
self.patch(paramiko, 'SSHClient', return_value=self.ssh_client)
|
|
|
|
def test_init(self):
|
|
fixture = self.fixture
|
|
self.assertIs(self.expected_host, fixture.host)
|
|
self.assertIs(self.expected_proxy_client, fixture.proxy_client)
|
|
self.assertIsNone(fixture.host_config)
|
|
self.assertIsNone(fixture.global_host_config)
|
|
self.assertIsNone(fixture.connect_parameters)
|
|
|
|
def test_setup(self):
|
|
fixture = self.fixture
|
|
if not self.expected_host:
|
|
fixture.host = 'some-host'
|
|
fixture.username = 'some-username'
|
|
fixture.setUp()
|
|
|
|
ssh_config = paramiko.SSHConfig()
|
|
for ssh_config_file in CONF.tobiko.ssh.config_files:
|
|
ssh_config_file = tobiko.tobiko_config_path(ssh_config_file)
|
|
if os.path.exists(ssh_config_file):
|
|
with io.open(ssh_config_file, 'rt',
|
|
encoding="utf-8") as f:
|
|
ssh_config.parse(f)
|
|
expected_host_config = ssh_config.lookup(fixture.host)
|
|
expected_host_config.pop('include', None)
|
|
self.assertEqual(fixture.host, fixture.global_host_config.host)
|
|
self.assertEqual(expected_host_config,
|
|
fixture.global_host_config.host_config)
|