class TestUsageProvider(unittest.TestCase):
def setUp(self):
self.provider = UsageProvider(':memory:')
def tearDown(self):
self.provider.close()
def test_add_usage_item(self):
now = get_tz_aware_now()
later = now + timedelta(minutes=30)
umts, bytes_recv, bytes_sent = True, 12345460, 12333211
item = self.provider.add_usage_item(now, later, bytes_recv,
bytes_sent, umts)
usage_items = self.provider.get_usage_for_day(date.today())
self.assertIn(item, usage_items)
self.provider.delete_usage_item(item)
def test_delete_usage_item(self):
now = get_tz_aware_now()
later = now + timedelta(minutes=60)
umts, bytes_recv, bytes_sent = True, 12345470, 12333212
item = self.provider.add_usage_item(now, later, bytes_recv,
bytes_sent, umts)
usage_items = self.provider.get_usage_for_day(date.today())
self.assertIn(item, usage_items)
self.provider.delete_usage_item(item)
# now check that it is indeed gone
usage_items = self.provider.get_usage_for_day(date.today())
self.assertNotIn(item, usage_items)
def test_get_usage_for_day(self):
# add one usage item for today (45m)
now1 = get_tz_aware_now()
later1 = now1 + timedelta(minutes=45)
umts1, bytes_recv1, bytes_sent1 = True, 1200034, 124566
item1 = self.provider.add_usage_item(now1, later1, bytes_recv1,
bytes_sent1, umts1)
# add another usage item for today (17m) 55 minutes later
now2 = get_tz_aware_now() + timedelta(minutes=55)
later2 = now1 + timedelta(minutes=17)
umts2, bytes_recv2, bytes_sent2 = True, 12000, 1245
item2 = self.provider.add_usage_item(now2, later2, bytes_recv2,
bytes_sent2, umts2)
# add another usage item for tomorrow (25m)
now3 = get_tz_aware_now() + timedelta(days=1)
later3 = now3 + timedelta(minutes=25)
umts3, bytes_recv3, bytes_sent3 = True, 14000, 1785
item3 = self.provider.add_usage_item(now3, later3, bytes_recv3,
bytes_sent3, umts3)
# now get the usage for today
today_items = self.provider.get_usage_for_day(date.today())
self.assertIn(item1, today_items)
self.assertIn(item2, today_items)
self.assertNotIn(item3, today_items)
# get the usage for tomorrow
tomorrow = date.today() + timedelta(days=1)
tomorrow_items = self.provider.get_usage_for_day(tomorrow)
self.assertNotIn(item1, tomorrow_items)
self.assertNotIn(item2, tomorrow_items)
self.assertIn(item3, tomorrow_items)
# leave it as we found it
for i in [item1, item2, item3]:
self.provider.delete_usage_item(i)
def test_get_usage_for_month(self):
current_month = date.today().month
current_year = date.today().year
# add one usage item for day 12 of this month (45m)
now1 = datetime(date.today().year, current_month, 12, 13, 10,
tzinfo=timezone('UTC'))
later1 = now1 + timedelta(minutes=45)
umts1, bytes_recv1, bytes_sent1 = True, 1200034, 124566
item1 = self.provider.add_usage_item(now1, later1, bytes_recv1,
bytes_sent1, umts1)
# add another usage item for day 13 of this month (17m)
now2 = datetime(date.today().year, current_month, 13, 15, 10,
tzinfo=timezone('UTC'))
later2 = now1 + timedelta(minutes=17)
umts2, bytes_recv2, bytes_sent2 = True, 12000, 1245
item2 = self.provider.add_usage_item(now2, later2, bytes_recv2,
bytes_sent2, umts2)
# add another usage item for next month
if current_month < 12:
month = current_month + 1
year = current_year
else:
month = 1
year = current_year + 1
# next month at 6.50am (25m)
now3 = datetime(year, month, 2, 6, 50, tzinfo=timezone('UTC'))
later3 = now3 + timedelta(minutes=25)
umts3, bytes_recv3, bytes_sent3 = True, 14000, 1785
item3 = self.provider.add_usage_item(now3, later3, bytes_recv3,
bytes_sent3, umts3)
# now get the usage for this month
this_month_items = self.provider.get_usage_for_month(now1.date())
self.assertIn(item1, this_month_items)
self.assertIn(item2, this_month_items)
self.assertNotIn(item3, this_month_items)
# now get the usage for next month
next_month_items = self.provider.get_usage_for_month(now3)
self.assertNotIn(item1, next_month_items)
self.assertNotIn(item2, next_month_items)
self.assertIn(item3, next_month_items)
# leave it as we found it
for i in [item1, item2, item3]:
self.provider.delete_usage_item(i)
def test_get_total_usage(self):
current_month = date.today().month
current_year = date.today().year
# add one usage item for day 12 of this month (45m)
now1 = datetime(current_year, current_month, 12, 13, 10,
tzinfo=timezone('UTC'))
later1 = now1 + timedelta(minutes=45)
umts1, bytes_recv1, bytes_sent1 = True, 1200034, 124566
item1 = self.provider.add_usage_item(now1, later1, bytes_recv1,
bytes_sent1, umts1)
# add another usage item for day 13 of this month (17m), one year ago
now2 = datetime(current_year - 1, current_month, 13, 15, 10,
tzinfo=timezone('UTC'))
later2 = now1 + timedelta(minutes=17)
umts2, bytes_recv2, bytes_sent2 = True, 12000, 1245
item2 = self.provider.add_usage_item(now2, later2, bytes_recv2,
bytes_sent2, umts2)
items = self.provider.get_total_usage()
self.assertIn(item1, items)
self.assertIn(item2, items)
# leave it as we found it
for i in [item1, item2]:
self.provider.delete_usage_item(i)
def test_get_total_usage_passing_a_date(self):
current_month = date.today().month
current_year = date.today().year
# add one usage item for day 12 of this month (45m)
now1 = datetime(current_year, current_month, 12, 13, 10,
tzinfo=timezone('UTC'))
later1 = now1 + timedelta(minutes=45)
umts1, bytes_recv1, bytes_sent1 = True, 1200034, 124566
item1 = self.provider.add_usage_item(now1, later1, bytes_recv1,
bytes_sent1, umts1)
# add another usage item for day 13 of this month (17m), one year ago
now2 = datetime(current_year - 1, current_month, 13, 15, 10,
tzinfo=timezone('UTC'))
later2 = now1 + timedelta(minutes=17)
umts2, bytes_recv2, bytes_sent2 = True, 12000, 1245
item2 = self.provider.add_usage_item(now2, later2, bytes_recv2,
bytes_sent2, umts2)
items = self.provider.get_total_usage(now1.date())
self.assertIn(item1, items)
self.assertNotIn(item2, items)
# leave it as we found it
for i in [item1, item2]:
self.provider.delete_usage_item(i)
