6
6
)
7
7
from labelbox .schema .user_group import UserGroup , UserGroupColor
8
8
9
- from libs .labelbox .tests .conftest import AdminClient
9
+ from libs .labelbox .tests .conftest import AdminClient , Environ
10
10
import os
11
11
12
12
data = faker .Faker ()
@@ -32,8 +32,9 @@ def user_group(client):
32
32
33
33
@pytest .fixture (scope = "module" )
34
34
def test_member (client , current_member , admin_client : AdminClient ):
35
+ admin_client = admin_client (Environ .STAGING )
35
36
admin_client ._create_user (client .get_organization ().uid )
36
- members = list (Member (client = client ).get_members ())
37
+ members = list (Member (client = client ).get_members (search = "email@email.com" ))
37
38
test_member = None
38
39
for member in members :
39
40
if member .id != current_member .id :
@@ -56,7 +57,10 @@ def test_throw_error_when_deleting_self(current_member):
56
57
current_member .delete ()
57
58
58
59
59
- @pytest .mark .skipif (condition = os .environ ["LABELBOX_TEST_ENVIRON" ] != "staging" )
60
+ @pytest .mark .skipif (
61
+ condition = os .environ ["LABELBOX_TEST_ENVIRON" ] != "staging" ,
62
+ reason = "admin client only works in staging" ,
63
+ )
60
64
def test_update_member (client , test_member , project_pack , user_group ):
61
65
labeler_role = client .get_roles ()["LABELER" ]
62
66
reviewer_role = client .get_roles ()["REVIEWER" ]
@@ -103,7 +107,10 @@ def test_update_member(client, test_member, project_pack, user_group):
103
107
assert updated_member .can_access_all_projects
104
108
105
109
106
- @pytest .mark .skipif (condition = os .environ ["LABELBOX_TEST_ENVIRON" ] != "staging" )
110
+ @pytest .mark .skipif (
111
+ condition = os .environ ["LABELBOX_TEST_ENVIRON" ] != "staging" ,
112
+ reason = "admin client only works in staging" ,
113
+ )
107
114
def test_get_members (test_member , current_member , client ):
108
115
member_ids = [
109
116
member .id
@@ -120,13 +127,16 @@ def test_get_members(test_member, current_member, client):
120
127
assert current_member .id in member_ids
121
128
122
129
123
- @pytest .mark .skipif (condition = os .environ ["LABELBOX_TEST_ENVIRON" ] != "staging" )
124
- def test_delete_member (test_member , current_member , client ):
130
+ @pytest .mark .skipif (
131
+ condition = os .environ ["LABELBOX_TEST_ENVIRON" ] != "staging" ,
132
+ reason = "admin client only works in staging" ,
133
+ )
134
+ def test_delete_member (test_member , current_member ):
125
135
email = test_member .email
126
136
id = test_member .id
127
137
test_member .delete ()
128
138
member_ids = [
129
- member .id for member in Member ( client = client ) .get_members (search = email )
139
+ member .id for member in current_member .get_members (search = email )
130
140
]
131
141
assert id not in member_ids
132
142
0 commit comments