def test_update_when_venture_id_is_none(self):
data = {
'device_id': 1,
'virtual_cores': 1,
'virtual_disk': 1,
'virtual_memory': 1,
}
update(data, self._get_usages(), date.today())
daily_usages = DailyUsage.objects.all()
self.assertEqual(daily_usages.count(), 0)
for daily_usage in daily_usages:
self.assertEqual(daily_usage.pricing_venture, None)
def test_update_when_device_id_is_none(self):
update({}, self._get_usages(), date.today())
self.assertItemsEqual(Device.objects.all(), [self.device])
