class SignatureTest(VKBaseTest):
def setUp(self):
self.signature = Signature('secretkey')
self.params = {'c': 'a', 'b': 'v', 'a': 'z'}
def test_string(self):
self.eq(self.signature.string(self.params),
'a=zb=vc=a' + self.signature.key)
def test_md5(self):
self.eq(self.signature.md5(self.params),
md5(self.signature.string(self.params)).hexdigest())
def test_check(self):
self.true(self.signature.check(self.params,
self.signature.md5(self.params)))
self.false(self.signature.check(self.params, 'qwe'))
class VKPayment(BasePayment):
def __init__(self, name, goods, secret, callback):
self.signature = Signature(secret)
self.info = Info(goods)
self.order = VKOrder(name, callback)
super(VKPayment, self).__init__(goods)
def request(self, args):
notification_type = args.get('notification_type')
try:
if not self.signature.check(args, args.pop('sig')):
raise SignatureError()
item_id, name, count = self.get_name_count(args['item'])
if notification_type.startswith(GET_ITEM):
return self.info(name, item_id)
if notification_type.startswith(ORDER):
return self.order(args['order_id'], args['receiver_id'],
name, item_id, args['status'])
except (ItemFormatError, UnknownItemError, InvalidCountError,
CallbackError, SignatureError) as error:
return error.response()
