apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-06-25 08:48:13 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat: limit campaign concurrency to number of CLIs

Browse source
This commit is contained in:
Abhishek Kumar 2026-02-07 13:45:21 +05:30
parent 6711dcb3ea
commit 3cdede0f45
18 changed files with 846 additions and 462 deletions
Download patch file Download diff file Expand all files Collapse all files

114
api/tests/test_campaign_call_dispatcher.py
View file

@ -195,6 +195,24 @@ def mock_rate_limiter():
async def mock_delete_mapping(*args, **kwargs):
pass
async def mock_initialize_from_number_pool(*args, **kwargs):
return True
async def mock_acquire_from_number(*args, **kwargs):
return "+15551234567"
async def mock_release_from_number(*args, **kwargs):
return True
async def mock_store_from_number_mapping(*args, **kwargs):
return True
async def mock_get_from_number_mapping(*args, **kwargs):
return None
async def mock_delete_from_number_mapping(*args, **kwargs):
return True
return {
"acquire_token": mock_acquire_token,
"try_acquire_concurrent_slot": mock_try_acquire_slot,
@ -202,6 +220,12 @@ def mock_rate_limiter():
"store_workflow_slot_mapping": mock_store_mapping,
"get_workflow_slot_mapping": mock_get_mapping,
"delete_workflow_slot_mapping": mock_delete_mapping,
"initialize_from_number_pool": mock_initialize_from_number_pool,
"acquire_from_number": mock_acquire_from_number,
"release_from_number": mock_release_from_number,
"store_workflow_from_number_mapping": mock_store_from_number_mapping,
"get_workflow_from_number_mapping": mock_get_from_number_mapping,
"delete_workflow_from_number_mapping": mock_delete_from_number_mapping,
}
@ -242,6 +266,24 @@ class TestProcessBatchBasic:
mock_rl.delete_workflow_slot_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_slot_mapping"]
)
mock_rl.initialize_from_number_pool = AsyncMock(
side_effect=mock_rate_limiter["initialize_from_number_pool"]
)
mock_rl.acquire_from_number = AsyncMock(
side_effect=mock_rate_limiter["acquire_from_number"]
)
mock_rl.release_from_number = AsyncMock(
side_effect=mock_rate_limiter["release_from_number"]
)
mock_rl.store_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["store_workflow_from_number_mapping"]
)
mock_rl.get_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["get_workflow_from_number_mapping"]
)
mock_rl.delete_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_from_number_mapping"]
)
dispatcher = CampaignCallDispatcher()
@ -307,6 +349,24 @@ class TestProcessBatchConcurrency:
mock_rl.delete_workflow_slot_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_slot_mapping"]
)
mock_rl.initialize_from_number_pool = AsyncMock(
side_effect=mock_rate_limiter["initialize_from_number_pool"]
)
mock_rl.acquire_from_number = AsyncMock(
side_effect=mock_rate_limiter["acquire_from_number"]
)
mock_rl.release_from_number = AsyncMock(
side_effect=mock_rate_limiter["release_from_number"]
)
mock_rl.store_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["store_workflow_from_number_mapping"]
)
mock_rl.get_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["get_workflow_from_number_mapping"]
)
mock_rl.delete_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_from_number_mapping"]
)
dispatcher = CampaignCallDispatcher()
@ -379,6 +439,24 @@ class TestProcessBatchConcurrency:
mock_rl.delete_workflow_slot_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_slot_mapping"]
)
mock_rl.initialize_from_number_pool = AsyncMock(
side_effect=mock_rate_limiter["initialize_from_number_pool"]
)
mock_rl.acquire_from_number = AsyncMock(
side_effect=mock_rate_limiter["acquire_from_number"]
)
mock_rl.release_from_number = AsyncMock(
side_effect=mock_rate_limiter["release_from_number"]
)
mock_rl.store_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["store_workflow_from_number_mapping"]
)
mock_rl.get_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["get_workflow_from_number_mapping"]
)
mock_rl.delete_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_from_number_mapping"]
)
dispatcher = CampaignCallDispatcher()
@ -450,6 +528,24 @@ class TestProcessBatchConcurrency:
mock_rl.delete_workflow_slot_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_slot_mapping"]
)
mock_rl.initialize_from_number_pool = AsyncMock(
side_effect=mock_rate_limiter["initialize_from_number_pool"]
)
mock_rl.acquire_from_number = AsyncMock(
side_effect=mock_rate_limiter["acquire_from_number"]
)
mock_rl.release_from_number = AsyncMock(
side_effect=mock_rate_limiter["release_from_number"]
)
mock_rl.store_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["store_workflow_from_number_mapping"]
)
mock_rl.get_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["get_workflow_from_number_mapping"]
)
mock_rl.delete_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_from_number_mapping"]
)
dispatcher = CampaignCallDispatcher()
@ -513,6 +609,24 @@ class TestProcessBatchConcurrency:
mock_rl.delete_workflow_slot_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_slot_mapping"]
)
mock_rl.initialize_from_number_pool = AsyncMock(
side_effect=mock_rate_limiter["initialize_from_number_pool"]
)
mock_rl.acquire_from_number = AsyncMock(
side_effect=mock_rate_limiter["acquire_from_number"]
)
mock_rl.release_from_number = AsyncMock(
side_effect=mock_rate_limiter["release_from_number"]
)
mock_rl.store_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["store_workflow_from_number_mapping"]
)
mock_rl.get_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["get_workflow_from_number_mapping"]
)
mock_rl.delete_workflow_from_number_mapping = AsyncMock(
side_effect=mock_rate_limiter["delete_workflow_from_number_mapping"]
)
dispatcher = CampaignCallDispatcher()