134 lines
4.9 KiB
Python
134 lines
4.9 KiB
Python
import random
|
|
import string
|
|
|
|
from django.test import TestCase
|
|
|
|
from .models import OpenNebulaManager
|
|
from .serializers import VirtualMachineSerializer
|
|
from utils.models import CustomUser
|
|
|
|
|
|
class OpenNebulaManagerTestCases(TestCase):
|
|
"""This class defines the test suite for the opennebula manager model."""
|
|
|
|
def setUp(self):
|
|
"""Define the test client and other test variables."""
|
|
self.email = '{}@ungleich.ch'.format(''.join(random.choices(string.ascii_uppercase, k=10)))
|
|
self.password = ''.join(random.choices(string.ascii_uppercase + string.digits, k=20))
|
|
|
|
self.user = CustomUser.objects.create(name='test', email=self.email,
|
|
password=self.password)
|
|
|
|
self.vm_specs = {}
|
|
self.vm_specs['cpu'] = 1
|
|
self.vm_specs['memory'] = 2
|
|
self.vm_specs['disk_size'] = 10
|
|
|
|
self.manager = OpenNebulaManager()
|
|
|
|
def test_connect_to_server(self):
|
|
"""Test the opennebula manager can connect to a server."""
|
|
try:
|
|
ver = self.manager.oneadmin_client.version()
|
|
except:
|
|
ver = None
|
|
self.assertTrue(ver is not None)
|
|
|
|
def test_get_user(self):
|
|
"""Test the opennebula manager can get a existing user."""
|
|
self.manager.create_user(self.user)
|
|
user = self.manager._get_user(self.user)
|
|
name = user.name
|
|
self.assertNotEqual(name, None)
|
|
|
|
def test_create_and_delete_user(self):
|
|
"""Test the opennebula manager can create and delete a new user."""
|
|
old_count = len(self.manager._get_user_pool())
|
|
self.manager = OpenNebulaManager(email=self.email,
|
|
password=self.password)
|
|
user_pool = self.manager._get_user_pool()
|
|
new_count = len(user_pool)
|
|
# Remove the user afterwards
|
|
user = user_pool.get_by_name(self.email)
|
|
user.delete()
|
|
|
|
self.assertNotEqual(old_count, new_count)
|
|
|
|
def test_user_can_login(self):
|
|
""" Test the manager can login to a new created user"""
|
|
self.manager.create_user(self.user)
|
|
user = self.manager._get_user(self.user)
|
|
client = self.manager._get_client(self.user)
|
|
version = client.version()
|
|
|
|
# Cleanup
|
|
user.delete()
|
|
self.assertNotEqual(version, None)
|
|
|
|
def test_add_public_key_to_user(self):
|
|
""" Test the manager can add a new public key to an user """
|
|
self.manager.create_user(self.user)
|
|
user = self.manager._get_user(self.user)
|
|
public_key = 'test'
|
|
self.manager.add_public_key(self.user, public_key)
|
|
# Fetch new user information from opennebula
|
|
user.info()
|
|
user_public_key = user.template.ssh_public_key
|
|
# Cleanup
|
|
user.delete()
|
|
|
|
self.assertEqual(user_public_key, public_key)
|
|
|
|
def test_append_public_key_to_user(self):
|
|
""" Test the manager can append a new public key to an user """
|
|
self.manager.create_user(self.user)
|
|
user = self.manager._get_user(self.user)
|
|
public_key = 'test'
|
|
self.manager.add_public_key(self.user, public_key)
|
|
# Fetch new user information from opennebula
|
|
user.info()
|
|
old_public_key = user.template.ssh_public_key
|
|
self.manager.add_public_key(self.user, public_key, merge=True)
|
|
user.info()
|
|
new_public_key = user.template.ssh_public_key
|
|
# Cleanup
|
|
user.delete()
|
|
|
|
self.assertEqual(new_public_key, '{}\n{}'.format(old_public_key,
|
|
public_key))
|
|
|
|
def test_remove_public_key_to_user(self):
|
|
""" Test the manager can remove a public key from an user """
|
|
self.manager.create_user(self.user)
|
|
user = self.manager._get_user(self.user)
|
|
public_key = 'test'
|
|
self.manager.add_public_key(self.user, public_key)
|
|
self.manager.add_public_key(self.user, public_key, merge=True)
|
|
user.info()
|
|
old_public_key = user.template.ssh_public_key
|
|
self.manager.remove_public_key(self.user, public_key)
|
|
user.info()
|
|
new_public_key = user.template.ssh_public_key
|
|
# Cleanup
|
|
user.delete()
|
|
|
|
self.assertEqual(new_public_key,
|
|
old_public_key.replace('{}\n'.format(public_key), '', 1))
|
|
|
|
def test_requires_ssh_key_for_new_vm(self):
|
|
"""Test the opennebula manager requires the user to have a ssh key when
|
|
creating a new vm"""
|
|
|
|
|
|
class VirtualMachineSerializerTestCase(TestCase):
|
|
def setUp(self):
|
|
"""Define the test client and other test variables."""
|
|
self.manager = OpenNebulaManager(email=None, password=None)
|
|
|
|
def test_serializer_strips_of_public(self):
|
|
""" Test the serialized virtual machine object contains no 'public-'."""
|
|
|
|
for vm in self.manager.get_vms():
|
|
serialized = VirtualMachineSerializer(vm)
|
|
self.assertEqual(serialized.data.get('name'), vm.name.strip('public-'))
|
|
break
|