add tests to konviva

This commit is contained in:
2025-05-29 18:20:47 -03:00
parent 590cf10777
commit efd962cdba
5 changed files with 11 additions and 7 deletions

View File

@@ -20,7 +20,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
new_image = event.detail['new_image']
now_ = now()
with enrollment_layer.transact_items() as transact:
with enrollment_layer.transact_writer() as transact:
transact.update(
key=KeyPair(new_image['id'], '0'),
update_expr='SET #status = :archived, update_date = :update_date',
@@ -41,6 +41,5 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
'create_date': now_,
},
)
transact.write_items()
return True

View File

@@ -20,7 +20,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
new_image = event.detail['new_image']
now_ = now()
with enrollment_layer.transact_items() as transact:
with enrollment_layer.transact_writer() as transact:
transact.update(
key=KeyPair(new_image['id'], '0'),
update_expr='SET #status = :expired, update_date = :update_date',
@@ -43,6 +43,4 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
},
)
transact.write_items()
return True

View File

@@ -35,7 +35,7 @@ def token(username: str) -> KonvivaToken:
r.raise_for_status()
# Because Konviva does not return the proper HTTP status code
if err := glom(r.json(), 'errors.0', default=None):
if err := glom(r.json(), 'errors', default=None):
raise KonvivaError(err)
return KonvivaToken(**r.json())

View File

@@ -23,7 +23,7 @@ Globals:
Architectures:
- x86_64
Layers:
- !Sub arn:aws:lambda:sa-east-1:336641857101:layer:layercake:68
- !Sub arn:aws:lambda:sa-east-1:336641857101:layer:layercake:72
Environment:
Variables:
TZ: America/Sao_Paulo

View File

@@ -1,3 +1,5 @@
import pytest
import konviva
from config import KONVIVA_API_URL
@@ -9,3 +11,8 @@ def test_konviva_token():
redirect_uri = konviva.redirect_uri(token)
assert KONVIVA_API_URL in redirect_uri
def test_konviva_token_error():
with pytest.raises(konviva.KonvivaError):
konviva.token('sergio@doesnotexist.com')