|
1 | 1 | import pytest |
| 2 | +from faker import Faker |
2 | 3 |
|
3 | 4 | from labelbox import ProjectRole |
4 | | -from faker import Faker |
5 | 5 |
|
6 | 6 | faker = Faker() |
7 | 7 |
|
@@ -67,25 +67,34 @@ def test_org_invite(client, organization, environ, queries, org_invite): |
67 | 67 | invite, invite_limit = org_invite |
68 | 68 | role = client.get_roles()["LABELER"] |
69 | 69 |
|
| 70 | + assert ( |
| 71 | + invite.uid == "invited" |
| 72 | + ) # for security reasons we don't return the invite uid |
| 73 | + |
70 | 74 | if environ.value == "prod": |
71 | 75 | invite_limit_after = organization.invite_limit() |
72 | 76 | # One user added |
73 | 77 | assert invite_limit.remaining - invite_limit_after.remaining == 1 |
74 | 78 | # An invite shouldn't effect the user count until after it is accepted |
75 | 79 |
|
76 | 80 | outstanding_invites = queries.get_invites(client) |
77 | | - in_list = False |
78 | 81 |
|
| 82 | + found_invite = None |
79 | 83 | for outstanding_invite in outstanding_invites: |
80 | | - if outstanding_invite.uid == invite.uid: |
81 | | - in_list = True |
82 | | - org_role = outstanding_invite.organization_role_name.lower() |
83 | | - assert ( |
84 | | - org_role == role.name.lower() |
85 | | - ), "Role should be labeler. Found {org_role} " |
86 | | - assert in_list, "Invite not found" |
| 84 | + if outstanding_invite.email == invite.email: |
| 85 | + found_invite = outstanding_invite |
| 86 | + break |
| 87 | + |
| 88 | + assert found_invite is not None, "Invite not found" |
| 89 | + org_role = found_invite.organization_role_name.lower() |
| 90 | + assert ( |
| 91 | + org_role == role.name.lower() |
| 92 | + ), "Role should be labeler. Found {org_role} " |
87 | 93 |
|
88 | 94 |
|
| 95 | +@pytest.mark.skip( |
| 96 | + "Test disabled - due to security reasons, invite uid is not returned" |
| 97 | +) |
89 | 98 | def test_cancel_invite( |
90 | 99 | client, |
91 | 100 | organization, |
|
0 commit comments