/
tests.py
60 lines (53 loc) · 2.16 KB
/
tests.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import unittest
import application
import schema
import settings
import utils
import logging
import time
class CoreTestCase(unittest.TestCase):
def setUp(self):
self.client = application.app.test_client()
self.db = application.get_db_connection()
salt = int(time.time())
self._test_user = 'test_user_{0}'.format(salt)
self._test_user_password = 'test123'
self._test_user_email = '{0}@lumentica.com'.format(self._test_user)
self._test_user_role = 'testrole_{0}'.format(salt)
self._project_name = 'test_project_{0}'.format(salt)
self._project_type = 'live'
self._project_distro_id = 0
self._project_version = '1'
self._project_arch = 'i386'
self._project_packages = ['blender', 'gimp']
def test_index(self):
resp = self.client.get('/')
assert(resp.status_code == 200 or resp.status_code == 302)
def test_create_user(self):
assert utils.create_user(self._test_user, self._test_user_email, \
self._test_user_password, self._test_user_role)
user = utils.get_user(self._test_user)
assert user != None
assert user['username'] == self._test_user
assert user['email'] == self._test_user_email
assert user['role'] == self._test_user_role
assert utils.delete_user(self._test_user)
def test_create_role(self):
assert utils.create_role(self._test_user_role)
assert utils.delete_role(self._test_user_role)
def test_toggle_user(self):
assert utils.create_user(self._test_user, password=self._test_user_password, role=self._test_user_role)
assert utils.toggle_user(self._test_user, False)
user = utils.get_user(self._test_user)
assert user != None
assert user['enabled'] == False
assert utils.toggle_user(self._test_user, True)
user = utils.get_user(self._test_user)
assert user != None
assert user['enabled'] == True
assert utils.delete_user(self._test_user)
def tearDown(self):
if utils.get_user(self._test_user):
utils.delete_user(self._test_user)
if __name__=="__main__":
unittest.main()