"""Integration tests for lambda function."""
import json


def test_clean(invoke_lambda):
    """Assert lambda action."""
    assert invoke_lambda['StatusCode'] == 200
    response = json.loads(invoke_lambda['Payload'].read())
    assert response == {'has_virus': False}


def test_virus(invoke_lambda_virus):
    """Assert lambda action."""
    assert invoke_lambda_virus['StatusCode'] == 200
    response = json.loads(invoke_lambda_virus['Payload'].read())
    assert response == {'has_virus': True}