mirror of
https://github.com/hanzoai/sentinel.git
synced 2026-09-29 22:34:33 +00:00
28 lines
1001 B
Python
28 lines
1001 B
Python
from django.urls import reverse
|
|
|
|
from sentry.auth.authenticators.totp import TotpInterface
|
|
from sentry.testutils.cases import APITestCase
|
|
from sentry.testutils.silo import control_silo_test
|
|
|
|
|
|
@control_silo_test
|
|
class UserAuthenticatorIndexTest(APITestCase):
|
|
def test_list_all_authenticators(self):
|
|
user = self.create_user(email="[email protected]", is_superuser=True)
|
|
self.login_as(user=user, superuser=True)
|
|
url = reverse("sentry-api-0-user-authenticator-index", kwargs={"user_id": "me"})
|
|
|
|
resp = self.client.get(url, format="json")
|
|
|
|
assert resp.status_code == 200
|
|
interface = [i for i in resp.data if i["id"] == "totp"][0]
|
|
assert not interface["isEnrolled"]
|
|
|
|
# Enroll in Totp - should still be listed
|
|
TotpInterface().enroll(user)
|
|
|
|
resp = self.client.get(url, format="json")
|
|
assert resp.status_code == 200
|
|
interface = [i for i in resp.data if i["id"] == "totp"][0]
|
|
assert interface["isEnrolled"]
|