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)