"""Test format error util.""" from unittest.mock import Mock from ledger.constants.error import ERROR_ENTITY_DOES_NOT_EXIST from ledger.utils.format_error import format_error_response, validation_error def test_format_error_response(): """Test to formatted error response.""" accounting_period_id = 123 error_message = ERROR_ENTITY_DOES_NOT_EXIST.format( object_id=accounting_period_id, object_type='Accounting Period' ) mock_response = Mock() mock_response.status_code = 200 mock_response.json.return_value = { 'message': error_message, 'code': 'error', 'status': 400, } res = format_error_response(mock_response) assert res.status == 400 assert res.errors['message'] == error_message def test_validation_error(): """Test validation error creates an error response.""" res = validation_error('How dare you?!') assert res.errors['code'] == 'error' assert res.errors['message'] == 'How dare you?!' assert res.status == 400