2
2
3
3
from tests .utils .fixtures .mock_session import MockSession
4
4
from tests .utils .fixtures .mock_user import MockUser
5
- from workos .users import Users
5
+ from workos .user_management import UserManagement
6
6
7
7
8
- class TestUsers (object ):
8
+ class TestUserManagement (object ):
9
9
@pytest .fixture (autouse = True )
10
10
def setup (self , set_api_key , set_client_id ):
11
- self .users = Users ()
11
+ self .user_management = UserManagement ()
12
12
13
13
@pytest .fixture
14
14
def mock_user (self ):
@@ -31,7 +31,7 @@ def mock_users(self):
31
31
"order" : None ,
32
32
"default_limit" : True ,
33
33
},
34
- "method" : Users .list_users ,
34
+ "method" : UserManagement .list_users ,
35
35
},
36
36
}
37
37
return dict_response
@@ -52,10 +52,10 @@ def mock_users_with_limit(self):
52
52
"after" : None ,
53
53
"order" : None ,
54
54
},
55
- "method" : Users .list_users ,
55
+ "method" : UserManagement .list_users ,
56
56
},
57
57
}
58
- return self .users .construct_from_response (dict_response )
58
+ return self .user_management .construct_from_response (dict_response )
59
59
60
60
@pytest .fixture
61
61
def mock_users_with_default_limit (self ):
@@ -75,10 +75,10 @@ def mock_users_with_default_limit(self):
75
75
"order" : None ,
76
76
"default_limit" : True ,
77
77
},
78
- "method" : Users .list_users ,
78
+ "method" : UserManagement .list_users ,
79
79
},
80
80
}
81
- return self .users .construct_from_response (dict_response )
81
+ return self .user_management .construct_from_response (dict_response )
82
82
83
83
@pytest .fixture
84
84
def mock_users_pagination_response (self ):
@@ -96,7 +96,7 @@ def mock_users_pagination_response(self):
96
96
"order" : None ,
97
97
"default_limit" : True ,
98
98
},
99
- "method" : Users .list_users ,
99
+ "method" : UserManagement .list_users ,
100
100
},
101
101
}
102
102
@@ -134,14 +134,14 @@ def test_create_user(self, mock_user, mock_request_method):
134
134
"password" : "password" ,
135
135
"email_verified" : False ,
136
136
}
137
- user = self .users .create_user (payload )
137
+ user = self .user_management .create_user (payload )
138
138
139
139
assert user ["id" ] == "user_01H7ZGXFP5C6BBQY6Z7277ZCT0"
140
140
141
141
def test_get_user (self , mock_user , capture_and_mock_request ):
142
142
url , request_kwargs = capture_and_mock_request ("get" , mock_user , 200 )
143
143
144
- user = self .users .get_user ("user_01H7ZGXFP5C6BBQY6Z7277ZCT0" )
144
+ user = self .user_management .get_user ("user_01H7ZGXFP5C6BBQY6Z7277ZCT0" )
145
145
146
146
assert url [0 ].endswith ("users/user_01H7ZGXFP5C6BBQY6Z7277ZCT0" )
147
147
assert user ["id" ] == "user_01H7ZGXFP5C6BBQY6Z7277ZCT0"
@@ -177,7 +177,7 @@ def test_list_users_returns_metadata(
177
177
):
178
178
mock_request_method ("get" , mock_users , 200 )
179
179
180
- users = self .users .list_users (
180
+ users = self .user_management .list_users (
181
181
182
182
organization = "foo-corp.com" ,
183
183
)
@@ -189,15 +189,15 @@ def test_list_users_returns_metadata(
189
189
def test_delete_user (self , capture_and_mock_request ):
190
190
url , request_kwargs = capture_and_mock_request ("delete" , None , 200 )
191
191
192
- user = self .users .delete_user ("user_01H7ZGXFP5C6BBQY6Z7277ZCT0" )
192
+ user = self .user_management .delete_user ("user_01H7ZGXFP5C6BBQY6Z7277ZCT0" )
193
193
194
194
assert url [0 ].endswith ("users/user_01H7ZGXFP5C6BBQY6Z7277ZCT0" )
195
195
assert user is None
196
196
197
197
def test_update_user (self , mock_user , capture_and_mock_request ):
198
198
url , request = capture_and_mock_request ("put" , mock_user , 200 )
199
199
200
- user = self .users .update_user (
200
+ user = self .user_management .update_user (
201
201
"user_01H7ZGXFP5C6BBQY6Z7277ZCT0" ,
202
202
{
203
203
"first_name" : "Marcelina" ,
@@ -215,7 +215,7 @@ def test_update_user(self, mock_user, capture_and_mock_request):
215
215
def test_update_user_password (self , mock_user , capture_and_mock_request ):
216
216
url , request = capture_and_mock_request ("put" , mock_user , 200 )
217
217
218
- user = self .users .update_user_password (
218
+ user = self .user_management .update_user_password (
219
219
"user_01H7ZGXFP5C6BBQY6Z7277ZCT0" , "pass_123"
220
220
)
221
221
@@ -226,15 +226,15 @@ def test_update_user_password(self, mock_user, capture_and_mock_request):
226
226
def test_add_user_to_organization (self , capture_and_mock_request , mock_user ):
227
227
url , _ = capture_and_mock_request ("post" , mock_user , 200 )
228
228
229
- user = self .users .add_user_to_organization ("user_123" , "org_123" )
229
+ user = self .user_management .add_user_to_organization ("user_123" , "org_123" )
230
230
231
231
assert url [0 ].endswith ("users/user_123/organization/org_123" )
232
232
assert user ["id" ] == "user_01H7ZGXFP5C6BBQY6Z7277ZCT0"
233
233
234
234
def test_remove_user_from_organization (self , capture_and_mock_request , mock_user ):
235
235
url , _ = capture_and_mock_request ("delete" , mock_user , 200 )
236
236
237
- user = self .users .remove_user_from_organization ("user_123" , "org_123" )
237
+ user = self .user_management .remove_user_from_organization ("user_123" , "org_123" )
238
238
239
239
assert url [0 ].endswith ("users/user_123/organization/org_123" )
240
240
assert user ["id" ] == "user_01H7ZGXFP5C6BBQY6Z7277ZCT0"
@@ -249,7 +249,7 @@ def test_authenticate_with_magic_auth(
249
249
250
250
url , request = capture_and_mock_request ("post" , mock_auth_response , 200 )
251
251
252
- response = self .users .authenticate_with_magic_auth (
252
+ response = self .user_management .authenticate_with_magic_auth (
253
253
code = code ,
254
254
user = user ,
255
255
user_agent = user_agent ,
@@ -279,7 +279,7 @@ def test_authenticate_with_password(
279
279
280
280
url , request = capture_and_mock_request ("post" , mock_auth_response , 200 )
281
281
282
- response = self .users .authenticate_with_password (
282
+ response = self .user_management .authenticate_with_password (
283
283
email = email ,
284
284
password = password ,
285
285
user_agent = user_agent ,
@@ -303,7 +303,7 @@ def test_authenticate_with_code(self, capture_and_mock_request, mock_auth_respon
303
303
304
304
url , request = capture_and_mock_request ("post" , mock_auth_response , 200 )
305
305
306
- response = self .users .authenticate_with_code (
306
+ response = self .user_management .authenticate_with_code (
307
307
code = code ,
308
308
user_agent = user_agent ,
309
309
ip_address = ip_address ,
@@ -328,7 +328,7 @@ def test_create_password_challenge(
328
328
"post" , mock_password_challenge_response , 200
329
329
)
330
330
331
- response = self .users .create_password_reset_challenge (
331
+ response = self .user_management .create_password_reset_challenge (
332
332
email = email ,
333
333
password_reset_url = password_reset_url ,
334
334
)
@@ -345,7 +345,7 @@ def test_complete_password_reset(self, capture_and_mock_request, mock_user):
345
345
346
346
url , request = capture_and_mock_request ("post" , mock_user , 200 )
347
347
348
- response = self .users .complete_password_reset (
348
+ response = self .user_management .complete_password_reset (
349
349
token = token ,
350
350
new_password = new_password ,
351
351
)
@@ -360,7 +360,7 @@ def test_send_verification_email(self, capture_and_mock_request, mock_user):
360
360
361
361
url , _ = capture_and_mock_request ("post" , mock_user , 200 )
362
362
363
- response = self .users .send_verification_email (user = user )
363
+ response = self .user_management .send_verification_email (user = user )
364
364
365
365
assert url [0 ].endswith (
366
366
"users/user_01H7ZGXFP5C6BBQY6Z7277ZCT0/send_verification_email"
@@ -373,7 +373,7 @@ def test_verify_email_code(self, capture_and_mock_request, mock_auth_response):
373
373
374
374
url , request = capture_and_mock_request ("post" , mock_auth_response , 200 )
375
375
376
- response = self .users .verify_email_code (user = user , code = code )
376
+ response = self .user_management .verify_email_code (user = user , code = code )
377
377
378
378
assert url [0 ].endswith ("users/verify_email_code" )
379
379
assert request ["json" ]["user_id" ] == user
@@ -385,7 +385,7 @@ def test_send_magic_auth_code(self, capture_and_mock_request, mock_user):
385
385
386
386
url , request = capture_and_mock_request ("post" , mock_user , 200 )
387
387
388
- response = self .users .send_magic_auth_code (email = email )
388
+ response = self .user_management .send_magic_auth_code (email = email )
389
389
390
390
assert url [0 ].endswith ("users/magic_auth/send" )
391
391
assert request ["json" ]["email_address" ] == email
0 commit comments