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.assertIsNone(self.pm.connection_data.keyfile) 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 _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)