mirror of https://github.com/ceph/ceph-ansible.git
152 lines
4.7 KiB
Python
152 lines
4.7 KiB
Python
import os
|
|
import sys
|
|
from mock.mock import patch, MagicMock
|
|
import pytest
|
|
sys.path.append('./library')
|
|
import radosgw_zone # noqa: E402
|
|
|
|
|
|
fake_binary = 'radosgw-admin'
|
|
fake_cluster = 'ceph'
|
|
fake_container_binary = 'podman'
|
|
fake_container_image = 'docker.io/ceph/daemon:latest'
|
|
fake_container_cmd = [
|
|
fake_container_binary,
|
|
'run',
|
|
'--rm',
|
|
'--net=host',
|
|
'-v', '/etc/ceph:/etc/ceph:z',
|
|
'-v', '/var/lib/ceph/:/var/lib/ceph/:z',
|
|
'-v', '/var/log/ceph/:/var/log/ceph/:z',
|
|
'--entrypoint=' + fake_binary,
|
|
fake_container_image
|
|
]
|
|
fake_realm = 'foo'
|
|
fake_zonegroup = 'bar'
|
|
fake_zone = 'z1'
|
|
fake_endpoints = ['http://192.168.1.10:8080', 'http://192.168.1.11:8080']
|
|
fake_params = {'cluster': fake_cluster,
|
|
'name': fake_zone,
|
|
'realm': fake_realm,
|
|
'zonegroup': fake_zonegroup,
|
|
'endpoints': fake_endpoints,
|
|
'default': True,
|
|
'master': True}
|
|
|
|
|
|
class TestRadosgwZoneModule(object):
|
|
|
|
@patch.dict(os.environ, {'CEPH_CONTAINER_BINARY': fake_container_binary})
|
|
def test_container_exec(self):
|
|
cmd = radosgw_zone.container_exec(fake_binary, fake_container_image)
|
|
assert cmd == fake_container_cmd
|
|
|
|
def test_not_is_containerized(self):
|
|
assert radosgw_zone.is_containerized() is None
|
|
|
|
@patch.dict(os.environ, {'CEPH_CONTAINER_IMAGE': fake_container_image})
|
|
def test_is_containerized(self):
|
|
assert radosgw_zone.is_containerized() == fake_container_image
|
|
|
|
@pytest.mark.parametrize('image', [None, fake_container_image])
|
|
@patch.dict(os.environ, {'CEPH_CONTAINER_BINARY': fake_container_binary})
|
|
def test_pre_generate_radosgw_cmd(self, image):
|
|
if image:
|
|
expected_cmd = fake_container_cmd
|
|
else:
|
|
expected_cmd = [fake_binary]
|
|
|
|
assert radosgw_zone.pre_generate_radosgw_cmd(image) == expected_cmd
|
|
|
|
@pytest.mark.parametrize('image', [None, fake_container_image])
|
|
@patch.dict(os.environ, {'CEPH_CONTAINER_BINARY': fake_container_binary})
|
|
def test_generate_radosgw_cmd(self, image):
|
|
if image:
|
|
expected_cmd = fake_container_cmd
|
|
else:
|
|
expected_cmd = [fake_binary]
|
|
|
|
expected_cmd.extend([
|
|
'--cluster',
|
|
fake_cluster,
|
|
'zone'
|
|
])
|
|
assert radosgw_zone.generate_radosgw_cmd(fake_cluster, [], image) == expected_cmd
|
|
|
|
def test_create_zone(self):
|
|
fake_module = MagicMock()
|
|
fake_module.params = fake_params
|
|
expected_cmd = [
|
|
fake_binary,
|
|
'--cluster', fake_cluster,
|
|
'zone', 'create',
|
|
'--rgw-realm=' + fake_realm,
|
|
'--rgw-zonegroup=' + fake_zonegroup,
|
|
'--rgw-zone=' + fake_zone,
|
|
'--endpoints=' + ','.join(fake_endpoints),
|
|
'--default',
|
|
'--master'
|
|
]
|
|
|
|
assert radosgw_zone.create_zone(fake_module) == expected_cmd
|
|
|
|
def test_modify_zone(self):
|
|
fake_module = MagicMock()
|
|
fake_module.params = fake_params
|
|
expected_cmd = [
|
|
fake_binary,
|
|
'--cluster', fake_cluster,
|
|
'zone', 'modify',
|
|
'--rgw-realm=' + fake_realm,
|
|
'--rgw-zonegroup=' + fake_zonegroup,
|
|
'--rgw-zone=' + fake_zone,
|
|
'--endpoints=' + ','.join(fake_endpoints),
|
|
'--default',
|
|
'--master'
|
|
]
|
|
|
|
assert radosgw_zone.modify_zone(fake_module) == expected_cmd
|
|
|
|
def test_get_zone(self):
|
|
fake_module = MagicMock()
|
|
fake_module.params = fake_params
|
|
expected_cmd = [
|
|
fake_binary,
|
|
'--cluster', fake_cluster,
|
|
'zone', 'get',
|
|
'--rgw-realm=' + fake_realm,
|
|
'--rgw-zonegroup=' + fake_zonegroup,
|
|
'--rgw-zone=' + fake_zone,
|
|
'--format=json'
|
|
]
|
|
|
|
assert radosgw_zone.get_zone(fake_module) == expected_cmd
|
|
|
|
def test_get_zonegroup(self):
|
|
fake_module = MagicMock()
|
|
fake_module.params = fake_params
|
|
expected_cmd = [
|
|
fake_binary,
|
|
'--cluster', fake_cluster,
|
|
'zonegroup', 'get',
|
|
'--rgw-realm=' + fake_realm,
|
|
'--rgw-zonegroup=' + fake_zonegroup,
|
|
'--format=json'
|
|
]
|
|
|
|
assert radosgw_zone.get_zonegroup(fake_module) == expected_cmd
|
|
|
|
def test_remove_zone(self):
|
|
fake_module = MagicMock()
|
|
fake_module.params = fake_params
|
|
expected_cmd = [
|
|
fake_binary,
|
|
'--cluster', fake_cluster,
|
|
'zone', 'delete',
|
|
'--rgw-realm=' + fake_realm,
|
|
'--rgw-zonegroup=' + fake_zonegroup,
|
|
'--rgw-zone=' + fake_zone
|
|
]
|
|
|
|
assert radosgw_zone.remove_zone(fake_module) == expected_cmd
|