class TestOSDFactory(TestCase):
salt_local_client = MagicMock(run_job=MagicMock())
salt_local_client.return_value = salt_local_client
salt_local_client.run_job.return_value = {'jid': 12345}
def setUp(self):
fake_cluster_monitor = MagicMock()
attributes = {'name': 'I am a fake',
'fsid': 12345,
'get_sync_object.return_value': fake_cluster_monitor,
'osds_by_id': {0: {'up': True}, 1: {'up': False}}}
fake_cluster_monitor.configure_mock(**attributes)
self.osd_request_factory = OsdRequestFactory(fake_cluster_monitor)
def testCreate(self):
self.assertNotEqual(OsdRequestFactory(0), None, 'Test creating an OSDRequest')
@patch('cthulhu.manager.user_request.LocalClient', salt_local_client)
def testScrub(self):
scrub = self.osd_request_factory.scrub(0)
self.assertIsInstance(scrub, UserRequest, 'Testing Scrub')
scrub.submit(54321)
self.salt_local_client.run_job.assert_called_with(54321, 'ceph.rados_commands', [12345, 'I am a fake', [('osd scrub', {'who': '0'})]])
def testDeepScrub(self):
deep_scrub = self.osd_request_factory.deep_scrub(0)
self.assertIsInstance(deep_scrub, UserRequest, 'Failed to make a deep scrub request')
def test_validate_scrub(self):
self.assertEqual(self.osd_request_factory.get_valid_commands([0]), {0: {'valid_commands': OSD_IMPLEMENTED_COMMANDS}})
def test_validate_scrub_on_down_osd(self):
self.assertEqual(self.osd_request_factory.get_valid_commands([1]), {1: {'valid_commands': []}})
def test_validate_op_key_error(self):
self.assertRaises(KeyError, self.osd_request_factory.get_valid_commands, [2])
