Files
saladeaula.digital/http-api/tests/routes/test_enrollments.py
2025-04-13 01:11:44 -03:00

53 lines
1.5 KiB
Python

from http import HTTPMethod, HTTPStatus
from layercake.dynamodb import (
ComposeKey,
DynamoDBCollection,
DynamoDBPersistenceLayer,
KeyPair,
PartitionKey,
)
from ..conftest import HttpApiProxy, LambdaContext
def test_cancel_enrollment(
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='/enrollments/43ea4475-c369-4f90-b576-135b7df5106b/cancel',
method=HTTPMethod.PATCH,
headers={'X-Tenant': '*'},
body={
'id': '43ea4475-c369-4f90-b576-135b7df5106b',
'lock_hash': 'f8f7996aa99d50eb85266be5a9fca1db',
'course': {
'id': '123',
'name': 'NR-10',
'time_in_days': 720,
},
'vacancy': {
'sk': 'QV4sXY3DvSTUMGJ4QqsrwJ#43ea4475-c369-4f90-b576-135b7df5106b',
'id': 'vacancies#cJtK9SsnJhKPyxESe7g3DG',
},
},
),
lambda_context,
)
assert r['statusCode'] == HTTPStatus.OK
collect = DynamoDBCollection(dynamodb_persistence_layer)
enrollment = collect.get_item(KeyPair('43ea4475-c369-4f90-b576-135b7df5106b', '0'))
assert enrollment['status'] == 'CANCELED'
vacancies = collect.query(
PartitionKey(ComposeKey('cJtK9SsnJhKPyxESe7g3DG', 'vacancies'))
)
assert len(vacancies['items']) == 1