1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
from django.test.client import Client
from django.urls import reverse
import pytest
@pytest.fixture(params=["app", "browser"])
def headless_client(request):
return request.param
@pytest.fixture
def headless_reverse(headless_client):
def rev(viewname, **kwargs):
viewname = viewname.replace("headless:", f"headless:{headless_client}:")
return reverse(viewname, **kwargs)
return rev
class AppClient(Client):
session_token = None
def generic(self, *args, **kwargs):
if self.session_token:
kwargs["HTTP_X_SESSION_TOKEN"] = self.session_token
resp = super().generic(*args, **kwargs)
if resp["content-type"] == "application/json":
data = resp.json()
session_token = data.get("meta", {}).get("session_token")
if session_token:
self.session_token = session_token
return resp
def force_login(self, user):
ret = super().force_login(user)
self.session_token = self.session.session_key
return ret
def headless_session(self):
from allauth.headless.internal import sessionkit
return sessionkit.session_store(self.session_token)
@pytest.fixture
def app_client():
return AppClient()
@pytest.fixture
def client(headless_client):
if headless_client == "browser":
client = Client()
client.headless_session = lambda: client.session
return client
return AppClient()
@pytest.fixture
def auth_client(client, user):
client.force_login(user)
return client
|
