def test_1(self):
case_id = conf.merchant_id + "001"
response = chargeback_document.upload_document(case_id, self.document_to_upload1, config=conf)
self.assertEquals('000', response['responseCode'])
self.assertEquals('Success', response['responseMessage'])
response = chargeback_document.upload_document(case_id, self.document_to_upload2, config=conf)
self.assertEquals('000', response['responseCode'])
self.assertEquals('Success', response['responseMessage'])
response = chargeback_document.upload_document(case_id, self.document_to_upload3, config=conf)
self.assertEquals('000', response['responseCode'])
self.assertEquals('Success', response['responseMessage'])
response = chargeback_document.list_documents(case_id, config=conf)
document_list = response['documentIds']
self.assertIn("test.jpg", document_list)
self.assertIn("test.gif", document_list)
self.assertIn("test.pdf", document_list)
document_to_retrieve = package_root + "/tests/test1.tiff"
chargeback_document.retrieve_document(case_id, "test.jpg", document_to_retrieve, config=conf)
self.assertTrue(os.path.exists(document_to_retrieve))
os.remove(document_to_retrieve)
chargeback_document.retrieve_document(case_id, "test.gif", document_to_retrieve, config=conf)
self.assertTrue(os.path.exists(document_to_retrieve))
os.remove(document_to_retrieve)
chargeback_document.retrieve_document(case_id, "test.pdf", document_to_retrieve, config=conf)
self.assertTrue(os.path.exists(document_to_retrieve))
os.remove(document_to_retrieve)
response = chargeback_document.replace_document(case_id, "test.jpg", self.document_to_upload4, config=conf)
self.assertEquals('000', response['responseCode'])
self.assertEquals('Success', response['responseMessage'])
chargeback_document.retrieve_document(case_id, "test.tiff", document_to_retrieve, config=conf)
self.assertTrue(os.path.exists(document_to_retrieve))
os.remove(document_to_retrieve)
response = chargeback_document.delete_document(case_id, "test.gif", config=conf)
self.assertEquals('000', response['responseCode'])
self.assertEquals('Success', response['responseMessage'])
response = chargeback_document.list_documents(case_id, config=conf)
document_list = response['documentIds']
self.assertIn("test.pdf", document_list)
self.assertIn("test.tiff", document_list)
def test_upload_chargebackDocument(self):
response = chargeback_document.upload_document("10000",
document_to_upload2)
self.assertEquals('013', response['responseCode'])
self.assertEquals('Invalid File Content', response['responseMessage'])
self.assertEquals('test.jpg', response['documentId'])
self.assertEquals('10000', response['caseId'])
def test_4(self):
case_id = conf.merchant_id + "004"
document_maxsize = package_root + "/tests/maxsize.tif"
open(document_maxsize, "w+").close()
response = chargeback_document.upload_document(case_id, document_maxsize, config=conf)
self.assertEquals('005', response['responseCode'])
self.assertEquals('Document already exists', response['responseMessage'])
os.remove(document_maxsize)
document_maxsize = package_root + "/tests/maxsize1.tif"
with open(document_maxsize, "w+") as f:
f.seek(2100000)
f.write("\0")
response = chargeback_document.upload_document(case_id, document_maxsize, config=conf)
self.assertEquals('012', response['responseCode'])
self.assertEquals('Filesize exceeds limit of 1MB', response['responseMessage'])
os.remove(document_maxsize)
response = chargeback_document.upload_document(case_id, self.document_to_upload1, config=conf)
self.assertEquals('008', response['responseCode'])
self.assertEquals('Max Document Limit Per Case Reached', response['responseMessage'])
def test_upload_document(self, mock_http_post_document_request):
mock_http_post_document_request.return_value = OrderedDict([
(u'@xmlns', u'http://www.vantivcnp.com/chargebacks'),
(u'merchantId', u'999'), (u'caseId', u'10000'),
(u'documentId', u'doc.tiff'), (u'responseCode', u'000'),
(u'responseMessage', u'Success')
])
response = chargeback_document.upload_document("10000",
document_to_upload)
args = mock_http_post_document_request.call_args
self.assertEquals(args[0][0],
"/services/chargebacks/upload/10000/doc.tiff")
self.assertEquals(response['responseCode'], '000')
self.assertEquals(response['responseMessage'], 'Success')
self.assertEquals(response['caseId'], '10000')
self.assertEquals(response['documentId'], 'doc.tiff')
def test_error_response(self):
response = chargeback_document.upload_document("10001",
document_to_upload1)
self.assertEquals('001', response['responseCode'])
self.assertEquals('Invalid Merchant', response['responseMessage'])
try:
chargeback_document.retrieve_document(123009, "logo.tiff",
"test.tiff")
except utils.ChargebackDocumentError as e:
self.assertEquals("Document Not Found", e.message)
self.assertEquals("009", e.code)
try:
chargeback_document.retrieve_document(123404, "logo.tiff",
"test.tiff")
except utils.ChargebackWebError as e:
self.assertEquals("Could not find requested object.", e.message)
self.assertEquals("404", e.code)
def test_3(self):
case_id = conf.merchant_id + "003"
response = chargeback_document.upload_document(case_id, self.document_to_upload1, config=conf)
self.assertEquals('004', response['responseCode'])
self.assertEquals('Case Not In Merchant Queue', response['responseMessage'])
def test_2(self):
case_id = conf.merchant_id + "002"
response = chargeback_document.upload_document(case_id, self.document_to_upload1, config=conf)
self.assertEquals('010', response['responseCode'])
self.assertEquals('Case not in valid cycle', response['responseMessage'])