add user to org

This commit is contained in:
2025-11-10 22:08:26 -03:00
parent 648da66c90
commit c00a42ea39
6 changed files with 139 additions and 19 deletions

View File

@@ -1,6 +1,13 @@
import json
from http import HTTPMethod, HTTPStatus
from layercake.dynamodb import (
DynamoDBPersistenceLayer,
PartitionKey,
SortKey,
TransactKey,
)
from ..conftest import HttpApiProxy, LambdaContext
@@ -40,6 +47,93 @@ def test_get_orgs(
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,
@@ -57,14 +151,14 @@ def test_add_user(
},
'org': {
'id': 'f6000f79-6e5c-49a0-952f-3bda330ef278',
'name': 'Branco do Brasil',
'name': 'Banco do Brasil',
'cnpj': '00000000000191',
},
},
),
lambda_context,
)
assert r['statusCode'] == HTTPStatus.CREATED
assert r['statusCode'] == HTTPStatus.CONFLICT
def test_org_not_found(

View File

@@ -1,11 +1,10 @@
// Users
{"id": "15bacf02-1535-4bee-9022-19d106fd7518", "sk": "0", "name": "Sérgio R Siqueira", "email": "sergio@somosbeta.com.br", "cpf": "07879819908"}
// User emails
{"id": "15bacf02-1535-4bee-9022-19d106fd7518", "sk": "emails#sergio@somosbeta.com.br"}
{"id": "15bacf02-1535-4bee-9022-19d106fd7518", "sk": "emails#sergio@somosbeta.com.br", "email_primary": true, "mx_record_exists": true}
{"id": "213a6682-2c59-4404-9189-12eec0a846d4", "sk": "orgs#f6000f79-6e5c-49a0-952f-3bda330ef278", "name": "Banco do Brasil", "cnpj": "00000000000191"}
// User orgs
{"id": "213a6682-2c59-4404-9189-12eec0a846d4", "sk": "orgs#286f7729-7765-482a-880a-0b153ea799be", "name": "Banco do Brasil", "cnpj": "00000000000191"}
{"id": "15bacf02-1535-4bee-9022-19d106fd7518", "sk": "orgs#286f7729-7765-482a-880a-0b153ea799be", "name": "Banco do Brasil", "cnpj": "00000000000191"}
// Enrollments
{"id": "578ec87f-94c7-4840-8780-bb4839cc7e64", "sk": "0", "course": {"id": "af3258f0-bccf-4781-aec6-d4c618d234a7", "name": "pytest", "access_period": 180}, "user": {"id": "068b4600-cc36-4b55-b832-bb620021705a", "name": "Benjamin Burnley", "email": "burnley@breakingbenjamin.com"}}
@@ -14,9 +13,15 @@
{"id": "2a8963fc-4694-4fe2-953a-316d1b10f1f5", "sk": "0", "name": "pytest", "cnpj": "04978826000180"}
{"id": "f6000f79-6e5c-49a0-952f-3bda330ef278", "sk": "0", "name": "Banco do Brasil", "cnpj": "00000000000191"}
{"id": "orgmembers#f6000f79-6e5c-49a0-952f-3bda330ef278", "sk": "15bacf02-1535-4bee-9022-19d106fd7518"}
// Indicies
// CNPJs
{"id": "cnpj", "sk": "04978826000180", "org_id": "2a8963fc-4694-4fe2-953a-316d1b10f1f5"}
{"id": "cnpj", "sk": "00000000000191", "org_id": "6000f79-6e5c-49a0-952f-3bda330ef278"}
// CPFs
{"id": "cpf", "sk": "07879819908", "user_id": "15bacf02-1535-4bee-9022-19d106fd7518""}
{"id": "cpf", "sk": "07879819908", "user_id": "15bacf02-1535-4bee-9022-19d106fd7518"}
// Emails
{"id": "email", "sk": "sergio@somosbeta.com.br", "user_id": "15bacf02-1535-4bee-9022-19d106fd7518"}