SIENTIAPDE-988
Refactor ingestor lease management and update Docker configurations - Updated CMD in Dockerfile to use the correct application entry point. - Added docker compose command to remove volumes in README. - Removed unnecessary restart policy from ingestor service in docker-compose.yaml. - Refactored manage_leases method in Ingestor class to improve clarity and parameter naming. - Added get_number_of_leases method in IngestorManager to retrieve active leases. - Implemented get_all_leases method in ResourceManager to fetch active leases from Redis. - Enhanced unit tests for lease management in test_ingestor_manager and test_resource_manager.
This commit is contained in:
@@ -183,6 +183,22 @@ def test_get_active_ingestors_empty(ingestor_manager):
|
||||
ingestor_manager.resource_manager.get_all_ingestors.assert_called_once()
|
||||
|
||||
|
||||
def test_get_number_of_leases_success(ingestor_manager):
|
||||
ingestor_manager.resource_manager.get_all_leases = MagicMock(
|
||||
return_value=["lease1", "lease2"])
|
||||
result = ingestor_manager.get_number_of_leases()
|
||||
assert result == 2
|
||||
ingestor_manager.resource_manager.get_all_leases.assert_called_once()
|
||||
|
||||
|
||||
def test_get_number_of_leases_empty(ingestor_manager):
|
||||
ingestor_manager.resource_manager.get_all_leases = MagicMock(
|
||||
return_value=None)
|
||||
result = ingestor_manager.get_number_of_leases()
|
||||
assert result == 0
|
||||
ingestor_manager.resource_manager.get_all_leases.assert_called_once()
|
||||
|
||||
|
||||
def test_get_number_of_slots_success(ingestor_manager):
|
||||
ingestor_manager.resource_manager.get_all_slots = MagicMock(
|
||||
return_value=["slot1", "slot2"])
|
||||
|
||||
Reference in New Issue
Block a user