def test_keypair_cleanup(self, mock_safe_remove, mock_osclients):
keypair_ctx = keypair.Keypair(self.ctx_with_keys)
keypair_ctx.cleanup()
mock_clients = mock_osclients.return_value
mock_nova = mock_clients.nova.return_value
self.assertEqual([mock.call(mock_nova)] * self.users,
mock_safe_remove.mock_calls)
def test_keypair_safe_remove(self):
mock_nova = mock.MagicMock()
keypair_ctx = keypair.Keypair(self.ctx_without_keys)
keypair_ctx._keypair_safe_remove(mock_nova)
self.assertEqual(
[mock.call.delete('rally_ssh_key')],
mock_nova.keypairs.mock_calls)
def test_keypair_generate(self, mock_osclients, mock_safe_remove):
keypair_ctx = keypair.Keypair(self.ctx_without_keys)
keypair_ctx._generate_keypair('endpoint')
mock_clients = mock_osclients.return_value
mock_nova = mock_clients.nova.return_value
self.assertIn(mock.call().nova().keypairs.create('rally_ssh_key'),
mock_osclients.mock_calls)
mock_safe_remove.assert_called_once_with(mock_nova)
def test_keypair_setup(self, mock_generate):
mock_generate.side_effect = [
{
"id": "key_id",
"key": "key",
"name": self.keypair_name
},
{
"id": "key_id",
"key": "key",
"name": self.keypair_name
},
]
keypair_ctx = keypair.Keypair(self.ctx_without_keys)
keypair_ctx.setup()
self.assertEqual(self.ctx_with_keys, keypair_ctx.context)
self.assertEqual([mock.call("endpoint")] * 2, mock_generate.mock_calls)
def test_keypair_generate(self, mock_osclients):
mock_keypairs = mock_osclients.return_value.nova.return_value.keypairs
mock_keypair = mock_keypairs.create.return_value
mock_keypair.public_key = "public_key"
mock_keypair.private_key = "private_key"
mock_keypair.id = "key_id"
keypair_ctx = keypair.Keypair(self.ctx_without_keys)
key = keypair_ctx._generate_keypair("endpoint")
self.assertEqual(
{
"id": "key_id",
"name": "rally_ssh_key_foo_task_id",
"private": "private_key",
"public": "public_key"
}, key)
mock_osclients.assert_has_calls([
mock.call().nova().keypairs.delete("rally_ssh_key_foo_task_id"),
mock.call().nova().keypairs.create("rally_ssh_key_foo_task_id"),
])
def test_keypair_setup(self, mock_generate):
mock_generate.return_value = "key"
keypair_ctx = keypair.Keypair(self.ctx_without_keys)
keypair_ctx.setup()
self.assertEqual(self.ctx_without_keys, self.ctx_with_keys)
def test_keypair_cleanup(self, mock_cleanup):
keypair_ctx = keypair.Keypair(self.ctx_with_keys)
keypair_ctx.cleanup()
mock_cleanup.assert_called_once_with(names=["nova.keypairs"],
users=self.ctx_with_keys["users"])
