def test_receive_text_plain_with_mdn_signed(self):
headers = TestPartnerReceiveMessage.received_text_plain_headers
body = TestPartnerReceiveMessage.received_text_plain_content
pm = PartnerManager(headers, body)
resp_headers, resp_content = pm.receive()
assert_not_equal(None, resp_headers)
assert_not_equal(None, resp_content)
def post_message_receive_by_id(domain, id):
logger.debug(
'received as2 message from domain:{0} , uri:{1} ,query_string:{2}'.
format(domain, request.url, request.query_string))
try:
headers = dict(request.headers)
body = get_request_body()
mgr = PartnerManager(headers, body)
headers, body = mgr.receive(id)
return build_response(200, headers, body)
except:
logger.exception('message receive via agreement-id failed')
return build_response(500, None, str(sys.exc_info()[1]))
def test_receive_with_decompress_error(self,
mock_receive_decode,
mock_mdn_encode
):
headers = TestPartnerReceiveMessage.received_text_plain_headers
body = TestPartnerReceiveMessage.received_text_plain_content
mock_mdn_encode.side_effect = mock.Mock(return_value=({'x': 'x-decompress-error'}, "mdn: decompress error"))
mock_receive_decode.side_effect = AS2DeCompressException('decompress error')
pm = PartnerManager(headers, body)
mdn_header, mdn_content = pm.receive('A01')
assert_not_equal(mdn_header, None)
assert_not_equal(mdn_content, None)
assert_equal(mdn_header, {'x': 'x-decompress-error'})
assert_equal(mdn_content, "mdn: decompress error")
def test_receive_with_verify_error(self,
mock_receive_decode,
mock_mdn_encode
):
headers = TestPartnerReceiveMessage.received_text_plain_headers
body = TestPartnerReceiveMessage.received_text_plain_content
mock_receive_decode.side_effect = AS2VerifySignatureException('verify signature error')
mock_mdn_encode.side_effect = mock.Mock(
return_value=({'x': 'x-verify-signature-error'}, "mdn: verify signature error"))
pm = PartnerManager(headers, body)
mdn_header, mdn_content = pm.receive('A01')
assert_not_equal(mdn_header, None)
assert_not_equal(mdn_content, None)
assert_equal(mdn_header, {'x': 'x-verify-signature-error'})
assert_equal(mdn_content, "mdn: verify signature error")
def test_receive_failed_when_agreement_not_found(self, headers, body):
pm = PartnerManager(headers, body)
with assert_raises(Exception):
pm.receive()
def test_receive_failed_when_agreement_id_not_found(self):
headers = TestPartnerReceiveMessage.received_text_plain_headers
body = TestPartnerReceiveMessage.received_text_plain_content
pm = PartnerManager(headers, body)
pm.receive(agreement_id='A-ID-Not-Found', is_test=True)
