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])