fixing test_traffic get_total

This commit is contained in:
Cayo Puigdefabregas 2021-02-08 19:39:53 +01:00 committed by Santiago Lamora
parent c952d782cd
commit 091120d3c2
1 changed files with 9 additions and 9 deletions

View File

@ -77,11 +77,11 @@ class TrafficBillingTest(BaseTrafficBillingTest):
with freeze_time(now+relativedelta(months=1)):
bill = account.orders.bill(proforma=True)[0]
self.report_traffic(account, 10**10*9)
self.assertEqual(0, bill.get_total())
self.assertEqual(0, bill.total)
with freeze_time(now+relativedelta(months=3)):
bill = account.orders.bill(proforma=True)[0]
self.assertEqual((90-10)*10, bill.get_total())
self.assertEqual((90-10)*10, bill.total)
def test_multiple_traffics(self):
self.create_traffic_service()
@ -93,7 +93,7 @@ class TrafficBillingTest(BaseTrafficBillingTest):
with freeze_time(timezone.now()+relativedelta(months=1)):
bill1 = account1.orders.bill().pop()
bill2 = account2.orders.bill().pop()
self.assertNotEqual(bill1.get_total(), bill2.get_total())
self.assertNotEqual(bill1.total, bill2.total)
class TrafficPrepayBillingTest(BaseTrafficBillingTest):
@ -139,31 +139,31 @@ class TrafficPrepayBillingTest(BaseTrafficBillingTest):
self.create_prepay(10, account=account)
bill = account.orders.bill(proforma=True)[0]
self.assertEqual(10*50, bill.get_total())
self.assertEqual(10*50, bill.total)
self.report_traffic(account, 10**10)
with freeze_time(now+relativedelta(months=1)):
bill = account.orders.bill(proforma=True, new_open=True)[0]
self.assertEqual(2*10*50 + 0*10, bill.get_total())
self.assertEqual(2*10*50 + 0*10, bill.total)
# TODO RuntimeWarning: DateTimeField MetricStorage.updated_on received a naive
self.report_traffic(account, 10**10)
with freeze_time(now+relativedelta(months=1)):
bill = account.orders.bill(proforma=True, new_open=True)[0]
self.assertEqual(2*10*50 + 0*10, bill.get_total())
self.assertEqual(2*10*50 + 0*10, bill.total)
self.report_traffic(account, 10**10)
with freeze_time(now+relativedelta(months=1)):
bill = account.orders.bill(proforma=True, new_open=True)[0]
self.assertEqual(2*10*50 + (30-10-10)*10, bill.get_total())
self.assertEqual(2*10*50 + (30-10-10)*10, bill.total)
with freeze_time(now+relativedelta(months=2)):
self.report_traffic(account, 10**11)
with freeze_time(now+relativedelta(months=1)):
bill = account.orders.bill(proforma=True, new_open=True)[0]
self.assertEqual(2*10*50 + (30-10-10)*10, bill.get_total())
self.assertEqual(2*10*50 + (30-10-10)*10, bill.total)
with freeze_time(now+relativedelta(months=3)):
bill = account.orders.bill(proforma=True, new_open=True)[0]
self.assertEqual(4*10*50 + (30-10-10)*10 + (100-10-10)*10, bill.get_total())
self.assertEqual(4*10*50 + (30-10-10)*10 + (100-10-10)*10, bill.total)