import json
from http import HTTPMethod, HTTPStatus

from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair

from ..conftest import HttpApiProxy, LambdaContext


def test_get_policies(
    mock_app,
    dynamodb_seeds,
    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = mock_app.lambda_handler(
        http_api_proxy(
            raw_path='/orgs/cJtK9SsnJhKPyxESe7g3DG/policies',
            method=HTTPMethod.GET,
            headers={'X-Tenant': '*'},
        ),
        lambda_context,
    )
    assert r['statusCode'] == HTTPStatus.OK
    assert json.loads(r['body']) == {
        'billing_policy': {'billing_day': 1, 'payment_method': 'PIX'},
        'payment_policy': {'due_days': 90},
    }


def test_put_policies(
    mock_app,
    dynamodb_seeds,
    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = mock_app.lambda_handler(
        http_api_proxy(
            raw_path='/orgs/cJtK9SsnJhKPyxESe7g3DG/policies',
            method=HTTPMethod.PUT,
            headers={'X-Tenant': '*'},
            body={'payment_policy': None},
        ),
        lambda_context,
    )
    assert r['statusCode'] == HTTPStatus.OK

    course = dynamodb_persistence_layer.collection.get_item(
        KeyPair('cJtK9SsnJhKPyxESe7g3DG', '0')
    )
    assert course['name'] == 'EDUSEG'


def test_get_address(
    mock_app,
    dynamodb_seeds,
    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = mock_app.lambda_handler(
        http_api_proxy(
            raw_path='/orgs/cJtK9SsnJhKPyxESe7g3DG/address',
            method=HTTPMethod.GET,
            body={'payment_policy': None},
        ),
        lambda_context,
    )
    address = {
        'id': 'cJtK9SsnJhKPyxESe7g3DG',
        'sk': 'metadata#address',
        'postcode': '88101001',
        'address1': 'Av. Presidente Kennedy, 815',
        'address2': 'Sala 1',
        'neighborhood': 'Campinas',
        'city': 'São José',
        'state': 'SC',
    }
    assert r['statusCode'] == HTTPStatus.OK
    assert json.loads(r['body']) == address


def test_post_address(
    mock_app,
    dynamodb_seeds,
    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = mock_app.lambda_handler(
        http_api_proxy(
            raw_path='/orgs/cJtK9SsnJhKPyxESe7g3DG/address',
            method=HTTPMethod.POST,
            body={
                'postcode': '81280350',
                'address1': 'Rua Monsenhor Ivo Zanlorenzi, 5190',
                'address2': 'ap 1802',
                'neighborhood': 'Cidade Industrial',
                'city': 'Curitiba',
                'state': 'PR',
            },
        ),
        lambda_context,
    )
    assert r['statusCode'] == HTTPStatus.OK

    data = dynamodb_persistence_layer.collection.get_item(
        KeyPair('cJtK9SsnJhKPyxESe7g3DG', 'metadata#address')
    )
    assert data['address1'] == 'Rua Monsenhor Ivo Zanlorenzi, 5190'