blurb_it/tests/test_util.py at a346755955cbc1f6a2f73ada5fe8707e744afdd1 · python/blurb_it · GitHub

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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
import string

import pytest

from blurb_it import util, error
from aiohttp.test_utils import make_mocked_request
from aiohttp_session import Session, SESSION_KEY


class FakeGH:
    def __init__(self, *, getiter=None, getitem=None, post=None):
        self._getitem_return = getitem
        self._getiter_return = getiter
        self._post_return = post
        self.getitem_url = None
        self.getiter_url = None
        self.post_url = []
        self.post_data = []

    async def getiter(self, url, jwt=None, accept=None):
        self.getiter_url = url
        for item in self._getiter_return:
            yield item


async def test_nonceify():
    body = (
        "Lorem ipsum dolor amet flannel squid normcore tbh raclette enim"
        "pabst tumblr wolf farm-to-table bitters. Bitters keffiyeh next"
        "level proident normcore, et all of +1 90's in blue bottle"
        "chillwave lorem. Id keffiyeh microdosing cupidatat pour-over"
        "paleo farm-to-table tumeric sriracha +1. Raclette in poutine,"
        "bushwick kitsch id pariatur hexagon. Thundercats shaman beard,"
        "nulla swag echo park organic microdosing. Hot chicken tbh pop-up"
        "tacos, asymmetrical tilde veniam bespoke reprehenderit ut do."
    )

    nonce = await util.nonceify(body)
    assert nonce == "Ps4kgC"


async def test_get_misc_news_filename():
    path = await util.get_misc_news_filename(
        issue_number=123,
        section="Library",
        body="Lorem ipsum dolor amet flannel squid normcore tbh raclette enim"
        "pabst tumblr wolf farm-to-table bitters. Bitters keffiyeh next"
        "level proident normcore, et all of +1 90's in blue bottle"
        "chillwave lorem. Id keffiyeh microdosing cupidatat pour-over"
        "paleo farm-to-table tumeric sriracha +1. Raclette in poutine,"
        "bushwick kitsch id pariatur hexagon. Thundercats shaman beard,"
        "nulla swag echo park organic microdosing. Hot chicken tbh pop-up"
        "tacos, asymmetrical tilde veniam bespoke reprehenderit ut do.",
    )

    assert path.startswith("Misc/NEWS.d/next/Library/")
    assert path.endswith(".gh-issue-123.Ps4kgC.rst")


async def test_has_session():
    request = mock_request_session()

    has_session = await util.has_session(request)
    assert has_session


async def test_session_context():
    request = mock_request_session()

    session_context = await util.get_session_context(request)
    assert session_context == {"username": "blurb", "token": "124"}


async def test_no_session_context():
    request = mock_request_no_session()

    session_context = await util.get_session_context(request)
    assert session_context == {}


def mock_request_session():
    request = make_mocked_request("GET", "/")
    session = Session("identity", data=None, new=False)
    session["token"] = "124"
    session["username"] = "blurb"
    request[SESSION_KEY] = session

    return request


def mock_request_no_session():
    request = make_mocked_request("GET", "/")
    session = Session("identity", data=None, new=False)
    request[SESSION_KEY] = session
    return request


def test_get_csrf_token__not_existing(mocker):
    mocker.patch("blurb_it.util.create_csrf_token", return_value="foobar")

    assert util.get_csrf_token({}) == "foobar"


def test_get_csrf_token__existing():

    assert util.get_csrf_token({"csrf": "foobar"}) == "foobar"


def test_create_csrf_token():
    token = util.create_csrf_token()
    assert len(token) == 43
    assert set(token) <= set(string.ascii_letters + string.digits + "-_=")


@pytest.mark.parametrize("token, match", [("a", True), ("b", False)])
def test_compare_csrf_tokens__match(token, match):
    assert util.compare_csrf_tokens("a", token) is match


async def test_get_installation():
    app_installations = [
        {
            "id": 1,
            "account": {
                "login": "octocat",
                "id": 1,
            },
        }
    ]
    gh = FakeGH(getiter=app_installations)
    result = await util.get_installation(gh, "fake_jwt", "octocat")
    assert result == app_installations[0]


async def test_get_installation_not_found():
    app_installations = [
        {
            "id": 1,
            "account": {
                "login": "octocat",
                "id": 1,
            },
        },
        {
            "id": 1,
            "account": {
                "login": "octosaurus",
                "id": 1,
            },
        },
    ]
    gh = FakeGH(getiter=app_installations)
    with pytest.raises(error.InstallationNotFound) as exc:
        await util.get_installation(gh, "fake_jwt", "octonauts")
    assert exc.value.args[0] == "Can't find installation by that user: octonauts"