pedro/django-orchestra-test
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fixing test_traffic get_total

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

18
orchestra/contrib/services/tests/functional_tests/test_traffic.py
View file

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