def get_delete_pool_flow_internal(self, name): """Create a flow to delete a pool, etc. :returns: The flow for deleting a pool """ delete_pool_flow = linear_flow.Flow(constants.DELETE_POOL_FLOW) # health monitor should cascade # members should cascade delete_pool_flow.add(database_tasks.MarkPoolPendingDeleteInDB( name='mark_pool_pending_delete_in_db_' + name, requires=constants.POOL, rebind={constants.POOL: name})) delete_pool_flow.add(database_tasks.CountPoolChildrenForQuota( name='count_pool_children_for_quota_' + name, requires=constants.POOL, provides=constants.POOL_CHILD_COUNT, rebind={constants.POOL: name})) delete_pool_flow.add(model_tasks.DeleteModelObject( name='delete_model_object_' + name, rebind={constants.OBJECT: name})) delete_pool_flow.add(database_tasks.DeletePoolInDB( name='delete_pool_in_db_' + name, requires=constants.POOL, rebind={constants.POOL: name})) delete_pool_flow.add(database_tasks.DecrementPoolQuota( name='decrement_pool_quota_' + name, requires=[constants.POOL, constants.POOL_CHILD_COUNT], rebind={constants.POOL: name})) return delete_pool_flow
def get_delete_pool_flow(self): """Create a flow to delete a pool :returns: The flow for deleting a pool """ delete_pool_flow = linear_flow.Flow(constants.DELETE_POOL_FLOW) delete_pool_flow.add(lifecycle_tasks.PoolToErrorOnRevertTask( requires=[constants.POOL, constants.LISTENERS, constants.LOADBALANCER])) delete_pool_flow.add(database_tasks.MarkPoolPendingDeleteInDB( requires=constants.POOL)) delete_pool_flow.add(database_tasks.CountPoolChildrenForQuota( requires=constants.POOL, provides=constants.POOL_CHILD_COUNT)) delete_pool_flow.add(model_tasks.DeleteModelObject( rebind={constants.OBJECT: constants.POOL})) delete_pool_flow.add(amphora_driver_tasks.ListenersUpdate( requires=[constants.LOADBALANCER, constants.LISTENERS])) delete_pool_flow.add(database_tasks.DeletePoolInDB( requires=constants.POOL)) delete_pool_flow.add(database_tasks.DecrementPoolQuota( requires=[constants.POOL, constants.POOL_CHILD_COUNT])) delete_pool_flow.add(database_tasks.MarkLBAndListenersActiveInDB( requires=[constants.LOADBALANCER, constants.LISTENERS])) return delete_pool_flow
def get_delete_pool_flow_internal(self, pool_id): """Create a flow to delete a pool, etc. :returns: The flow for deleting a pool """ delete_pool_flow = linear_flow.Flow(constants.DELETE_POOL_FLOW + '-' + pool_id) # health monitor should cascade # members should cascade delete_pool_flow.add(database_tasks.MarkPoolPendingDeleteInDB( name='mark_pool_pending_delete_in_db_' + pool_id, requires=constants.POOL_ID, inject={constants.POOL_ID: pool_id})) delete_pool_flow.add(database_tasks.CountPoolChildrenForQuota( name='count_pool_children_for_quota_' + pool_id, requires=constants.POOL_ID, provides=constants.POOL_CHILD_COUNT, inject={constants.POOL_ID: pool_id})) delete_pool_flow.add(database_tasks.DeletePoolInDB( name='delete_pool_in_db_' + pool_id, requires=constants.POOL_ID, inject={constants.POOL_ID: pool_id})) delete_pool_flow.add(database_tasks.DecrementPoolQuota( name='decrement_pool_quota_' + pool_id, requires=[constants.PROJECT_ID, constants.POOL_CHILD_COUNT])) return delete_pool_flow
def get_delete_pool_flow(self): """Create a flow to delete a pool :returns: The flow for deleting a pool """ delete_pool_flow = linear_flow.Flow(constants.DELETE_POOL_FLOW) delete_pool_flow.add( lifecycle_tasks.PoolToErrorOnRevertTask(requires=[ constants.POOL, constants.LISTENERS, constants.LOADBALANCER ])) delete_pool_flow.add( database_tasks.MarkPoolPendingDeleteInDB(requires=constants.POOL)) delete_pool_flow.add( database_tasks.CountPoolChildrenForQuota( requires=constants.POOL, provides=constants.POOL_CHILD_COUNT)) delete_pool_flow.add( model_tasks.DeleteModelObject( rebind={constants.OBJECT: constants.POOL})) # Get VThunder details from database delete_pool_flow.add( a10_database_tasks.GetVThunderByLoadBalancer( requires=constants.LOADBALANCER, provides=a10constants.VTHUNDER)) delete_pool_flow.add( handler_virtual_port.ListenersUpdate(requires=[ constants.LOADBALANCER, constants.LISTENERS, a10constants.VTHUNDER ])) delete_pool_flow.add( handler_service_group.PoolDelete( requires=[constants.POOL, a10constants.VTHUNDER])) delete_pool_flow.add( database_tasks.DeletePoolInDB(requires=constants.POOL)) delete_pool_flow.add( database_tasks.DecrementPoolQuota( requires=[constants.POOL, constants.POOL_CHILD_COUNT])) delete_pool_flow.add( database_tasks.MarkLBAndListenersActiveInDB( requires=[constants.LOADBALANCER, constants.LISTENERS])) return delete_pool_flow
def test_decrement_pool_quota_pool_children(self, mock_check_quota_met, mock_decrement_quota): pool_child_count = {'HM': 1, 'member': 2} project_id = uuidutils.generate_uuid() task = database_tasks.DecrementPoolQuota() mock_session = mock.MagicMock() with mock.patch('octavia.db.api.' 'get_session') as mock_get_session_local: mock_get_session_local.return_value = mock_session task.execute(project_id, pool_child_count) calls = [mock.call(mock_session, data_models.Pool, project_id), mock.call(mock_session, data_models.HealthMonitor, project_id), mock.call(mock_session, data_models.Member, project_id, quantity=2)] mock_decrement_quota.assert_has_calls(calls) mock_session.commit.assert_called_once_with() # revert mock_session.reset_mock() with mock.patch('octavia.db.api.' 'get_session') as mock_get_session_local: mock_lock_session = mock.MagicMock() mock_get_session_local.side_effect = [mock_session, mock_lock_session, mock_lock_session, mock_lock_session, mock_lock_session] task.revert(project_id, pool_child_count, None) calls = [mock.call(mock_session, mock_lock_session, data_models.Pool, project_id), mock.call(mock_session, mock_lock_session, data_models.HealthMonitor, project_id), mock.call(mock_session, mock_lock_session, data_models.Member, project_id), mock.call(mock_session, mock_lock_session, data_models.Member, project_id)] mock_check_quota_met.assert_has_calls(calls) self.assertEqual(4, mock_lock_session.commit.call_count) # revert with health monitor quota exception mock_session.reset_mock() mock_check_quota_met.side_effect = [None, Exception('fail'), None, None] with mock.patch('octavia.db.api.' 'get_session') as mock_get_session_local: mock_lock_session = mock.MagicMock() mock_get_session_local.side_effect = [mock_session, mock_lock_session, mock_lock_session, mock_lock_session, mock_lock_session] task.revert(project_id, pool_child_count, None) calls = [mock.call(mock_session, mock_lock_session, data_models.Pool, project_id), mock.call(mock_session, mock_lock_session, data_models.HealthMonitor, project_id), mock.call(mock_session, mock_lock_session, data_models.Member, project_id), mock.call(mock_session, mock_lock_session, data_models.Member, project_id)] mock_check_quota_met.assert_has_calls(calls) self.assertEqual(3, mock_lock_session.commit.call_count) self.assertEqual(1, mock_lock_session.rollback.call_count) # revert with member quota exception mock_session.reset_mock() mock_check_quota_met.side_effect = [None, None, None, Exception('fail')] with mock.patch('octavia.db.api.' 'get_session') as mock_get_session_local: mock_lock_session = mock.MagicMock() mock_get_session_local.side_effect = [mock_session, mock_lock_session, mock_lock_session, mock_lock_session, mock_lock_session] task.revert(project_id, pool_child_count, None) calls = [mock.call(mock_session, mock_lock_session, data_models.Pool, project_id), mock.call(mock_session, mock_lock_session, data_models.HealthMonitor, project_id), mock.call(mock_session, mock_lock_session, data_models.Member, project_id), mock.call(mock_session, mock_lock_session, data_models.Member, project_id)] mock_check_quota_met.assert_has_calls(calls) self.assertEqual(3, mock_lock_session.commit.call_count) self.assertEqual(1, mock_lock_session.rollback.call_count)
def test_decrement_pool_quota(self): project_id = uuidutils.generate_uuid() task = database_tasks.DecrementPoolQuota() data_model = data_models.Pool self._test_decrement_quota(task, data_model, project_id=project_id)
def test_decrement_pool_quota(self): task = database_tasks.DecrementPoolQuota() data_model = data_models.Pool self._test_decrement_quota(task, data_model)