import json
from http import HTTPMethod, HTTPStatus

from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, PartitionKey

from ...conftest import HttpApiProxy, LambdaContext

# Check the seeds, if necessary.
org_id = '2a0f83b6-9d72-4fc0-952c-acbcfba39016'
seat = {
    'id': '389a282f-0a1e-4c9e-9502-d3131b1c2e57',
    'user': {
        'id': '15bacf02-1535-4bee-9022-19d106fd7518',
        'name': 'Eddie Vedder',
        'email': 'eddie@pearljam.band',
        'cpf': '07879819908',
    },
    'course': {
        'id': 'c27d1b4f-575c-4b6b-82a1-9b91ff369e0b',
        'name': 'NR-18 PEMT Plataforma Móvel de Trabalho Aéreo',
        'access_period': '360',
        'unit_price': '149',
    },
    'seat': {'order_id': 'c556e2f2-f65b-4959-ad04-e789de107ac5'},
}


def test_enroll_from_subscription(
    app,
    seeds,
    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = app.lambda_handler(
        http_api_proxy(
            raw_path='/enrollments',
            method=HTTPMethod.POST,
            body={
                'org_id': org_id,
                'subscription': {
                    'billing_day': 6,
                },
                'enrollments': [
                    {
                        'id': '44ff9ac1-a7cd-447b-a284-53cdc5929d7f',
                        'user': {
                            'id': '15bacf02-1535-4bee-9022-19d106fd7518',
                            'name': 'Eddie Vedder',
                            'email': 'eddie@pearljam.band',
                            'cpf': '07879819908',
                        },
                        'course': {
                            'id': 'c27d1b4f-575c-4b6b-82a1-9b91ff369e0b',
                            'name': 'NR-18 PEMT Plataforma Móvel de Trabalho Aéreo',
                            'access_period': '360',
                            'unit_price': '149',
                        },
                        'scheduled_for': '2028-01-01',
                    },
                    {
                        'id': 'd0349bbe-cef3-44f7-b20e-3cb4476ab4c5',
                        'user': {
                            'id': '15bacf02-1535-4bee-9022-19d106fd7518',
                            'name': 'Sérgio R Siqueira',
                            'email': 'sergio@somosbeta.com.br',
                            'cpf': '07879819908',
                        },
                        'course': {
                            'id': '9b1bd8e1-b6da-4f68-9a83-c8d5b8f3b628',
                            'name': 'CIPA',
                            'access_period': '360',
                            'unit_price': '99',
                        },
                        'deduplication_window': {
                            'offset_days': '45',
                        },
                    },
                ],
            },
        ),
        lambda_context,
    )
    assert r['statusCode'] == HTTPStatus.OK
    body = json.loads(r['body'])
    submission = dynamodb_persistence_layer.get_item(KeyPair(body['id'], body['sk']))
    assert submission['sk'] == body['sk']

    enrolled = dynamodb_persistence_layer.collection.query(
        PartitionKey('d0349bbe-cef3-44f7-b20e-3cb4476ab4c5')
    )

    assert len(enrolled['items']) == 7

    scheduled = dynamodb_persistence_layer.collection.query(
        PartitionKey(f'SCHEDULED#ORG#{org_id}')
    )

    assert len(scheduled['items']) == 1


def test_enroll_for_from_seats(
    app,
    seeds,
    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = app.lambda_handler(
        http_api_proxy(
            raw_path='/enrollments',
            method=HTTPMethod.POST,
            body={
                'org_id': org_id,
                'enrollments': [{**seat}],
            },
        ),
        lambda_context,
    )
    body = json.loads(r['body'])
    assert r['statusCode'] == HTTPStatus.OK
    item = body['enrolled'][0]
    assert item['status'] == 'success'

    r = dynamodb_persistence_layer.collection.get_item(
        KeyPair(
            'c556e2f2-f65b-4959-ad04-e789de107ac5',
            'ENROLLMENT#389a282f-0a1e-4c9e-9502-d3131b1c2e57',
        )
    )
    assert r['status'] == 'EXECUTED'


def test_schedule_for_from_seats(
    app,
    seeds,
    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = app.lambda_handler(
        http_api_proxy(
            raw_path='/enrollments',
            method=HTTPMethod.POST,
            body={
                'org_id': org_id,
                'enrollments': [
                    {
                        **seat,
                        'scheduled_for': '2028-01-01',
                    }
                ],
            },
        ),
        lambda_context,
    )
    body = json.loads(r['body'])
    assert r['statusCode'] == HTTPStatus.OK

    item = body['scheduled'][0]
    assert item['status'] == 'success'

    r = dynamodb_persistence_layer.collection.get_item(
        KeyPair(
            'c556e2f2-f65b-4959-ad04-e789de107ac5',
            'ENROLLMENT#389a282f-0a1e-4c9e-9502-d3131b1c2e57',
        )
    )
    assert r['status'] == 'SCHEDULED'