saladeaula.digital/api.saladeaula.digital/tests/routes/test_users.py

import json
from http import HTTPMethod, HTTPStatus

from layercake.dynamodb import (
    DynamoDBPersistenceLayer,
    PartitionKey,
    SortKey,
    TransactKey,
)

from ..conftest import HttpApiProxy, LambdaContext


def test_get_emails(
    app,
    seeds,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = app.lambda_handler(
        http_api_proxy(
            raw_path='/users/15bacf02-1535-4bee-9022-19d106fd7518/emails',
            method=HTTPMethod.GET,
        ),
        lambda_context,
    )
    assert r['statusCode'] == HTTPStatus.OK

    body = json.loads(r['body'])
    assert len(body['items']) == 1


def test_get_orgs(
    app,
    seeds,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = app.lambda_handler(
        http_api_proxy(
            raw_path='/users/213a6682-2c59-4404-9189-12eec0a846d4/orgs',
            method=HTTPMethod.GET,
        ),
        lambda_context,
    )
    assert r['statusCode'] == HTTPStatus.OK


def test_add_user(
    app,
    seeds,
    http_api_proxy: HttpApiProxy,
    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
    lambda_context: LambdaContext,
):
    r = app.lambda_handler(
        http_api_proxy(
            raw_path='/users',
            method=HTTPMethod.POST,
            body={
                'user': {
                    'name': 'Scott Weiland',
                    'email': 'scott@stonetemplopilots.com',
                    'cpf': '40245650016',
                },
                'org': {
                    'id': 'f6000f79-6e5c-49a0-952f-3bda330ef278',
                    'name': 'Branco do Brasil',
                    'cnpj': '00000000000191',
                },
            },
        ),
        lambda_context,
    )
    assert r['statusCode'] == HTTPStatus.CREATED
    r = dynamodb_persistence_layer.collection.query(
        PartitionKey('orgmembers#f6000f79-6e5c-49a0-952f-3bda330ef278')
    )
    user_id = r['items'][0]['sk']
    user = dynamodb_persistence_layer.collection.get_items(
        TransactKey(user_id)
        + SortKey('0')
        + SortKey('emails#scott@stonetemplopilots.com')
    )
    assert user['name'] == 'Scott Weiland'
    assert 'email' in user
    assert 'email_verified' in user
    assert 'created_at' in user
    assert 'org_id' in user
    assert 'emails#scott@stonetemplopilots.com' in user


def test_user_exists(
    app,
    seeds,
    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = app.lambda_handler(
        http_api_proxy(
            raw_path='/users',
            method=HTTPMethod.POST,
            body={
                'user': {
                    'name': 'Sérgio R Siqueira',
                    'email': 'sergio@somosbeta.com.br',
                    'cpf': '07879819908',
                },
                'org': {
                    'id': '2a8963fc-4694-4fe2-953a-316d1b10f1f5',
                    'name': 'pytest',
                    'cnpj': '04978826000180',
                },
            },
        ),
        lambda_context,
    )
    assert r['statusCode'] == HTTPStatus.NO_CONTENT

    r = dynamodb_persistence_layer.collection.query(
        PartitionKey('orgmembers#2a8963fc-4694-4fe2-953a-316d1b10f1f5')
    )
    user_id = r['items'][0]['sk']

    user = dynamodb_persistence_layer.collection.get_items(
        TransactKey(user_id)
        + SortKey('0')
        + SortKey('emails#sergio@somosbeta.com.br', rename_key='email')
        + SortKey('orgs#2a8963fc-4694-4fe2-953a-316d1b10f1f5', rename_key='org')
    )
    assert 'mx_record_exists' in user['email']
    assert 'cnpj' in user['org']


def test_user_conflict(
    app,
    seeds,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = app.lambda_handler(
        http_api_proxy(
            raw_path='/users',
            method=HTTPMethod.POST,
            body={
                'user': {
                    'name': 'Sérgio R Siqueira',
                    'email': 'sergio@somosbeta.com.br',
                    'cpf': '07879819908',
                },
                'org': {
                    'id': 'f6000f79-6e5c-49a0-952f-3bda330ef278',
                    'name': 'Banco do Brasil',
                    'cnpj': '00000000000191',
                },
            },
        ),
        lambda_context,
    )
    assert r['statusCode'] == HTTPStatus.CONFLICT


def test_org_not_found(
    app,
    seeds,
    http_api_proxy: HttpApiProxy,
    lambda_context: LambdaContext,
):
    r = app.lambda_handler(
        http_api_proxy(
            raw_path='/users',
            method=HTTPMethod.POST,
            body={
                'user': {
                    'name': 'Sérgio R Siqueira',
                    'email': 'sergio@somosbeta.com.br',
                    'cpf': '07879819908',
                },
                'org': {
                    'id': '123',
                    'name': 'Banco do Brasil',
                    'cnpj': '00000000000191',
                },
            },
        ),
        lambda_context,
    )
    body = json.loads(r['body'])
    assert body['type'] == 'OrgMissingError'