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)
