38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from layercake.dateutils import now
|
|
from layercake.dynamodb import DynamoDBPersistenceLayer, PartitionKey
|
|
|
|
import events.batch.csv_into_chunks as app
|
|
|
|
|
|
def test_chunk_csv(
|
|
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
|
|
lambda_context,
|
|
):
|
|
pk = 'BATCH_JOB#ORG#1411844c-10d6-456e-959d-e91775145461'
|
|
sk = 'FILE#2025-11-13T16:04:53.024743'
|
|
event = {
|
|
'detail': {
|
|
'new_image': {
|
|
'id': pk,
|
|
'sk': sk,
|
|
's3_uri': 's3://saladeaula.digital/samples/large_users.csv',
|
|
'columns': {
|
|
'1:email',
|
|
'2:cpf',
|
|
'3:name',
|
|
},
|
|
'org': {
|
|
'id': '1411844c-10d6-456e-959d-e91775145461',
|
|
'name': 'EDUSEG',
|
|
'cnpj': '15608435000190',
|
|
},
|
|
'created_at': now(),
|
|
},
|
|
},
|
|
}
|
|
|
|
app.lambda_handler(event, lambda_context) # type: ignore
|
|
|
|
r = dynamodb_persistence_layer.collection.query(PartitionKey(pk), limit=100)
|
|
assert len(r['items']) == 67
|