mirror of
https://opendev.org/openstack/ansible-collections-openstack.git
synced 2026-07-25 09:14:31 +00:00
Add host_aggregate_info module
This module allows operators to get informations on compute aggregates in a cluster, including availability zone informations and hosts membership. Change-Id: I14fa8d6914072bee314efa3753633933b21e9439 Signed-off-by: Bertrand Lanson <bertrand.lanson@infomaniak.com>
This commit is contained in:
@@ -0,0 +1,97 @@
|
||||
import unittest
|
||||
from unittest import mock
|
||||
|
||||
from ansible_collections.openstack.cloud.plugins.modules import host_aggregate_info
|
||||
from ansible_collections.openstack.cloud.tests.unit.modules.utils import (
|
||||
AnsibleExitJson,
|
||||
ModuleTestCase,
|
||||
set_module_args,
|
||||
)
|
||||
|
||||
|
||||
class FakeAggregate(dict):
|
||||
|
||||
def to_dict(self, computed=False):
|
||||
return dict(self)
|
||||
|
||||
|
||||
FAKE_AGGREGATE = FakeAggregate(
|
||||
id=7,
|
||||
uuid='583bf5e7-b228-4850-ad3e-9a0b7aa6c8da',
|
||||
name='dc1-a1',
|
||||
availability_zone='dc1-a1',
|
||||
hosts=['o3080-r01u07-1'],
|
||||
metadata={'availability_zone': 'dc1-a1'},
|
||||
created_at='2026-05-16T23:46:45.000000',
|
||||
updated_at=None,
|
||||
deleted_at=None,
|
||||
is_deleted=False,
|
||||
)
|
||||
|
||||
|
||||
class FakeSDK(object):
|
||||
class exceptions:
|
||||
class OpenStackCloudException(Exception):
|
||||
pass
|
||||
|
||||
|
||||
class TestHostAggregateInfo(ModuleTestCase):
|
||||
|
||||
def _run_module(self, module_args, compute):
|
||||
set_module_args(module_args)
|
||||
conn = mock.Mock()
|
||||
conn.compute = compute
|
||||
with mock.patch.object(
|
||||
host_aggregate_info.ComputeHostAggregateInfoModule,
|
||||
'openstack_cloud_from_module',
|
||||
return_value=(FakeSDK(), conn),
|
||||
):
|
||||
host_aggregate_info.main()
|
||||
|
||||
def _new_compute(self):
|
||||
compute = mock.Mock()
|
||||
compute.aggregates.return_value = [FAKE_AGGREGATE]
|
||||
compute.find_aggregate.return_value = FAKE_AGGREGATE
|
||||
return compute
|
||||
|
||||
def test_list_all_aggregates(self):
|
||||
compute = self._new_compute()
|
||||
|
||||
with self.assertRaises(AnsibleExitJson) as ex:
|
||||
self._run_module({'name': None}, compute)
|
||||
|
||||
result = ex.exception.args[0]
|
||||
self.assertFalse(result['changed'])
|
||||
self.assertEqual(1, len(result['aggregates']))
|
||||
self.assertEqual('dc1-a1', result['aggregates'][0]['name'])
|
||||
compute.aggregates.assert_called_once_with()
|
||||
compute.find_aggregate.assert_not_called()
|
||||
|
||||
def test_find_aggregate_by_name(self):
|
||||
compute = self._new_compute()
|
||||
|
||||
with self.assertRaises(AnsibleExitJson) as ex:
|
||||
self._run_module({'name': 'dc1-a1'}, compute)
|
||||
|
||||
result = ex.exception.args[0]
|
||||
self.assertFalse(result['changed'])
|
||||
self.assertEqual(1, len(result['aggregates']))
|
||||
self.assertEqual('dc1-a1', result['aggregates'][0]['name'])
|
||||
compute.find_aggregate.assert_called_once_with('dc1-a1')
|
||||
compute.aggregates.assert_not_called()
|
||||
|
||||
def test_find_aggregate_by_name_not_found(self):
|
||||
compute = self._new_compute()
|
||||
compute.find_aggregate.return_value = None
|
||||
|
||||
with self.assertRaises(AnsibleExitJson) as ex:
|
||||
self._run_module({'name': 'nonexistent'}, compute)
|
||||
|
||||
result = ex.exception.args[0]
|
||||
self.assertFalse(result['changed'])
|
||||
self.assertEqual([], result['aggregates'])
|
||||
compute.find_aggregate.assert_called_once_with('nonexistent')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user