ceph-ansible/library/test_ceph_crush.py

91 lines
3.5 KiB
Python

from . import ceph_crush
import pytest
class TestCephCrushModule(object):
def test_no_host(self):
location = [
("chassis", "monchassis"),
("rack", "monrack"),
("row", "marow"),
("pdu", "monpdu"),
("pod", "monpod"),
("room", "maroom"),
("datacenter", "mondc"),
("region", "maregion"),
("root", "maroute"),
]
with pytest.raises(Exception):
result = ceph_crush.sort_osd_crush_location(location, None)
def test_lower_than_two_bucket(self):
location = [
("chassis", "monchassis"),
]
with pytest.raises(Exception):
result = ceph_crush.sort_osd_crush_location(location, None)
def test_invalid_bucket_type(self):
location = [
("host", "monhost"),
("chassis", "monchassis"),
("rackyyyyy", "monrack"),
]
with pytest.raises(Exception):
result = ceph_crush.sort_osd_crush_location(location, None)
def test_ordering(self):
expected_result = [
("host", "monhost"),
("chassis", "monchassis"),
("rack", "monrack"),
("row", "marow"),
("pdu", "monpdu"),
("pod", "monpod"),
("room", "maroom"),
("datacenter", "mondc"),
("region", "maregion"),
("root", "maroute"),
]
expected_result_reverse = expected_result[::-1]
result = ceph_crush.sort_osd_crush_location(expected_result_reverse, None)
assert expected_result == result
def test_generate_commands(self):
cluster = "test"
expected_command_list = [
['ceph', '--cluster', cluster, 'osd', 'crush', "add-bucket", "monhost", "host"],
['ceph', '--cluster', cluster, 'osd', 'crush', "add-bucket", "monchassis", "chassis"],
['ceph', '--cluster', cluster, 'osd', 'crush', "move", "monhost", "chassis=monchassis"],
['ceph', '--cluster', cluster, 'osd', 'crush', "add-bucket", "monrack", "rack"],
['ceph', '--cluster', cluster, 'osd', 'crush', "move", "monchassis", "rack=monrack"],
]
location = [
("host", "monhost"),
("chassis", "monchassis"),
("rack", "monrack"),
]
result = ceph_crush.create_and_move_buckets_list(cluster, location)
assert result == expected_command_list
def test_generate_commands_container(self):
cluster = "test"
containerized = "docker exec -ti ceph-mon"
expected_command_list = [
['docker', 'exec', '-ti', 'ceph-mon', 'ceph', '--cluster', cluster, 'osd', 'crush', "add-bucket", "monhost", "host"],
['docker', 'exec', '-ti', 'ceph-mon', 'ceph', '--cluster', cluster, 'osd', 'crush', "add-bucket", "monchassis", "chassis"],
['docker', 'exec', '-ti', 'ceph-mon', 'ceph', '--cluster', cluster, 'osd', 'crush', "move", "monhost", "chassis=monchassis"],
['docker', 'exec', '-ti', 'ceph-mon', 'ceph', '--cluster', cluster, 'osd', 'crush', "add-bucket", "monrack", "rack"],
['docker', 'exec', '-ti', 'ceph-mon', 'ceph', '--cluster', cluster, 'osd', 'crush', "move", "monchassis", "rack=monrack"],
]
location = [
("host", "monhost"),
("chassis", "monchassis"),
("rack", "monrack"),
]
result = ceph_crush.create_and_move_buckets_list(cluster, location, containerized)
assert result == expected_command_list