/
users_test.py
96 lines (82 loc) · 3.78 KB
/
users_test.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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
import unittest # importing the unitttest module
from users import User # importing the users class
from accounts import Account
class TestUsers (unittest.TestCase):
# this class defines test cases for user class behaviours
# TestCase class helps in creating testcases
def setUp(self):
self.new_user = User(
'Vikki', 'Ireri', 'wambsviki@gmail.com', 'akisijui', [])
self.new_account = Account('Instagram', 'https://www.instagram.com/',
'wambsviki@gmail.com', '12345')
self.new_account1 = Account('Facebook', 'https://www.facebook.com/',
'wambsviki@gmail.com', '67890')
def test_init(self):
self.assertEqual(self.new_user.first_name, 'Vikki')
self.assertEqual(self.new_user.last_name, 'Ireri')
self.assertEqual(self.new_user.email, 'wambsviki@gmail.com')
self.assertEqual(self.new_user.password, 'akisijui')
self.assertEqual(self.new_user.accounts, [])
def test_save_user(self):
self.new_user.save_user()
# self.new_acAccountcount.save_account()
# print(self.new_user.save_user())
def test_add_acc_to_user(self):
user = self.new_user.save_user()
acc1 = self.new_account1.save_account()
acc = self.new_account.new_account(8)
self.new_user.add_acc_to_user(acc1, user)
self.new_user.add_acc_to_user(acc, user)
self.assertEqual(len(user['wambsviki@gmail.com']['accounts']), 2)
# print(user)
def test_add_user_to_list(self):
user = self.new_user.save_user()
acc1 = self.new_account1.save_account()
acc = self.new_account.save_account()
self.new_user.add_acc_to_user(acc1, user)
self.new_user.add_acc_to_user(acc, user)
self.new_user.add_user_to_list(user)
self.assertEqual(len(User.users_list), 1)
# print(User.users_list)
def test_delete_user_from_list(self):
user = self.new_user.save_user()
acc1 = self.new_account1.save_account()
acc = self.new_account.save_account()
self.new_user.add_acc_to_user(acc1, user)
self.new_user.add_acc_to_user(acc, user)
self.new_user.add_user_to_list(user)
self.new_user.delete_user_from_list()
self.assertEqual(len(User.users_list), 1)
# print(User.users_list)
def test_delete_account(self):
user = self.new_user.save_user()
acc1 = self.new_account1.save_account()
acc = self.new_account.save_account()
self.new_user.add_acc_to_user(acc1, user)
self.new_user.add_acc_to_user(acc, user)
self.new_user.add_user_to_list(user)
self.new_user.delete_account('Facebook', user)
# print(self.new_user.delete_account('Facebook', user))
def test_find_user(self):
user = self.new_user.save_user()
acc1 = self.new_account1.save_account()
acc = self.new_account.save_account()
self.new_user.add_acc_to_user(acc1, user)
self.new_user.add_acc_to_user(acc, user)
self.new_user.add_user_to_list(user)
self.new_user.find_user_by_email('wambsviki@gmail.com')
def test_find_account(self):
user = self.new_user.save_user()
acc1 = self.new_account1.save_account()
acc = self.new_account.save_account()
self.new_user.add_acc_to_user(acc1, user)
self.new_user.add_acc_to_user(acc, user)
self.new_user.add_user_to_list(user)
self.new_user.find_account_by_name(user, 'Facebook')
# print(user['wambsviki@gmail.com']['accounts'])
def test_find_by_name(self):
user = self.new_user.save_user()
acc1 = self.new_account1.save_account()
self.new_user.find_account_by_name(acc1, 'Instagram')
if __name__ == '__main__':
unittest.main()