6
6
)
7
7
from labelbox .schema .user_group import UserGroup , UserGroupColor
8
8
9
+ from libs .labelbox .tests .conftest import AdminClient
10
+ import os
11
+
9
12
data = faker .Faker ()
10
13
11
14
@@ -28,18 +31,18 @@ def user_group(client):
28
31
29
32
30
33
@pytest .fixture (scope = "module" )
31
- def test_member (client , current_member , user_group ):
34
+ def test_member (client , current_member , admin_client : AdminClient ):
35
+ admin_client ._create_user (client .get_organization ().uid )
32
36
members = list (Member (client = client ).get_members ())
33
37
test_member = None
34
38
for member in members :
35
39
if member .id != current_member .id :
36
40
test_member = member
37
- test_member .user_group_ids .add (user_group .id )
38
- updated_member = test_member .update ()
39
- yield updated_member
40
- # remove from any user_groups as clean up
41
- updated_member .user_group_ids = set ()
42
- updated_member .update ()
41
+ if test_member is None :
42
+ raise ValueError ("Valid member was not found" )
43
+ yield test_member
44
+ # delete member for clean up
45
+ test_member .delete ()
43
46
44
47
45
48
def test_get_member (current_member , client ):
@@ -48,11 +51,12 @@ def test_get_member(current_member, client):
48
51
assert current_member_eq .email == current_member .email
49
52
50
53
51
- def test_throw_error_when_deleting_self (current_member , client ):
54
+ def test_throw_error_when_deleting_self (current_member ):
52
55
with pytest .raises (ValueError ):
53
56
current_member .delete ()
54
57
55
58
59
+ @pytest .mark .skipif (condition = os .environ ["LABELBOX_TEST_ENVIRON" ] != "staging" )
56
60
def test_update_member (client , test_member , project_pack , user_group ):
57
61
labeler_role = client .get_roles ()["LABELER" ]
58
62
reviewer_role = client .get_roles ()["REVIEWER" ]
@@ -99,6 +103,7 @@ def test_update_member(client, test_member, project_pack, user_group):
99
103
assert updated_member .can_access_all_projects
100
104
101
105
106
+ @pytest .mark .skipif (condition = os .environ ["LABELBOX_TEST_ENVIRON" ] != "staging" )
102
107
def test_get_members (test_member , current_member , client ):
103
108
member_ids = [
104
109
member .id
@@ -115,6 +120,17 @@ def test_get_members(test_member, current_member, client):
115
120
assert current_member .id in member_ids
116
121
117
122
123
+ @pytest .mark .skipif (condition = os .environ ["LABELBOX_TEST_ENVIRON" ] != "staging" )
124
+ def test_delete_member (test_member , current_member , client ):
125
+ email = test_member .email
126
+ id = test_member .id
127
+ test_member .delete ()
128
+ member_ids = [
129
+ member .id for member in Member (client = client ).get_members (search = email )
130
+ ]
131
+ assert id not in member_ids
132
+
133
+
118
134
if __name__ == "__main__" :
119
135
import subprocess
120
136
0 commit comments