Skip to content

Commit 90eacec

Browse files
committed
fix: fixed property_approved event test
1 parent ae02eab commit 90eacec

File tree

1 file changed

+14
-5
lines changed

1 file changed

+14
-5
lines changed

unicorn_web/tests/unit/test_publication_approved_event_handler.py

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,15 @@
1010
from .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)
1423
def 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

Comments
 (0)