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)