summaryrefslogtreecommitdiffstats
path: root/users/tests/views/test_auth.py
blob: 22e1fb6e46eff87299a45f6fb8f6d6eaafbc16c8 (plain)
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
import mock
import pytest

from core.models import Config
from users.models import User


@pytest.fixture
def config_system():
    # TODO: Good enough for now, but a better Config mocking system is needed
    result = Config.load_system()
    with mock.patch("core.models.Config.load_system", return_value=result):
        yield result


@pytest.mark.django_db
def test_signup_disabled(client, config_system):
    # Signup disabled and no signup text
    config_system.signup_allowed = False
    resp = client.get("/auth/signup/")
    assert resp.status_code == 200
    content = str(resp.content)
    assert "Not accepting new users at this time" in content
    assert "<button>Create</button>" not in content

    # Signup disabled with signup text configured
    config_system.signup_text = "Go away!!!!!!"
    resp = client.get("/auth/signup/")
    assert resp.status_code == 200
    content = str(resp.content)
    assert "Go away!!!!!!" in content

    # Ensure direct POST doesn't side step guard
    resp = client.post(
        "/auth/signup/", data={"email": "test_signup_disabled@example.org"}
    )
    assert resp.status_code == 200
    assert not User.objects.filter(email="test_signup_disabled@example.org").exists()

    # Signup enabled
    config_system.signup_allowed = True
    resp = client.get("/auth/signup/")
    assert resp.status_code == 200
    content = str(resp.content)
    assert "Not accepting new users at this time" not in content
    assert "<button>Create</button>" in content


@pytest.mark.django_db
def test_signup_invite_only(client, config_system):
    config_system.signup_allowed = True
    config_system.signup_invite_only = True

    resp = client.get("/auth/signup/")
    assert resp.status_code == 200
    content = str(resp.content)
    assert 'name="invite_code"' in content

    # TODO: Actually test this