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
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
import asyncio
import functools
import random
import time
from testing import Client
from testing import default_test_setup
from testing import gen_data
from testing import gen_points
from testing import gen_series
from testing import InsertError
from testing import PoolError
from testing import QueryError
from testing import run_test
from testing import Series
from testing import Server
from testing import ServerError
from testing import SiriDB
from testing import TestBase
from testing import UserAuthError
from testing import parse_args
class TestUser(TestBase):
title = 'Test user object'
@default_test_setup(3)
async def run(self):
a = await self.client0.connect()
result = await self.client0.query('list users ')
self.assertEqual(result.pop('users'), [['iris', 'full']])
with self.assertRaises(QueryError):
await self.client0.query('create user "sasientje" ')
with self.assertRaisesRegex(
QueryError,
'User name should be at least 2 characters.'):
await self.client0.query('create user "s" set password "123456" ')
with self.assertRaisesRegex(
QueryError,
'User name contains illegal characters.*'):
await self.client0.query('create user " " set password "123456" ')
with self.assertRaisesRegex(
QueryError,
'Password should be at least 4 characters.'):
await self.client0.query('create user "aa" set password "123" ')
result = await self.client0.query(
'create user "sasientje" set password "blabla" ')
self.assertEqual(
result.pop('success_msg'),
"Successfully created user 'sasientje'.")
result = await self.client0.query('list users where access < modify ')
self.assertEqual(result.pop('users'), [['sasientje', 'no access']])
result = await self.client0.query('grant modify to user "sasientje" ')
self.assertEqual(
result.pop('success_msg'),
"Successfully granted permissions to user 'sasientje'.")
await self.db.add_replica(self.server1, 0)
await self.assertIsRunning(self.db, self.client0, timeout=20)
await asyncio.sleep(5)
await self.db.add_pool(self.server2)
await self.client1.connect()
await self.client2.connect()
await asyncio.sleep(20)
result = await self.client1.query('list users where access < full ')
self.assertEqual(result.pop('users'), [['sasientje', 'modify']])
result = await self.client1.query(
'revoke write from user "sasientje" ')
self.assertEqual(
result.pop('success_msg'),
"Successfully revoked permissions from user 'sasientje'.")
result = await self.client1.query(
'grant show, count to user "sasientje"')
result = await self.client1.query(
'list users where access < modify ')
self.assertEqual(
result.pop('users'),
[['sasientje', 'alter, count, drop and show']])
result = await self.client1.query(
'create user "pee" set password "hihihaha" ')
time.sleep(0.1)
result = await self.client0.query('list users where name ~ "p"')
self.assertEqual(result.pop('users'), [['pee', 'no access']])
self.client0.close()
result = await self.server0.stop()
self.assertTrue(result)
with self.assertRaisesRegex(
QueryError,
"Password should be at least 4 characters."):
result = await self.client1.query(
'alter user "sasientje" set password "dag" ')
result = await self.client1.query(
'alter user "sasientje" set password "dagdag"')
await self.server0.start(sleep=35)
self.client0 = Client(
self.db,
self.server0,
username="sasientje",
password="dagdag")
await self.client0.connect()
result = await self.client0.query("show who_am_i ")
self.assertEqual(result['data'][0]['value'], 'sasientje')
with self.assertRaisesRegex(
UserAuthError,
"Access denied. User 'sasientje' has no 'insert' privileges."):
result = await self.client0.insert({'no access test': [[1, 1.0]]})
result = await self.client1.query('drop user "sasientje" ')
self.assertEqual(
result.pop('success_msg'),
"Successfully dropped user 'sasientje'.")
time.sleep(0.1)
for client in (self.client0, self.client1, self.client2):
result = await client.query('count users')
self.assertEqual(
result.pop('users'), 2, msg='Expecting 2 users (iris and pee)')
result = await self.client0.query('count users where name == "pee"')
self.assertEqual(result.pop('users'), 1, msg='Expecting 1 user (pee)')
with self.assertRaisesRegex(
UserAuthError,
"Access denied. User 'sasientje' has no 'grant' privileges."):
result = await self.client0.query('grant full to user "pee" ')
with self.assertRaisesRegex(
QueryError,
"User name should be at least 2 characters."):
result = await self.client1.query('alter user "pee" set name "p" ')
with self.assertRaisesRegex(
QueryError,
"^User name contains illegal characters.*"):
result = await self.client1.query(
'alter user "pee" set name " p " ')
with self.assertRaisesRegex(
QueryError,
"User 'iris' already exists."):
result = await self.client1.query(
'alter user "pee" set name "iris" ')
with self.assertRaisesRegex(
QueryError,
"User 'iris' already exists."):
result = await self.client1.query(
'alter user "pee" set name "iris" ')
with self.assertRaisesRegex(
QueryError,
"Cannot find user: 'Pee'"):
result = await self.client1.query(
'alter user "Pee" set name "PPP" ')
result = await self.client1.query('alter user "pee" set name "Pee"')
self.assertEqual(
result.pop('success_msg'), "Successfully updated user 'Pee'.")
time.sleep(0.1)
result = await self.client2.query('list users where name == "Pee" ')
self.assertEqual(result.pop('users'), [['Pee', 'no access']])
self.client0.close()
self.client1.close()
self.client2.close()
if __name__ == '__main__':
parse_args()
run_test(TestUser())
|
