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

View File

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

View File

@@ -35,7 +35,7 @@ def token(username: str) -> KonvivaToken:
r.raise_for_status() r.raise_for_status()
# Because Konviva does not return the proper HTTP status code # 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) raise KonvivaError(err)
return KonvivaToken(**r.json()) return KonvivaToken(**r.json())

View File

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

View File

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