from braintree.util.crypto import Crypto
from braintree.webhook_notification import WebhookNotification
import base64
from datetime import datetime
class WebhookTestingGateway(object):
def __init__(self, gateway):
self.gateway = gateway
self.config = gateway.config
def sample_notification(self, kind, id):
payload = base64.encodestring(self.__sample_xml(kind, id))
hmac_payload = Crypto.sha1_hmac_hash(self.gateway.config.private_key, payload)
signature = "%s|%s" % (self.gateway.config.public_key, hmac_payload)
return signature, payload
def __sample_xml(self, kind, id):
timestamp = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
sample_xml = """
%s
%s
%s
""" % (timestamp, kind, self.__subject_sample_xml(kind, id))
return sample_xml.encode('utf-8')
def __subject_sample_xml(self, kind, id):
if kind == WebhookNotification.Kind.SubMerchantAccountApproved:
return self.__merchant_account_approved_sample_xml(id)
elif kind == WebhookNotification.Kind.SubMerchantAccountDeclined:
return self.__merchant_account_declined_sample_xml(id)
elif kind == WebhookNotification.Kind.TransactionDisbursed:
return self.__transaction_disbursed_sample_xml(id)
elif kind == WebhookNotification.Kind.PartnerMerchantConnected:
return self.__partner_merchant_connected_sample_xml()
elif kind == WebhookNotification.Kind.PartnerMerchantDisconnected:
return self.__partner_merchant_disconnected_sample_xml()
elif kind == WebhookNotification.Kind.PartnerMerchantDeclined:
return self.__partner_merchant_declined_sample_xml()
elif kind == WebhookNotification.Kind.DisbursementException:
return self.__disbursement_exception_sample_xml(id)
elif kind == WebhookNotification.Kind.Disbursement:
return self.__disbursement_sample_xml(id)
elif kind == WebhookNotification.Kind.DisputeOpened:
return self.__dispute_opened_sample_xml(id)
elif kind == WebhookNotification.Kind.DisputeLost:
return self.__dispute_lost_sample_xml(id)
elif kind == WebhookNotification.Kind.DisputeWon:
return self.__dispute_won_sample_xml(id)
else:
return self.__subscription_sample_xml(id)
def __transaction_disbursed_sample_xml(self, id):
return """
%s
100
10
100
10
2013-07-09T18:23:29Z
""" % id
def __disbursement_exception_sample_xml(self, id):
return """
%s
- afv56j
- kj8hjk
false
false
merchant_account_token
USD
false
active
100.00
2014-02-09
bank_rejected
update_funding_information
""" % id
def __disbursement_sample_xml(self, id):
return """
%s
- afv56j
- kj8hjk
true
false
merchant_account_token
USD
false
active
100.00
2014-02-09
""" % id
def __dispute_opened_sample_xml(self, id):
return """
250.00
USD
2014-03-01
2014-03-21
open
fraud
%s
%s
250.00
""" % (id, id)
def __dispute_lost_sample_xml(self, id):
return """
250.00
USD
2014-03-01
2014-03-21
lost
fraud
%s
%s
250.00
""" % (id, id)
def __dispute_won_sample_xml(self, id):
return """
250.00
USD
2014-03-01
2014-03-21
won
fraud
%s
%s
250.00
""" % (id, id)
def __subscription_sample_xml(self, id):
return """
%s
""" % id
def __merchant_account_approved_sample_xml(self, id):
return """
%s
active
master_ma_for_%s
active
""" % (id, id)
def __merchant_account_declined_sample_xml(self, id):
return """
Credit score is too low
82621
Credit score is too low
base
%s
suspended
master_ma_for_%s
suspended
""" % (id, id)
def __partner_merchant_connected_sample_xml(self):
return """
abc123
public_key
private_key
public_id
cse_key
"""
def __partner_merchant_disconnected_sample_xml(self):
return """
abc123
"""
def __partner_merchant_declined_sample_xml(self):
return """
abc123
"""