uncloud-mravi/uncloud_vm/tests.py

114 lines
4.3 KiB
Python

import datetime
import parsedatetime
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, RecurringPeriod
User = get_user_model()
cal = parsedatetime.Calendar()
# If you want to check the test database using some GUI/cli tool
# 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='serverx.placey.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):
one_month_later, parse_status = cal.parse("1 month later")
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(*one_month_later[:6], tzinfo=timezone.utc),
recurring_period=RecurringPeriod.PER_MONTH
)
)
# TODO: the logic tested by this test is not implemented yet.
# 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='pending_disk_image', 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='disk_image', 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
)
# TODO: the logic tested by this test is not implemented yet.
# 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='disk_image', is_os_image=True, is_public=True, size_in_gb=10,
# status='active'
# )
# )