Skip to content
Toggle navigation
P
Projects
G
Groups
S
Snippets
Help
Fukász Rómeó Ervin
/
cloud
This project
Loading...
Sign in
Toggle navigation
Go to a project
Project
Repository
Issues
0
Merge Requests
0
Pipelines
Wiki
Members
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Commit
64faefe2
authored
Jul 06, 2013
by
Dudás Ádám
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
one: tests for UserCloudDetails
parent
c26b7fc1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
82 additions
and
18 deletions
+82
-18
one/tests/test_models.py
+82
-18
No files found.
one/tests/test_models.py
View file @
64faefe2
from
datetime
import
datetime
,
timedelta
from
django.test
import
TestCase
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
):
def
setUp
(
self
):
user
=
User
.
objects
.
create
(
username
=
"testuser"
,
password
=
"testpass"
,
email
=
"test@mail.com
"
,
first_name
=
"Test"
,
last_name
=
"User"
)
user
=
User
.
objects
.
create
(
username
=
"testuser"
,
password
=
"testpass"
,
email
=
"test@mail.com"
,
first_name
=
"Test
"
,
last_name
=
"User"
)
disk1
=
Disk
.
objects
.
create
(
name
=
"testdsk1"
)
insttype
=
InstanceType
.
objects
.
create
(
name
=
"testtype"
,
CPU
=
4
,
RAM
=
4096
,
credit
=
4
)
RAM
=
4096
,
credit
=
4
)
ntwrk
=
Network
.
objects
.
create
(
name
=
"testntwrk"
,
nat
=
False
,
public
=
True
)
public
=
True
)
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
,
instance_type
=
insttype
,
network
=
ntwrk
,
owner
=
user
)
self
.
testinst1
=
Instance
.
objects
.
create
(
owner
=
user
,
template
=
tmplt1
,
state
=
"ACTIVE"
)
self
.
testinst2
=
Instance
.
objects
.
create
(
owner
=
user
,
template
=
tmplt2
,
state
=
"ACTIVE"
)
self
.
testdetails
=
UserCloudDetails
.
objects
.
get
(
user
=
user
)
instance_type
=
insttype
,
owner
=
user
,
network
=
ntwrk
)
self
.
testinst1
=
Instance
.
objects
.
create
(
owner
=
user
,
state
=
"ACTIVE"
,
template
=
tmplt1
)
self
.
testinst2
=
Instance
.
objects
.
create
(
owner
=
user
,
state
=
"ACTIVE"
,
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
):
credits
=
(
self
.
testinst1
.
template
.
instance_type
.
credit
+
self
.
testinst2
.
template
.
instance_type
.
credit
)
self
.
assertEqual
(
credits
,
self
.
testdetails
.
get_weighted_instance_count
())
self
.
testinst2
.
template
.
instance_type
.
credit
)
self
.
assertEqual
(
credits
,
self
.
userdetails
.
get_weighted_instance_count
())
self
.
testinst1
.
state
=
"STOPPED"
self
.
testinst1
.
save
()
self
.
assertEqual
(
self
.
testinst2
.
template
.
instance_type
.
credit
,
self
.
test
details
.
get_weighted_instance_count
())
self
.
user
details
.
get_weighted_instance_count
())
self
.
testinst2
.
state
=
"STOPPED"
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
())
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment