def test_aws_lambda_toggle_event_source_mapping_exception(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    client.update_event_source_mapping.side_effect = mock_client_error(
        "update_event_source_mapping")

    with pytest.raises(FailedActivity) as x:
        uuid = "6b08c7db-a0f5-404d-ae73-b116d9125b0e"
        toggle_event_source_mapping_state(uuid, False)
    assert "Test Error" in str(x)
def test_aws_lambda_toggle_event_source_mapping_enable(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    client.update_event_source_mapping.return_value = read_in_test_data(
        "update_event_source_mapping_enable.json")
    uuid = "6b08c7db-a0f5-404d-ae73-b116d9125b0e"
    response = toggle_event_source_mapping_state(event_uuid=uuid, enabled=True)
    assert response["State"] == "Enabling"
    client.update_event_source_mapping.assert_called_with(UUID=uuid,
                                                          Enabled=True)
def test_aws_lambda_toggle_event_source_mapping_disable(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    client.update_event_source_mapping.return_value = read_in_test_data(
        'update_event_source_mapping_disable.json')
    uuid = '6b08c7db-a0f5-404d-ae73-b116d9125b0e'
    response = toggle_event_source_mapping_state(event_uuid=uuid, enabled=False)
    assert response['State'] == 'Disabling'
    client.update_event_source_mapping.assert_called_with(
        UUID=uuid, Enabled=False)