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

import datetime
import time

from tests.consts.config import LAMBDA_NAME


def test_lambda_with_dummy_data(lambda_client, cloudwatch_log_client):
    """Assert lambda action."""
    data = {'key': datetime.datetime.now().isoformat()}
    lambda_client.invoke(LAMBDA_NAME, payload=data, assertion=True)
    time.sleep(5)
    # assert cloudwatch logs for this lambda.
    cloudwatch_log_client.assert_lambda_logs_message(LAMBDA_NAME, 'Unknown event!')


def test_lambda_with_project_insert(lambda_client, cloudwatch_log_client):
    """Assert lambda with real data."""
    data = {
        'Records': [
            {
                'kinesis': {
                    'kinesisSchemaVersion': '1.0',
                    'partitionKey': '5993248',
                    'sequenceNumber': '14946538883691542474157356460932289527906',
                    'data': 'eyJkYXRhYmFzZSI6ImFydF9yZWxhdGlvbnMiLCJ0YWJsZSI6InByb2plY3QiLCJ0eXBlIjoiaW5zZXJ0IiwidHMiOjE3Mjc5MzU4MDEsInhpZCI6MTA4MTU1MSwiY29tbWl0Ijp0cnVlLCJkYXRhIjp7InByb2plY3RfaWQiOjU5OTMyNDgsInByb2plY3RfY29kZSI6ImN5c0pZaFBxbXkwSjdqeCIsInZlbmRvcl9pZCI6NzEyMywic3ViYWNjb3VudF9pZCI6MCwicHJvamVjdF9uYW1lIjoiUHJvamVjdCBjeVRlc3Q1MjIwNDEwNjUwIiwiY3JlYXRlZF9kYXRlX3V0YyI6IjIwMjQtMTAtMDMgMDY6MTA6MDEuMTg4MDUyIiwidXBkYXRlZF9kYXRlX3V0YyI6IjIwMjQtMTAtMDMgMDY6MTA6MDEuMTg4MDUyIiwiY29ycmVsYXRpb25faWQiOiIxZjk0MjM1NC04MTRlLTExZWYtYjQ4My0wYTU4YTlmZWFjMDIuMSIsImFydGlzdF9pZCI6NTc4MTQ0LCJkZXNjcmlwdGlvbiI6bnVsbCwibGFzdF9tb2RpZmllZF9ieSI6MzgyNzEsInVzZXJfdHlwZSI6ImFsdyIsImRlbGV0aW9ucyI6Ik4ifX0=',  # noqa
                    'approximateArrivalTimestamp': 1727935801.45,
                },
                'eventSource': 'aws:kinesis',
                'eventVersion': '1.0',
                'eventID': 'shardId-000000000006:38883691542474157356460932289527906',
                'eventName': 'aws:kinesis:record',
                'invokeIdentityArn': 'arn:aws:iam::437795906767:role/integration-test',
                'awsRegion': 'us-east-1',
                'eventSourceARN': 'arn:aws:kinesis:us-east-1:437795906767:stream/integration-test',
            }
        ]
    }  # noqa
    lambda_client.invoke(LAMBDA_NAME, payload=data, assertion=True)
    time.sleep(5)
    # assert cloudwatch logs for this lambda.
    cloudwatch_log_client.assert_lambda_logs_message(LAMBDA_NAME, 'function "project_insert" took', max_tries=2)