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)