def test_registry_update(self): self._create_registry(cluster_id='FAKE_ID', check_type='NODE_STATUS_POLLING', interval=60, params={}, engine_id='DEAD_ENGINE') registries = db_api.registry_claim(self.ctx, engine_id='ENGINE_ID') self.assertTrue(registries[0].enabled) db_api.registry_update(self.ctx, 'FAKE_ID', {'enabled': False}) registries = db_api.registry_claim(self.ctx, engine_id='NEW_ENGINE_ID') self.assertFalse(registries[0].enabled)
def test_registry_delete(self): registry = self._create_registry('CLUSTER_ID', check_type='NODE_STATUS_POLLING', interval=60, params={}, engine_id='ENGINE_ID') db_api.registry_delete(self.ctx, 'CLUSTER_ID') self.assertEqual([], db_api.registry_claim(self.ctx, registry.engine_id))
def test_registry_delete(self): registry = self._create_registry('CLUSTER_ID', check_type='NODE_STATUS_POLLING', interval=60, params={}, engine_id='ENGINE_ID') db_api.registry_delete(self.ctx, 'CLUSTER_ID') self.assertEqual([], db_api.registry_claim(self.ctx, registry.engine_id))
def test_registry_claim(self): for i in range(2): cluster_id = 'cluster-%s' % i self._create_registry(cluster_id=cluster_id, check_type='NODE_STATUS_POLLING', interval=60, params={}, engine_id='DEAD_ENGINE') registries = db_api.registry_claim(self.ctx, engine_id='ENGINE_ID') self.assertEqual(2, len(registries)) self.assertEqual('ENGINE_ID', registries[0].engine_id) self.assertEqual('ENGINE_ID', registries[1].engine_id)
def test_registry_claim(self): for i in range(2): cluster_id = 'cluster-%s' % i self._create_registry(cluster_id=cluster_id, check_type='NODE_STATUS_POLLING', interval=60, params={}, engine_id='DEAD_ENGINE') registries = db_api.registry_claim(self.ctx, engine_id='ENGINE_ID') self.assertEqual(2, len(registries)) self.assertEqual('ENGINE_ID', registries[0].engine_id) self.assertEqual('ENGINE_ID', registries[1].engine_id)
def test_registry_create_get(self): registry = self._create_registry(cluster_id='CLUSTER_ID', check_type='NODE_STATUS_POLLING', interval=60, params={}, engine_id='ENGINE_ID') ret_registries = db_api.registry_claim(self.ctx, registry.engine_id) self.assertEqual(1, len(ret_registries)) ret_registry = ret_registries[0] self.assertEqual(registry.id, ret_registry.id) self.assertEqual(registry.cluster_id, ret_registry.cluster_id) self.assertEqual(registry.check_type, ret_registry.check_type) self.assertEqual(registry.interval, ret_registry.interval) self.assertEqual(registry.params, ret_registry.params) self.assertEqual(registry.engine_id, ret_registry.engine_id)
def test_registry_create_get(self): registry = self._create_registry(cluster_id='CLUSTER_ID', check_type='NODE_STATUS_POLLING', interval=60, params={}, engine_id='ENGINE_ID') ret_registries = db_api.registry_claim(self.ctx, registry.engine_id) self.assertEqual(1, len(ret_registries)) ret_registry = ret_registries[0] self.assertEqual(registry.id, ret_registry.id) self.assertEqual(registry.cluster_id, ret_registry.cluster_id) self.assertEqual(registry.check_type, ret_registry.check_type) self.assertEqual(registry.interval, ret_registry.interval) self.assertEqual(registry.params, ret_registry.params) self.assertEqual(registry.engine_id, ret_registry.engine_id)
def test_registry_claim_with_dead_engine(self, mock_check): db_api.service_create('SERVICE_ID_DEAD') self._create_registry( cluster_id='CLUSTER_1', check_type='NODE_STATUS_POLLING', interval=60, params={}, engine_id='SERVICE_ID') self._create_registry( cluster_id='CLUSTER_1', check_type='NODE_STATUS_POLLING', interval=60, params={}, engine_id='SERVICE_ID_DEAD') mock_check.side_effect = [False, True] registries = db_api.registry_claim(self.ctx, engine_id='ENGINE_ID') self.assertEqual(1, len(registries)) self.assertEqual('SERVICE_ID_DEAD', registries[0].engine_id)