def test_update_usage_when_there_is_no_value(self):
update_usage(
self.device,
self.venture,
self._get_usages()['virtual_cores'],
datetime.today(),
None,
)
self.assertItemsEqual(DailyUsage.objects.all(), [])
def test_update_usage(self):
usages = self._get_usages()
update_usage(
self.device,
self.venture,
usages['virtual_cores'],
date.today(),
1,
)
daily_usages = DailyUsage.objects.all()
self.assertEqual(daily_usages.count(), 1)
self.assertEqual(daily_usages[0].pricing_device, self.device)
self.assertEqual(daily_usages[0].pricing_venture, self.venture)
self.assertEqual(daily_usages[0].type, usages['virtual_cores'])
self.assertEqual(daily_usages[0].value, 1)
