Skip to content

Gelencsér Szabolcs / cloud

Go to a project
Commit 64faefe2 by Dudás Ádám
Options

one: tests for UserCloudDetails

parent c26b7fc1
Showing with 82 additions and 18 deletions
one/tests/test_models.py
from datetime import datetime, timedelta
from django.test import TestCase from django.test import TestCase
from django.contrib.auth.models import User from django.contrib.auth.models import User
from ..models import Disk, Instance, InstanceType, Network, Template, UserCloudDetails from ..models import (Disk, Instance, InstanceType, Network, Share,
Template, UserCloudDetails)
from school.models import Course, Group, Semester
class UserCloudDetailsTestCase(TestCase): class UserCloudDetailsTestCase(TestCase):
def setUp(self): def setUp(self):
user = User.objects.create(username="testuser", user = User.objects.create(username="testuser", password="testpass",
password="testpass", email="test@mail.com", email="test@mail.com", first_name="Test",
first_name="Test", last_name="User") last_name="User")
disk1 = Disk.objects.create(name="testdsk1") disk1 = Disk.objects.create(name="testdsk1")
insttype = InstanceType.objects.create(name="testtype", CPU=4, insttype = InstanceType.objects.create(name="testtype", CPU=4,
RAM=4096, credit=4) RAM=4096, credit=4)
ntwrk = Network.objects.create(name="testntwrk", nat=False, ntwrk = Network.objects.create(name="testntwrk", nat=False,
public=True) public=True)
tmplt1 = Template.objects.create(name="testtmplt1", disk=disk1, tmplt1 = Template.objects.create(name="testtmplt1", disk=disk1,
instance_type=insttype, network=ntwrk, owner=user) instance_type=insttype, owner=user,
network=ntwrk)
tmplt2 = Template.objects.create(name="testtmplt2", disk=disk1, tmplt2 = Template.objects.create(name="testtmplt2", disk=disk1,
instance_type=insttype, network=ntwrk, owner=user) instance_type=insttype, owner=user,
self.testinst1 = Instance.objects.create(owner=user, template=tmplt1, network=ntwrk)
state="ACTIVE") self.testinst1 = Instance.objects.create(owner=user, state="ACTIVE",
self.testinst2 = Instance.objects.create(owner=user, template=tmplt2, template=tmplt1)
state="ACTIVE") self.testinst2 = Instance.objects.create(owner=user, state="ACTIVE",
self.testdetails = UserCloudDetails.objects.get(user=user) template=tmplt2)
self.userdetails = user.cloud_details
date = datetime.now().date()
delta = timedelta(weeks=7)
sem = Semester.objects.create(name="testsem", start=date-delta,
end=date+delta)
course1 = Course.objects.create(code='tccode1', name='testcourse1',
short_name='tc1')
grp1 = Group.objects.create(name="testgroup1", semester=sem,
course=course1)
self.share1 = Share.objects.create(name="testshare1", group=grp1,
template=tmplt1, owner=user,
instance_limit=2, per_user_limit=1)
def test_reset_keys(self):
private_key = self.userdetails.ssh_private_key
public_key = self.userdetails.ssh_key.key
self.userdetails.reset_keys()
self.assertIsNotNone(self.userdetails.ssh_private_key)
self.assertNotEqual(private_key, self.userdetails.ssh_private_key)
self.assertIsNotNone(self.userdetails.ssh_key.key)
self.assertNotEqual(public_key, self.userdetails.ssh_key.key)
def test_reset_keys_without_key(self):
private_key = self.userdetails.ssh_private_key
self.userdetails.ssh_key = None
self.userdetails.save()
self.userdetails.reset_keys()
self.assertIsNotNone(self.userdetails.ssh_private_key)
self.assertNotEqual(private_key, self.userdetails.ssh_private_key)
self.assertIsNotNone(self.userdetails.ssh_key)
def test_reset_smb(self):
smb_password = self.userdetails.smb_password
self.userdetails.reset_smb()
self.assertIsNotNone(self.userdetails.smb_password)
self.assertNotEqual(smb_password, self.userdetails.smb_password)
def test_get_weighted_instance_count(self): def test_get_weighted_instance_count(self):
credits = (self.testinst1.template.instance_type.credit + credits = (self.testinst1.template.instance_type.credit +
self.testinst2.template.instance_type.credit) self.testinst2.template.instance_type.credit)
self.assertEqual(credits, self.testdetails self.assertEqual(credits,
.get_weighted_instance_count()) self.userdetails.get_weighted_instance_count())
self.testinst1.state = "STOPPED" self.testinst1.state = "STOPPED"
self.testinst1.save() self.testinst1.save()
self.assertEqual(self.testinst2.template.instance_type.credit, self.assertEqual(self.testinst2.template.instance_type.credit,
self.testdetails.get_weighted_instance_count()) self.userdetails.get_weighted_instance_count())
self.testinst2.state = "STOPPED" self.testinst2.state = "STOPPED"
self.testinst2.save() self.testinst2.save()
self.assertEqual(0, self.testdetails.get_weighted_instance_count()) self.assertEqual(0, self.userdetails.get_weighted_instance_count())
def test_get_instance_pc(self):
instance_pc = self.userdetails.get_instance_pc()
self.assertTrue(instance_pc >= 0)
def test_get_instance_pc_with_zero_instance_quota(self):
self.userdetails.instance_quota = 0
self.userdetails.save()
self.assertEqual(100, self.userdetails.get_instance_pc())
def test_get_weighted_share_count(self):
share = self.share1
count = share.template.instance_type.credit * share.instance_limit
self.assertEqual(count, self.userdetails.get_weighted_share_count())
def test_get_share_pc(self):
self.userdetails.share_quota = 50
share_pc = self.userdetails.get_share_pc()
self.assertTrue(share_pc >= 0)
def test_get_share_pc_with_zero_share_quota(self):
self.userdetails.share_quota = 0
self.userdetails.save()
self.assertEqual(100, self.userdetails.get_share_pc())
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment