def test_pool_manager_init(mock_pool_manager, mock_resource_groups):
assert mock_pool_manager.cluster == 'mesos-test'
assert mock_pool_manager.pool == 'bar'
assert mock_pool_manager.scheduler == 'mesos'
with staticconf.testing.MockConfiguration(
{
'scaling_limits': {
'max_tasks_to_kill': 'inf',
'max_weight_to_add': 100,
'max_weight_to_remove': 100,
'min_capacity': 3,
'max_capacity': 3,
},
},
namespace='bar.mesos_config',
), mock.patch(
'clusterman.aws.spot_fleet_resource_group.SpotFleetResourceGroup.load',
return_value={},
), mock.patch(
'clusterman.autoscaler.pool_manager.DrainingClient',
autospec=True,
), mock.patch(
'clusterman.autoscaler.pool_manager.PoolManager.reload_state'
):
mock_manager = PoolManager('mesos-test', 'bar', 'mesos')
mock_manager.resource_groups = mock_resource_groups
assert mock_manager.max_tasks_to_kill == float('inf')
def mock_pool_manager(mock_resource_groups):
with mock.patch(
'clusterman.aws.spot_fleet_resource_group.SpotFleetResourceGroup.load',
return_value={},
), mock.patch(
'clusterman.autoscaler.pool_manager.DrainingClient',
autospec=True,
), mock.patch(
'clusterman.autoscaler.pool_manager.PoolManager.reload_state'
), mock.patch(
'clusterman.autoscaler.pool_manager.ClusterConnector.load',
):
manager = PoolManager('mesos-test', 'bar', 'mesos')
manager.resource_groups = mock_resource_groups
return manager