def test_pool_mirroring_mode(self):
self.patch_object(ceph_rbd_mirror.ch_ceph, 'CephBrokerRq')
class FakeCephBrokerRq(object):
def __init__(self, raw_request_data=None):
request_data = json.loads(raw_request_data)
self.api_version = request_data['api-version']
self.request_id = request_data['request-id']
self.set_ops(request_data['ops'])
def set_ops(self, ops):
self.ops = ops
def add_op(self, op):
self.ops.append(op)
self.CephBrokerRq.side_effect = FakeCephBrokerRq
brq1_data = json.dumps({
'api-version':
1,
'request-id':
'broker_rq1',
'ops': [
{
'op': 'create-pool',
'name': 'pool-rq0',
'app-name': 'rbd-pool',
'rbd-mirroring-mode': 'pool'
},
]
})
brq2_data = json.dumps({
'api-version':
1,
'request-id':
'broker_rq2',
'ops': [
{
'op': 'create-pool',
'name': 'pool-rq1',
'app-name': 'rbd-image',
'rbd-mirroring-mode': 'image'
},
]
})
brq1 = self.CephBrokerRq(raw_request_data=brq1_data)
brq2 = self.CephBrokerRq(raw_request_data=brq2_data)
broker_requests = [brq1, brq2, None]
crmc = ceph_rbd_mirror.CephRBDMirrorCharm()
rq0 = crmc.pool_mirroring_mode('pool-rq0', broker_requests)
self.assertEqual('pool', rq0)
rq1 = crmc.pool_mirroring_mode('pool-rq1', broker_requests)
self.assertEqual('image', rq1)
def test__mirror_pool_info(self):
self.patch_object(ceph_rbd_mirror.socket, 'gethostname')
self.patch_object(ceph_rbd_mirror.subprocess, 'check_output')
self.gethostname.return_value = 'ahostname'
self.check_output.return_value = '{}'
crmc = ceph_rbd_mirror.CephRBDMirrorCharm()
crmc._mirror_pool_info('apool')
self.check_output.assert_called_once_with([
'rbd', '--id', 'rbd-mirror.ahostname', 'mirror', 'pool', 'info',
'--format', 'json', 'apool'
],
universal_newlines=True)
def test_custom_assess_status_check(self):
self.patch_object(ceph_rbd_mirror.socket, 'gethostname')
self.patch_object(ceph_rbd_mirror.reactive, 'is_flag_set')
self.is_flag_set.return_value = False
crmc = ceph_rbd_mirror.CephRBDMirrorCharm()
self.assertEqual(crmc.custom_assess_status_check(), (None, None))
self.is_flag_set.return_value = True
self.patch_object(ceph_rbd_mirror.reactive, 'endpoint_from_flag')
self.assertEqual(crmc.custom_assess_status_check(),
('waiting', 'Waiting for pools to be created'))
self.endpoint_from_flag.assert_called_once_with('ceph-local.available')
crmc.mirror_pools_summary = mock.MagicMock()
crmc.mirror_pools_summary.return_value = collections.OrderedDict({
'pool_health':
collections.OrderedDict({
'OK': 1,
'WARN': 1,
'ERROR': 1
}),
'image_states':
collections.OrderedDict({
'stopped': 2,
'replaying': 2
}),
})
result = crmc.custom_assess_status_check()
self.assertTrue('blocked' in result[0])
# the order of which the statuses appear in the string is undefined
self.assertTrue('OK (1)' in result[1])
self.assertTrue('WARN (1)' in result[1])
self.assertTrue('ERROR (1)' in result[1])
self.assertTrue('Primary (2)' in result[1])
self.assertTrue('Secondary (2)' in result[1])
crmc.mirror_pools_summary.return_value = collections.OrderedDict({
'pool_health':
collections.OrderedDict({'OK': 1}),
'image_states':
collections.OrderedDict({'stopped': 2}),
})
self.assertEqual(crmc.custom_assess_status_check(),
('active', 'Unit is ready (Pools OK (1) '
'Images Primary (2))'))
crmc.mirror_pools_summary.side_effect = subprocess.CalledProcessError(
42, [])
self.assertEqual(crmc.custom_assess_status_check(), (None, None))
def test_mirror_pool_enabled(self):
self.patch_object(ceph_rbd_mirror.socket, 'gethostname')
crmc = ceph_rbd_mirror.CephRBDMirrorCharm()
_mirror_pool_info = mock.MagicMock()
_mirror_pool_info.return_value = {
'mode':
'pool',
'peers': [{
'uuid':
'0e4dfe58-93fc-44f8-8c74-7e700f950118',
'cluster_name':
'remote',
'client_name':
'client.rbd-mirror.juju-c50b1a-zaza-4ce96f1e7e43-12'
}]
}
crmc._mirror_pool_info = _mirror_pool_info
self.assertTrue(crmc.mirror_pool_enabled('apool', mode='pool'))
_mirror_pool_info.assert_called_once_with('apool')
_mirror_pool_info.return_value = {'mode': 'disabled'}
self.assertFalse(crmc.mirror_pool_enabled('apool', mode='pool'))
def test_collapse_and_filter_broker_requests(self):
self.patch_object(ceph_rbd_mirror.ch_ceph, 'CephBrokerRq')
class FakeCephBrokerRq(object):
def __init__(self):
self.ops = []
def add_op(self, op):
self.ops.append(op)
self.CephBrokerRq.side_effect = FakeCephBrokerRq
broker_requests = [
{
'api-version':
1,
'ops': [
{
'op': 'create-pool',
'name': 'pool-rq0',
'app-name': 'rbd',
},
]
},
{
'api-version':
1,
'ops': [
{
'op': 'create-pool',
'name': 'pool-rq1',
'app-name': 'notrbd',
},
]
},
{
'api-version':
1,
'ops': [
{
'op': 'create-pool',
'name': 'pool-rq2',
'app-name': 'rbd',
'someotherkey': 'value',
},
]
},
]
crmc = ceph_rbd_mirror.CephRBDMirrorCharm()
rq = crmc.collapse_and_filter_broker_requests(
broker_requests,
set(('create-pool', )),
require_vp={'app-name': 'rbd'})
self.assertDictEqual(rq.ops[0], {
'app-name': 'rbd',
'name': 'pool-rq0',
'op': 'create-pool'
})
self.assertDictEqual(
rq.ops[1], {
'app-name': 'rbd',
'name': 'pool-rq2',
'op': 'create-pool',
'someotherkey': 'value'
})
self.assertTrue(len(rq.ops) == 2)
rq = crmc.collapse_and_filter_broker_requests(broker_requests,
set(('create-pool', )),
require_vp={
'app-name': 'rbd',
'someotherkey':
'value'
})
self.assertDictEqual(
rq.ops[0], {
'app-name': 'rbd',
'name': 'pool-rq2',
'op': 'create-pool',
'someotherkey': 'value'
})
self.assertTrue(len(rq.ops) == 1)
def test_pools_in_broker_request(self):
rq = mock.MagicMock()
rq.api_version = 1
rq.ops = [{'op': 'create-pool', 'name': 'fakepool'}]
crmc = ceph_rbd_mirror.CephRBDMirrorCharm()
self.assertIn('fakepool', crmc.pools_in_broker_request(rq))