class VKOrderTest(VKBaseTest):
def setUp(self):
super(VKOrderTest, self).setUp()
self.order = VKOrder(TEST_PAYMENT_NAME, self.engine.callback)
def test_order(self):
self.eq(self.order(1, 'uid', 'gems', 10, CHARGEABLE),
self.order.response({'order_id': 1}))
def test_db(self):
self.order(1, 'uid', 'gems', 10, CHARGEABLE)
self.eq(list(self.order.collection.find({'order_id': 1},
{'_id': False})),
[{'order_id': 1}])
def test_callback(self):
for _ in xrange(3):
self.order(1, 'uid', 'gems', 10, CHARGEABLE)
self.eq(self.engine.log, [('uid', 'gems', 10)])
def test_callback_not_chargeable(self):
self.order(1, 'uid', 'gems', 10, 'notchargeable')
self.eq(self.engine.log, [])
def test_order_callback_error(self):
self.payment.order.callback = self.error_callback
self.raises(CallbackError, self.payment.order,
1, 'uid', 'gems', 10, CHARGEABLE)