def first_greater_than(cls, base_size, exclude_id=None):
disks = DiskOffering.objects.filter(size_kb__gt=base_size).exclude(
id=exclude_id).order_by('size_kb')
if not disks:
raise NoDiskOfferingGreaterError(base_size)
return disks[0]
def test_no_disk_offering_greater(self):
size = 789
message = 'No disk offering greater than {}kb'.format(size)
no_disk_offering = NoDiskOfferingGreaterError(size=size)
self.assertEqual(no_disk_offering.message, message)