Add sample clean() for model + Add tests for uncloud_vm
This commit is contained in:
parent
3228b91038
commit
6c9c63e0da
2 changed files with 153 additions and 38 deletions
|
@ -1,19 +1,24 @@
|
||||||
|
import uuid
|
||||||
|
|
||||||
from django.db import models
|
from django.db import models
|
||||||
from django.contrib.auth import get_user_model
|
from django.contrib.auth import get_user_model
|
||||||
import uuid
|
|
||||||
|
# Uncomment if you override model's clean method
|
||||||
|
# from django.core.exceptions import ValidationError
|
||||||
|
|
||||||
from uncloud_pay.models import Product
|
from uncloud_pay.models import Product
|
||||||
|
|
||||||
STATUS_CHOICES = (
|
STATUS_CHOICES = (
|
||||||
('pending', 'Pending'), # Initial state
|
('pending', 'Pending'), # Initial state
|
||||||
('creating', 'Creating'), # Creating VM/image/etc.
|
('creating', 'Creating'), # Creating VM/image/etc.
|
||||||
('active', 'Active'), # Is usable / active
|
('active', 'Active'), # Is usable / active
|
||||||
('disabled', 'Disabled'), # Is usable, but cannot be used for new things
|
('disabled', 'Disabled'), # Is usable, but cannot be used for new things
|
||||||
('unusable', 'Unusable'), # Has some kind of error
|
('unusable', 'Unusable'), # Has some kind of error
|
||||||
('deleted', 'Deleted'), # Does not exist anymore, only DB entry as a log
|
('deleted', 'Deleted'), # Does not exist anymore, only DB entry as a log
|
||||||
)
|
)
|
||||||
|
|
||||||
STATUS_DEFAULT='pending'
|
STATUS_DEFAULT = 'pending'
|
||||||
|
|
||||||
|
|
||||||
class VMHost(models.Model):
|
class VMHost(models.Model):
|
||||||
uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
||||||
|
@ -30,19 +35,13 @@ class VMHost(models.Model):
|
||||||
# ram that can be used of the server
|
# ram that can be used of the server
|
||||||
usable_ram_in_gb = models.FloatField(default=0)
|
usable_ram_in_gb = models.FloatField(default=0)
|
||||||
|
|
||||||
|
status = models.CharField(max_length=32, choices=STATUS_CHOICES, default=STATUS_DEFAULT)
|
||||||
status = models.CharField(max_length=32,
|
|
||||||
choices=STATUS_CHOICES,
|
|
||||||
default=STATUS_DEFAULT
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class VMProduct(Product):
|
class VMProduct(Product):
|
||||||
vmhost = models.ForeignKey(VMHost,
|
vmhost = models.ForeignKey(
|
||||||
on_delete=models.CASCADE,
|
VMHost, on_delete=models.CASCADE, editable=False, blank=True, null=True
|
||||||
editable=False,
|
)
|
||||||
blank=True,
|
|
||||||
null=True)
|
|
||||||
|
|
||||||
cores = models.IntegerField()
|
cores = models.IntegerField()
|
||||||
ram_in_gb = models.FloatField()
|
ram_in_gb = models.FloatField()
|
||||||
|
@ -60,36 +59,30 @@ class VMDiskImageProduct(models.Model):
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
uuid = models.UUIDField(primary_key=True,
|
uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
||||||
default=uuid.uuid4,
|
owner = models.ForeignKey(get_user_model(), on_delete=models.CASCADE, editable=False)
|
||||||
editable=False)
|
|
||||||
owner = models.ForeignKey(get_user_model(),
|
|
||||||
on_delete=models.CASCADE,
|
|
||||||
editable=False)
|
|
||||||
|
|
||||||
name = models.CharField(max_length=256)
|
name = models.CharField(max_length=256)
|
||||||
is_os_image = models.BooleanField(default=False)
|
is_os_image = models.BooleanField(default=False)
|
||||||
is_public = models.BooleanField(default=False)
|
is_public = models.BooleanField(default=False)
|
||||||
|
|
||||||
size_in_gb = models.FloatField(null=True,
|
size_in_gb = models.FloatField(null=True, blank=True)
|
||||||
blank=True)
|
import_url = models.URLField(null=True, blank=True)
|
||||||
import_url = models.URLField(null=True,
|
|
||||||
blank=True)
|
|
||||||
|
|
||||||
|
storage_class = models.CharField(
|
||||||
storage_class = models.CharField(max_length=32,
|
max_length=32,
|
||||||
choices = (
|
choices=(
|
||||||
('hdd', 'HDD'),
|
('hdd', 'HDD'),
|
||||||
('ssd', 'SSD'),
|
('ssd', 'SSD'),
|
||||||
),
|
),
|
||||||
default='ssd'
|
default='ssd'
|
||||||
)
|
)
|
||||||
|
|
||||||
status = models.CharField(max_length=32,
|
status = models.CharField(
|
||||||
choices=STATUS_CHOICES,
|
max_length=32, choices=STATUS_CHOICES, default=STATUS_DEFAULT
|
||||||
default=STATUS_DEFAULT
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
class VMDiskProduct(models.Model):
|
class VMDiskProduct(models.Model):
|
||||||
"""
|
"""
|
||||||
The VMDiskProduct is attached to a VM.
|
The VMDiskProduct is attached to a VM.
|
||||||
|
@ -104,14 +97,29 @@ class VMDiskProduct(models.Model):
|
||||||
on_delete=models.CASCADE,
|
on_delete=models.CASCADE,
|
||||||
editable=False)
|
editable=False)
|
||||||
|
|
||||||
vm = models.ForeignKey(VMProduct, on_delete=models.CASCADE)
|
vm = models.ForeignKey(VMProduct, on_delete=models.CASCADE)
|
||||||
image = models.ForeignKey(VMDiskImageProduct, on_delete=models.CASCADE)
|
image = models.ForeignKey(VMDiskImageProduct, on_delete=models.CASCADE)
|
||||||
|
|
||||||
size_in_gb = models.FloatField(blank=True)
|
size_in_gb = models.FloatField(blank=True)
|
||||||
|
|
||||||
|
# Sample code for clean method
|
||||||
|
|
||||||
|
# Ensures that a VMDiskProduct can only be created from a VMDiskImageProduct
|
||||||
|
# that is in status 'active'
|
||||||
|
|
||||||
|
# def clean(self):
|
||||||
|
# if self.image.status != 'active':
|
||||||
|
# raise ValidationError({
|
||||||
|
# 'image': 'VM Disk must be created from an active disk image.'
|
||||||
|
# })
|
||||||
|
|
||||||
|
def save(self, *args, **kwargs):
|
||||||
|
self.full_clean()
|
||||||
|
super().save(*args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
class VMNetworkCard(models.Model):
|
class VMNetworkCard(models.Model):
|
||||||
vm = models.ForeignKey(VMProduct, on_delete=models.CASCADE)
|
vm = models.ForeignKey(VMProduct, on_delete=models.CASCADE)
|
||||||
|
|
||||||
mac_address = models.IntegerField()
|
mac_address = models.IntegerField()
|
||||||
|
|
||||||
|
|
107
uncloud/uncloud_vm/tests.py
Normal file
107
uncloud/uncloud_vm/tests.py
Normal file
|
@ -0,0 +1,107 @@
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
from django.test import TestCase
|
||||||
|
from django.contrib.auth import get_user_model
|
||||||
|
from django.utils import timezone
|
||||||
|
from django.core.exceptions import ValidationError
|
||||||
|
|
||||||
|
from uncloud_vm.models import VMDiskImageProduct, VMDiskProduct, VMProduct, VMHost
|
||||||
|
from uncloud_pay.models import Order
|
||||||
|
|
||||||
|
User = get_user_model()
|
||||||
|
|
||||||
|
|
||||||
|
# If you want to check the test database then use the following connecting parameters
|
||||||
|
|
||||||
|
# host: localhost
|
||||||
|
# database: test_uncloud
|
||||||
|
# user: root
|
||||||
|
# password:
|
||||||
|
# port: 5432
|
||||||
|
|
||||||
|
class VMTestCase(TestCase):
|
||||||
|
@classmethod
|
||||||
|
def setUpClass(cls):
|
||||||
|
# Setup vm host
|
||||||
|
cls.vm_host, created = VMHost.objects.get_or_create(
|
||||||
|
hostname='server1.place11.ungleich.ch', physical_cores=32, usable_cores=320,
|
||||||
|
usable_ram_in_gb=512.0, status='active'
|
||||||
|
)
|
||||||
|
super().setUpClass()
|
||||||
|
|
||||||
|
def setUp(self) -> None:
|
||||||
|
# Setup two users as it is common to test with different user
|
||||||
|
self.user = User.objects.create_user(
|
||||||
|
username='testuser', email='test@test.com', first_name='Test', last_name='User'
|
||||||
|
)
|
||||||
|
self.user2 = User.objects.create_user(
|
||||||
|
username='Meow', email='meow123@test.com', first_name='Meow', last_name='Cat'
|
||||||
|
)
|
||||||
|
super().setUp()
|
||||||
|
|
||||||
|
def create_sample_vm(self, owner):
|
||||||
|
return VMProduct.objects.create(
|
||||||
|
vmhost=self.vm_host, cores=2, ram_in_gb=4, owner=owner,
|
||||||
|
order=Order.objects.create(
|
||||||
|
owner=owner,
|
||||||
|
creation_date=datetime.datetime.now(tz=timezone.utc),
|
||||||
|
starting_date=datetime.datetime.now(tz=timezone.utc),
|
||||||
|
ending_date=datetime.datetime(2020, 4, 2, tzinfo=timezone.utc),
|
||||||
|
recurring_price=4.0, one_time_price=5.0, recurring_period='per_month'
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_disk_product(self):
|
||||||
|
"""Ensures that a VMDiskProduct can only be created from a VMDiskImageProduct
|
||||||
|
that is in status 'active'"""
|
||||||
|
|
||||||
|
vm = self.create_sample_vm(owner=self.user)
|
||||||
|
|
||||||
|
pending_disk_image = VMDiskImageProduct.objects.create(
|
||||||
|
owner=self.user, name='alpine3.11', is_os_image=True, is_public=True, size_in_gb=10,
|
||||||
|
status='pending'
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
vm_disk_product = VMDiskProduct.objects.create(
|
||||||
|
owner=self.user, vm=vm, image=pending_disk_image, size_in_gb=10
|
||||||
|
)
|
||||||
|
except ValidationError:
|
||||||
|
vm_disk_product = None
|
||||||
|
|
||||||
|
self.assertIsNone(
|
||||||
|
vm_disk_product,
|
||||||
|
msg='VMDiskProduct created with disk image whose status is not active.'
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_vm_disk_product_creation(self):
|
||||||
|
"""Ensure that a user can only create a VMDiskProduct for an existing VM"""
|
||||||
|
|
||||||
|
disk_image = VMDiskImageProduct.objects.create(
|
||||||
|
owner=self.user, name='alpine3.11', is_os_image=True, is_public=True, size_in_gb=10,
|
||||||
|
status='active'
|
||||||
|
)
|
||||||
|
|
||||||
|
with self.assertRaises(ValidationError, msg='User created a VMDiskProduct for non-existing VM'):
|
||||||
|
# Create VMProduct object but don't save it in database
|
||||||
|
vm = VMProduct()
|
||||||
|
|
||||||
|
vm_disk_product = VMDiskProduct.objects.create(
|
||||||
|
owner=self.user, vm=vm, image=disk_image, size_in_gb=10
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_vm_disk_product_creation_for_someone_else(self):
|
||||||
|
"""Ensure that a user can only create a VMDiskProduct for his/her own VM"""
|
||||||
|
|
||||||
|
# Create a VM which is ownership of self.user2
|
||||||
|
someone_else_vm = self.create_sample_vm(owner=self.user2)
|
||||||
|
|
||||||
|
# 'self.user' would try to create a VMDiskProduct for 'user2's VM
|
||||||
|
with self.assertRaises(ValidationError, msg='User created a VMDiskProduct for someone else VM.'):
|
||||||
|
vm_disk_product = VMDiskProduct.objects.create(
|
||||||
|
owner=self.user, vm=someone_else_vm,
|
||||||
|
size_in_gb=10,
|
||||||
|
image=VMDiskImageProduct.objects.create(
|
||||||
|
owner=self.user, name='alpine3.11', is_os_image=True, is_public=True, size_in_gb=10,
|
||||||
|
status='active'
|
||||||
|
)
|
||||||
|
)
|
Loading…
Reference in a new issue