|
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