def test_get_conn_success(self): self._create_node() self._create_pm() self._conn = self.pm._get_conn() self.mox.StubOutWithMock(connection, "ssh_connect") connection.ssh_connect(mox.IsA(self._conn)).AndReturn(True) self.mox.ReplayAll() self.pm._set_connection() self.assertEqual(self.pm.connection_data.host, "127.0.0.1") self.assertEqual(self.pm.connection_data.username, "user") self.assertEqual(self.pm.connection_data.password, "password") self.mox.VerifyAll()
def test_get_conn_success_pass(self): self._create_node() self._create_pm() self._conn = self.pm._get_conn() self.mox.StubOutWithMock(connection, 'ssh_connect') connection.ssh_connect(mox.IsA(self._conn)).AndReturn(True) self.mox.ReplayAll() self.pm._set_connection() self.assertEqual(self.pm.connection_data.host, '127.0.0.1') self.assertEqual(self.pm.connection_data.username, 'user') self.assertEqual(self.pm.connection_data.password, 'password') self.assertEqual(self.pm.connection_data.keyfile, None) self.mox.VerifyAll()
def test_get_conn_success(self): self.flags(virtual_power_ssh_host='127.0.0.1', group="baremetal") self.flags(virtual_power_host_user='******', group="baremetal") self.flags(virtual_power_host_pass='******', group="baremetal") self._create_node() self._create_pm() self._conn = self.pm._get_conn() self.mox.StubOutWithMock(connection, 'ssh_connect') connection.ssh_connect(mox.IsA(self._conn)).AndReturn(True) self.mox.ReplayAll() self.pm._set_connection() self.assertEqual(self.pm.connection_data.host, '127.0.0.1') self.assertEqual(self.pm.connection_data.username, 'user') self.assertEqual(self.pm.connection_data.password, 'password') self.mox.VerifyAll()
def test_get_conn_success_key(self): self.config(virtual_power_host_pass="", group="baremetal") self.config(virtual_power_host_key="/id_rsa_file.txt", group="baremetal") self._create_node() self._create_pm() self._conn = self.pm._get_conn() self.mox.StubOutWithMock(connection, "ssh_connect") connection.ssh_connect(mox.IsA(self._conn)).AndReturn(True) self.mox.ReplayAll() self.pm._set_connection() self.assertEqual(self.pm.connection_data.host, "127.0.0.1") self.assertEqual(self.pm.connection_data.username, "user") self.assertEqual(self.pm.connection_data.password, "") self.assertEqual(self.pm.connection_data.keyfile, "/id_rsa_file.txt") self.mox.VerifyAll()
def test_get_conn_success_key(self): self.flags(virtual_power_host_pass='', group="baremetal") self.flags(virtual_power_host_key='/id_rsa_file.txt', group="baremetal") self._create_node() self._create_pm() self._conn = self.pm._get_conn() self.mox.StubOutWithMock(connection, 'ssh_connect') connection.ssh_connect(mox.IsA(self._conn)).AndReturn(True) self.mox.ReplayAll() self.pm._set_connection() self.assertEqual(self.pm.connection_data.host, '127.0.0.1') self.assertEqual(self.pm.connection_data.username, 'user') self.assertEqual(self.pm.connection_data.password, '') self.assertEqual(self.pm.connection_data.keyfile, '/id_rsa_file.txt') self.mox.VerifyAll()
def __init__(self, connection): """Constructor. :param connection: common.Connection object with the information to connect to the remote ssh. """ self._connection = common.ssh_connect(connection) self.connection_data = connection
def _set_connection(self): if self._connection is None: self._connection = common.ssh_connect(self.connection_data)
def _set_connection(self): if self._connection is None: if self.connection_data is None: self.connection_data = self._get_conn() self._connection = connection.ssh_connect(self.connection_data)