2FA Log In about tavern HOT 2 CLOSED

import http.cookies
from django.test import Client


def django_cookies(client: Client) -> http.cookies.SimpleCookie:
    client.login("username", "password")

    return client.cookies

from unittest import mock

class MockOAuth():
    class MockOktaUserInfoResponse():
        def __init__(self, email):
            self.email = email


        def raise_for_status(*args, **kwargs):
            return None


        def json(self):
            if self.email:
                return { "email": self.email }
            else:
                return {}


    class MockOkta():
        def authorize_access_token(self, request):
            return request.POST.get("email")


        def get(self, userinfo_uri, token):
            return MockOAuth.MockOktaUserInfoResponse(email=token)


    def __init__(self):
        self.okta = MockOAuth.MockOkta()


    def register(*args, **kwargs):
        return True


@pytest.fixture(autouse=True)
def mock_okta():
    with mock.patch("authlib.integrations.django_client.OAuth", new=MockOAuth):
        yield