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