def test_modify_instance_fleet_invalid_fleet(self, aws_client):
fleet_id = 'i-JKEIDBGHSJ34'
mocked_response = mock_client_error(
op='ModifyInstanceFleet',
Code='ValidationException',
Message="Fleet id '%s' is not valid." % fleet_id)
client = MagicMock()
aws_client.return_value = client
client.modify_instance_fleet.side_effect = mocked_response
with pytest.raises(FailedActivity) as e:
modify_instance_fleet(cluster_id=self.cluster_id,
fleet_id=fleet_id,
on_demand_capacity=3,
spot_capacity=2)
assert "Fleet id '%s' is not valid" % fleet_id in str(e.value)
def test_modify_instance_fleet(self, aws_client):
mocked_response = read_configs('list_instance_fleets_1.json')
client = MagicMock()
aws_client.return_value = client
client.list_instance_fleets.return_value = mocked_response
client.modify_instance_fleet.return_value = None
modify_instance_fleet(cluster_id=self.cluster_id,
fleet_id=self.fleet_id,
on_demand_capacity=3,
spot_capacity=2)
client.modify_instance_fleet.assert_called_with(
ClusterId=self.cluster_id,
InstanceFleet={
'InstanceFleetId': self.fleet_id,
'TargetOnDemandCapacity': 3,
'TargetSpotCapacity': 2
})
def test_modify_instance_fleet_no_args(self):
with pytest.raises(FailedActivity) as e:
modify_instance_fleet(cluster_id=self.cluster_id,
fleet_id=self.fleet_id)
assert "Must provide at least one of" in str(e)