def disk_resize(cls, database, new_disk_offering, user): from physical.models import DiskOffering from notification.tasks import TaskRegister disk_offering = DiskOffering.objects.get(id=new_disk_offering) TaskRegister.database_disk_resize(database=database, user=user, disk_offering=disk_offering)
def test_database_disk_resize_with_task_name(self, mocked_delay, mocked_create): TaskRegister.database_disk_resize(database=MagicMock(), user='******', disk_offering='disk_offering', task_name='custom task name') self.assertTrue(mocked_create.called) create_task_param = mocked_create.call_args[0][0] self.assertEqual(create_task_param['task_name'], 'custom task name')
def test_database_disk_resize_with_task_name(self, mocked_delay, mocked_create): TaskRegister.database_disk_resize( database=MagicMock(), user='******', disk_offering='disk_offering', task_name='custom task name' ) self.assertTrue(mocked_create.called) create_task_param = mocked_create.call_args[0][0] self.assertEqual(create_task_param['task_name'], 'custom task name')
def disk_auto_resize(database, current_size, usage_percentage): from notification.tasks import TaskRegister disk = DiskOffering.first_greater_than(current_size + 1024, database.environment) if disk > DiskOffering.last_offering_available_for_auto_resize( environment=database.environment): raise DiskOfferingMaxAutoResize() if database.is_being_used_elsewhere(): raise BusyDatabaseError("") user = AccountUser.objects.get(username="******") task = TaskRegister.database_disk_resize( database=database, user=user, disk_offering=disk, task_name="database_disk_auto_resize", register_user=False, ) email_notifications.disk_resize_notification( database=database, new_disk=disk, usage_percentage=usage_percentage) return task
def disk_auto_resize(database, current_size, usage_percentage): from notification.tasks import TaskRegister disk = DiskOffering.first_greater_than(current_size + 1024) if disk > DiskOffering.last_offering_available_for_auto_resize(): raise DiskOfferingMaxAutoResize() if database.is_being_used_elsewhere(): raise BusyDatabaseError("") user = AccountUser.objects.get(username='******') task = TaskRegister.database_disk_resize( database=database, user=user, disk_offering=disk, task_name='database_disk_auto_resize', register_user=False ) email_notifications.disk_resize_notification( database=database, new_disk=disk, usage_percentage=usage_percentage ) return task