44from lib .fixtures import * # NOSONAR
55from lib .constants import LETTERS_ENDPOINT
66from lib .generators import Generators
7+ from lib .errorhandler import ErrorHandler
78
89@pytest .mark .test
910@pytest .mark .devtest
@@ -18,7 +19,8 @@ def test_200_get_letter_status(url, bearer_token):
1819 letter_id = get_letter_id .json ().get ("data" )[0 ].get ("id" )
1920 get_letter_data = requests .get (f"{ url } /{ LETTERS_ENDPOINT } /{ letter_id } /data" , headers = headers )
2021
21- assert get_letter_data .status_code == 200
22+ ErrorHandler .handle_retry (get_letter_data )
23+ assert get_letter_data .status_code == 200 , f"Response: { get_letter_data .status_code } : { get_letter_data .text } "
2224 assert get_letter_data .headers .get ("Content-Type" ) == "application/pdf"
2325
2426
@@ -31,6 +33,7 @@ def test_404_letter_does_not_exist(url, bearer_token):
3133 headers = Generators .generate_valid_headers (bearer_token .value )
3234 get_message_response = requests .get (f"{ url } /{ LETTERS_ENDPOINT } /xx" , headers = headers )
3335
36+ ErrorHandler .handle_retry (get_message_response )
3437 assert get_message_response .status_code == 404
3538 assert get_message_response .json ().get ("errors" )[0 ].get ("detail" ) == "No resource found with that ID"
3639
@@ -44,6 +47,7 @@ def test_404_letter_does_not_exist(url, bearer_token):
4447 headers = Generators .generate_valid_headers (bearer_token .value )
4548 get_message_response = requests .get (f"{ url } /{ LETTERS_ENDPOINT } /{ letter_id } /data" , headers = headers )
4649
50+ ErrorHandler .handle_retry (get_message_response )
4751 assert get_message_response .status_code == 404
4852 assert get_message_response .json ().get ("errors" )[0 ].get ("detail" ) == "No resource found with that ID"
4953
@@ -57,4 +61,5 @@ def test_500_letter_does_not_exist(url, bearer_token):
5761 headers = Generators .generate_valid_headers (bearer_token .value )
5862 get_message_response = requests .get (f"{ url } /{ LETTERS_ENDPOINT } /{ letter_id } /data" , headers = headers )
5963
64+ ErrorHandler .handle_retry (get_message_response )
6065 assert get_message_response .status_code == 500
0 commit comments