-
Notifications
You must be signed in to change notification settings - Fork 414
/
test_user_model.py
138 lines (115 loc) · 5.54 KB
/
test_user_model.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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import logging
import unittest
from unittest.mock import *
from typing import List
import io
import pytest
import tableauserverclient as TSC
class UserModelTests(unittest.TestCase):
def test_invalid_name(self):
self.assertRaises(ValueError, TSC.UserItem, None, TSC.UserItem.Roles.Publisher)
self.assertRaises(ValueError, TSC.UserItem, "", TSC.UserItem.Roles.Publisher)
user = TSC.UserItem("me", TSC.UserItem.Roles.Publisher)
with self.assertRaises(ValueError):
user.name = None
with self.assertRaises(ValueError):
user.name = ""
def test_invalid_auth_setting(self):
user = TSC.UserItem("me", TSC.UserItem.Roles.Publisher)
with self.assertRaises(ValueError):
user.auth_setting = "Hello"
def test_invalid_site_role(self):
user = TSC.UserItem("me", TSC.UserItem.Roles.Publisher)
with self.assertRaises(ValueError):
user.site_role = "Hello"
class UserDataTest(unittest.TestCase):
logger = logging.getLogger("UserDataTest")
role_inputs = [
["creator", "system", "yes", "SiteAdministrator"],
["None", "system", "no", "SiteAdministrator"],
["explorer", "SysTEm", "no", "SiteAdministrator"],
["creator", "site", "yes", "SiteAdministratorCreator"],
["explorer", "site", "yes", "SiteAdministratorExplorer"],
["creator", "SITE", "no", "SiteAdministratorCreator"],
["creator", "none", "yes", "Creator"],
["explorer", "none", "yes", "ExplorerCanPublish"],
["viewer", "None", "no", "Viewer"],
["explorer", "no", "yes", "ExplorerCanPublish"],
["EXPLORER", "noNO", "yes", "ExplorerCanPublish"],
["explorer", "no", "no", "Explorer"],
["unlicensed", "none", "no", "Unlicensed"],
["Chef", "none", "yes", "Unlicensed"],
["yes", "yes", "yes", "Unlicensed"],
]
valid_import_content = [
"username, pword, fname, creator, site, yes, email",
"username, pword, fname, explorer, none, no, email",
"",
"u",
"p",
]
valid_username_content = ["jfitzgerald@tableau.com"]
usernames = [
"valid",
"valid@email.com",
"domain/valid",
"domain/valid@tmail.com",
"va!@#$%^&*()lid",
"in@v@lid",
"in valid",
"",
]
def test_validate_usernames(self):
TSC.UserItem.validate_username_or_throw(UserDataTest.usernames[0])
TSC.UserItem.validate_username_or_throw(UserDataTest.usernames[1])
TSC.UserItem.validate_username_or_throw(UserDataTest.usernames[2])
TSC.UserItem.validate_username_or_throw(UserDataTest.usernames[3])
TSC.UserItem.validate_username_or_throw(UserDataTest.usernames[4])
with self.assertRaises(AttributeError):
TSC.UserItem.validate_username_or_throw(UserDataTest.usernames[5])
with self.assertRaises(AttributeError):
TSC.UserItem.validate_username_or_throw(UserDataTest.usernames[6])
def test_evaluate_role(self):
for line in UserDataTest.role_inputs:
actual = TSC.UserItem.CSVImport._evaluate_site_role(line[0], line[1], line[2])
assert actual == line[3], line + [actual]
def test_get_user_detail_empty_line(self):
test_line = ""
test_user = TSC.UserItem.CSVImport.create_user_from_line(test_line)
assert test_user is None
def test_get_user_detail_standard(self):
test_line = "username, pword, fname, license, admin, pub, email"
test_user: TSC.UserItem = TSC.UserItem.CSVImport.create_user_from_line(test_line)
assert test_user.name == "username", test_user.name
assert test_user.fullname == "fname", test_user.fullname
assert test_user.site_role == "Unlicensed", test_user.site_role
assert test_user.email == "email", test_user.email
def test_get_user_details_only_username(self):
test_line = "username"
test_user: TSC.UserItem = TSC.UserItem.CSVImport.create_user_from_line(test_line)
def test_populate_user_details_only_some(self):
values = "username, , , creator, admin"
user = TSC.UserItem.CSVImport.create_user_from_line(values)
assert user.name == "username"
def test_validate_user_detail_standard(self):
test_line = "username, pword, fname, creator, site, 1, email"
TSC.UserItem.CSVImport._validate_import_line_or_throw(test_line, UserDataTest.logger)
TSC.UserItem.CSVImport.create_user_from_line(test_line)
# for file handling
def _mock_file_content(self, content: List[str]) -> io.TextIOWrapper:
# the empty string represents EOF
# the tests run through the file twice, first to validate then to fetch
mock = MagicMock(io.TextIOWrapper)
content.append("") # EOF
mock.readline.side_effect = content
mock.name = "file-mock"
return mock
def test_validate_import_file(self):
test_data = self._mock_file_content(UserDataTest.valid_import_content)
valid, invalid = TSC.UserItem.CSVImport.validate_file_for_import(test_data, UserDataTest.logger)
assert valid == 2, "Expected two lines to be parsed, got {}".format(valid)
assert invalid == [], "Expected no failures, got {}".format(invalid)
def test_validate_usernames_file(self):
test_data = self._mock_file_content(UserDataTest.usernames)
valid, invalid = TSC.UserItem.CSVImport.validate_file_for_import(test_data, UserDataTest.logger)
assert valid == 5, "Exactly 5 of the lines were valid, counted {}".format(valid + invalid)