1010from .helper import load_event , return_env_vars_dict , create_ddb_table_property_web
1111
1212
13+ def get_property_pk_sk (property_id ):
14+ country , city , street , number = property_id .split ('/' )
15+ pk_details = f"{ country } #{ city } " .replace (' ' , '-' ).lower ()
16+ return {
17+ 'PK' : f"PROPERTY#{ pk_details } " ,
18+ 'SK' : f"{ street } #{ str (number )} " .replace (' ' , '-' ).lower (),
19+ }
20+
21+
1322@mock .patch .dict (os .environ , return_env_vars_dict (), clear = True )
1423def test_property_approved (dynamodb , mocker ):
1524 eventbridge_event = load_event ('events/property_approved.json' )
@@ -20,8 +29,8 @@ def test_property_approved(dynamodb, mocker):
2029
2130 create_ddb_table_property_web (dynamodb )
2231
23- context = LambdaContext ()
24- ret = app . lambda_handler ( apigw_event , context ) # type: ignore
25- assert 'result' in ret
26- result = ret [ 'result' ]
27- assert "success" in result . lower ()
32+ ret = app . lambda_handler ( eventbridge_event , LambdaContext ()) # type: ignore
33+ assert ret [ 'result' ] == 'Successfully updated property status'
34+
35+ ddbitem_after = dynamodb . Table ( 'table1' ). get_item ( Key = get_property_pk_sk ( property_id ))
36+ assert ddbitem_after [ 'Item' ][ 'status' ] == 'APPROVED'
0 commit comments