"""Integration tests for lambda function."""

import os

import pytest

from tests.integration.conftest import dockerized_lambda_api_client, table_has_records


@pytest.mark.jira('ACC-8900')
def test_invoke_lambda(basic_headers, snowflake_conn):
    """Assert lambda action."""
    dockerized_lambda = dockerized_lambda_api_client(basic_headers)

    assert not table_has_records(
        snowflake_conn, os.environ['SNOWFLAKE_DATABASE'], os.environ['SNOWFLAKE_SCHEMA']
    )

    body = {
        'abacus_event_id': 1,
        'target_type': 'statement_period',
        'target_id': '318',
        'event_name': 'snapshot_account_contracts',
    }

    res = dockerized_lambda.post_lambda('2015-03-31', body)
    assert res.status_code == 200
    assert 'status' in res.json(), (
        f'error type was: {res.json()["errorType"]}, '
        f'error message was: {res.json()["errorMessage"]}'
    )
    assert res.json()['status'] == 'OK'

    assert table_has_records(
        snowflake_conn, os.environ['SNOWFLAKE_DATABASE'], os.environ['SNOWFLAKE_SCHEMA']
    )