browsertrix/backend/test/test_invites.py
Tessa Walsh 14b349443f
Make pending invites expire via TTL index (#568)
* Make invites expire after configurable window

The value can be set in EXPIRE_AFTER_SECONDS env var and via
helm chart values, and defaults to 7 days.

* Create nightly test CI and add invite expiration test to it

* Update 404 error message for missing or expired invite

---------

Co-authored-by: sua yoo <sua@suayoo.com>
2023-02-14 16:07:14 -05:00

44 lines
1.3 KiB
Python

import requests
import time
import pytest
from .conftest import API_PREFIX
def test_pending_invites(admin_auth_headers, default_org_id):
r = requests.get(f"{API_PREFIX}/users/invites", headers=admin_auth_headers)
assert r.status_code == 200
data = r.json()
assert data["pending_invites"] == []
# Add a pending invite and check it's returned
INVITE_EMAIL = "invite-pending@example.com"
r = requests.post(
f"{API_PREFIX}/orgs/{default_org_id}/invite",
headers=admin_auth_headers,
json={"email": INVITE_EMAIL, "role": 20},
)
assert r.status_code == 200
data = r.json()
assert data["invited"] == "new_user"
r = requests.get(f"{API_PREFIX}/users/invites", headers=admin_auth_headers)
assert r.status_code == 200
data = r.json()
invites = data["pending_invites"]
assert len(invites) == 1
invite = invites[0]
assert invite["id"]
assert invite["email"] == INVITE_EMAIL
assert invite["oid"] == default_org_id
assert invite["created"]
assert invite["role"]
def test_pending_invites_crawler(crawler_auth_headers, default_org_id):
# Verify that only superusers can see pending invites
r = requests.get(f"{API_PREFIX}/users/invites", headers=crawler_auth_headers)
assert r.status_code == 403