def test_total_corrections_with_no_corrections(self):
order_round = OrderRoundFactory()
self.assertEqual(order_round.total_corrections(), {
'supplier_inc': 0,
'voko_inc': 0,
'supplier_exc': 0
})
def test_total_corrections(self):
round = OrderRoundFactory()
corr1 = OrderProductCorrectionFactory(order_product__order__order_round=round, charge_supplier=True)
corr2 = OrderProductCorrectionFactory(order_product__order__order_round=round, charge_supplier=False)
self.assertEqual(round.total_corrections(),
{'supplier_inc': corr1.calculate_refund(),
'voko_inc': corr2.calculate_refund(),
'supplier_exc': corr1.calculate_supplier_refund()})
def test_total_corrections(self):
order_round = OrderRoundFactory()
corr1 = OrderProductCorrectionFactory(
order_product__order__order_round=order_round,
charge_supplier=True)
corr2 = OrderProductCorrectionFactory(
order_product__order__order_round=order_round,
charge_supplier=False)
self.assertEqual(
order_round.total_corrections(), {
'supplier_inc': corr1.calculate_refund(),
'voko_inc': corr2.calculate_refund(),
'supplier_exc': corr1.calculate_supplier_refund()
})
def test_total_corrections_with_no_corrections(self):
round = OrderRoundFactory()
self.assertEqual(round.total_corrections(),
{'supplier_inc': 0, 'voko_inc': 0, 'supplier_exc': 0})
