171 lines
5.0 KiB
Python
171 lines
5.0 KiB
Python
from http import HTTPMethod, HTTPStatus
|
|
|
|
from layercake.dateutils import now
|
|
from layercake.dynamodb import (
|
|
DynamoDBPersistenceLayer,
|
|
KeyPair,
|
|
PartitionKey,
|
|
SortKey,
|
|
TransactKey,
|
|
)
|
|
|
|
from .conftest import HttpApiProxy, LambdaContext
|
|
|
|
|
|
def test_start_progress(
|
|
app,
|
|
seeds,
|
|
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
|
|
http_api_proxy: HttpApiProxy,
|
|
lambda_context: LambdaContext,
|
|
):
|
|
# This data was added from seeds
|
|
r = app.lambda_handler(
|
|
http_api_proxy(
|
|
raw_path='/',
|
|
method=HTTPMethod.POST,
|
|
body={
|
|
'ID_MATRICULA': '123',
|
|
'APROVEITAMENTO': '23.152173913043477',
|
|
'ANDAMENTO': '38.888888888888886',
|
|
'status': 'IN_PROGRESS',
|
|
},
|
|
),
|
|
lambda_context,
|
|
)
|
|
assert r['statusCode'] == HTTPStatus.NO_CONTENT
|
|
|
|
# Check `seeds.jsonl` for sample data related to this query
|
|
r = dynamodb_persistence_layer.collection.query(
|
|
PartitionKey('d9da85f2-e09f-472d-9515-3d91d70f1e8a')
|
|
)
|
|
docx = next((x for x in r['items'] if x['sk'] == '0'), {})
|
|
assert 'started_at' in docx
|
|
assert any(
|
|
item.get('sk') == 'SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS'
|
|
for item in r['items']
|
|
)
|
|
assert len(r['items']) == 2
|
|
|
|
|
|
def test_update_progress(
|
|
app,
|
|
seeds,
|
|
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
|
|
http_api_proxy: HttpApiProxy,
|
|
lambda_context: LambdaContext,
|
|
):
|
|
# This data was added from seeds
|
|
r = app.lambda_handler(
|
|
http_api_proxy(
|
|
raw_path='/',
|
|
method=HTTPMethod.POST,
|
|
body={
|
|
'ID_MATRICULA': '456',
|
|
'APROVEITAMENTO': '23.152173913043477',
|
|
'ANDAMENTO': '12.888888888888886',
|
|
'status': 'IN_PROGRESS',
|
|
},
|
|
),
|
|
lambda_context,
|
|
)
|
|
assert r['statusCode'] == HTTPStatus.NO_CONTENT
|
|
|
|
# Check `seeds.jsonl` for sample data related to this query
|
|
r = dynamodb_persistence_layer.collection.query(
|
|
PartitionKey('197991aa-52e2-4e0c-b2a4-a7c53bcfee02')
|
|
)
|
|
assert len(r['items']) == 2
|
|
assert any(
|
|
item.get('sk') == 'SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS'
|
|
for item in r['items']
|
|
)
|
|
|
|
|
|
def test_set_as_completed(
|
|
app,
|
|
seeds,
|
|
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
|
|
http_api_proxy: HttpApiProxy,
|
|
lambda_context: LambdaContext,
|
|
):
|
|
# Update `created_at` to avoid future test failures
|
|
dynamodb_persistence_layer.update_item(
|
|
key=KeyPair('6c7e3d9b-f5d1-4da4-9e55-0825bb6ff2b8', '0'),
|
|
update_expr='SET created_at = :created_at',
|
|
expr_attr_values={':created_at': now()},
|
|
)
|
|
|
|
# This data was added from seeds
|
|
r = app.lambda_handler(
|
|
http_api_proxy(
|
|
raw_path='/',
|
|
method=HTTPMethod.POST,
|
|
body={
|
|
'ID_MATRICULA': '567',
|
|
'APROVEITAMENTO': '89.152173913043477',
|
|
'ANDAMENTO': '100',
|
|
'status': 'COMPLETED',
|
|
},
|
|
),
|
|
lambda_context,
|
|
)
|
|
assert r['statusCode'] == HTTPStatus.NO_CONTENT
|
|
|
|
# Check `seeds.jsonl` for sample data related to this query
|
|
r = dynamodb_persistence_layer.collection.query(
|
|
PartitionKey('6c7e3d9b-f5d1-4da4-9e55-0825bb6ff2b8')
|
|
)
|
|
|
|
docx = next((x for x in r['items'] if x['sk'] == '0'), {})
|
|
assert 'completed_at' in docx
|
|
assert len(r['items']) == 5
|
|
assert any(item.get('sk') == 'LOCK' for item in r['items'])
|
|
assert any(
|
|
item.get('sk') == 'SCHEDULE#REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS'
|
|
for item in r['items']
|
|
)
|
|
|
|
r = dynamodb_persistence_layer.collection.query(PartitionKey('LOCK'))
|
|
assert len(r['items']) == 1
|
|
|
|
|
|
def test_set_as_failed(
|
|
app,
|
|
seeds,
|
|
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
|
|
http_api_proxy: HttpApiProxy,
|
|
lambda_context: LambdaContext,
|
|
):
|
|
# Update `created_at` to avoid future test failures
|
|
dynamodb_persistence_layer.update_item(
|
|
key=KeyPair('6c7e3d9b-f5d1-4da4-9e55-0825bb6ff2b8', '0'),
|
|
update_expr='SET created_at = :created_at',
|
|
expr_attr_values={':created_at': now()},
|
|
)
|
|
|
|
# This data was added from seeds
|
|
r = app.lambda_handler(
|
|
http_api_proxy(
|
|
raw_path='/',
|
|
method=HTTPMethod.POST,
|
|
body={
|
|
'ID_MATRICULA': '567',
|
|
'APROVEITAMENTO': '12.152173913043477',
|
|
'ANDAMENTO': '100',
|
|
'status': 'COMPLETED',
|
|
},
|
|
),
|
|
lambda_context,
|
|
)
|
|
assert r['statusCode'] == HTTPStatus.NO_CONTENT
|
|
|
|
# Check `seeds.jsonl` for sample data related to this query
|
|
r = dynamodb_persistence_layer.collection.get_items(
|
|
TransactKey('6c7e3d9b-f5d1-4da4-9e55-0825bb6ff2b8')
|
|
+ SortKey('0')
|
|
+ SortKey('METADATA#DEDUPLICATION_WINDOW', rename_key='dedup_window')
|
|
)
|
|
assert 'failed_at' in r
|
|
assert 'dedup_window' in r
|