def test_delete_aggregate(self, mock_nova):
mock_nova.aggregates.list.return_value = [
fakes.FakeAggregate('1234', 'name')
]
self.assertTrue(self.cloud.delete_aggregate('1234'))
mock_nova.aggregates.list.assert_called_once_with()
mock_nova.aggregates.delete.assert_called_once_with(aggregate='1234')
def test_remove_host_from_aggregate(self, mock_nova):
hostname = 'host1'
mock_nova.aggregates.list.return_value = [
fakes.FakeAggregate('1234', 'name', hosts=[hostname])
]
self.cloud.remove_host_from_aggregate('1234', hostname)
mock_nova.aggregates.remove_host.assert_called_once_with(
aggregate='1234', host=hostname)
def test_add_host_to_aggregate(self, mock_nova):
hostname = 'host1'
mock_nova.aggregates.list.return_value = [
fakes.FakeAggregate('1234', 'name')
]
self.cloud.add_host_to_aggregate('1234', hostname)
mock_nova.aggregates.add_host.assert_called_once_with(aggregate='1234',
host=hostname)
def test_set_aggregate_metadata(self, mock_nova):
metadata = {'key', 'value'}
mock_nova.aggregates.list.return_value = [
fakes.FakeAggregate('1234', 'name')
]
self.cloud.set_aggregate_metadata('1234', metadata)
mock_nova.aggregates.set_metadata.assert_called_once_with(
aggregate='1234', metadata=metadata)
def test_update_aggregate_unset_az(self, mock_nova):
mock_nova.aggregates.list.return_value = [
fakes.FakeAggregate('1234', 'name', availability_zone='az')
]
self.cloud.update_aggregate('1234', availability_zone=None)
mock_nova.aggregates.update.assert_called_once_with(
aggregate='1234',
values={'availability_zone': None},
)
