File: test_auth.py

package info (click to toggle)
python-mastodon 2.1.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 22,836 kB
  • sloc: python: 9,438; makefile: 206; sql: 98; sh: 27
file content (102 lines) | stat: -rw-r--r-- 4,313 bytes parent folder | download
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
import pytest
from mastodon.Mastodon import MastodonIllegalArgumentError
from mastodon import Mastodon
try:
    from urllib.parse import urlparse, parse_qs
except ImportError:
    from urlparse import urlparse, parse_qs
import vcr

@pytest.mark.vcr()
def test_auth_request_url(api):
    url = api.auth_request_url(allow_http=True)
    url2 = api.auth_request_url(skip_server_info=True, allow_http=True)
    assert url == url2
    parse = urlparse(url)
    assert parse.path == '/oauth/authorize'
    query = parse_qs(parse.query)
    assert query['client_id'][0] == api.client_id
    assert query['response_type'][0] == 'code'
    assert query['redirect_uri'][0] == 'urn:ietf:wg:oauth:2.0:oob'
    assert set(query['scope'][0].split()) == set(('read', 'write', 'follow', 'push'))

    with pytest.raises(MastodonIllegalArgumentError):
        api.auth_request_url(allow_http=False)

@pytest.mark.vcr()
def test_log_in_none(api_anonymous):
    with pytest.raises(MastodonIllegalArgumentError):
        api_anonymous.log_in()

@pytest.mark.vcr()
def test_log_in_password(api_anonymous):
    # No password login after 4.4.0, so this can't be tested anymore against newer servers
    with vcr.use_cassette('test_log_in_password.yaml', cassette_library_dir='tests/cassettes_pre_4_4_0', record_mode='none'):
        token = api_anonymous.log_in(
            username='mastodonpy_test_2@localhost',
            password='5fc638e0e53eafd9c4145b6bb852667d',
            allow_http=True
        )
        assert token

@pytest.mark.vcr()
def test_log_in_password_incorrect(api_anonymous):
    # No password login after 4.4.0, so this can't be tested anymore against newer servers
    with vcr.use_cassette('test_log_in_password_incorrect.yaml', cassette_library_dir='tests/cassettes_pre_4_4_0', record_mode='none'):
        with pytest.raises(MastodonIllegalArgumentError):
            api_anonymous.log_in(
                username='admin@localhost',
                password='hunter2',
                allow_http=True
            )

@pytest.mark.vcr()
def test_log_in_password_to_file(api_anonymous, tmpdir):
    # No password login after 4.4.0, so this can't be tested anymore against newer servers
    with vcr.use_cassette('test_log_in_password_to_file.yaml', cassette_library_dir='tests/cassettes_pre_4_4_0', record_mode='none'):    
        filepath = tmpdir.join('token')
        api_anonymous.log_in(
            username='mastodonpy_test_2@localhost',
            password='5fc638e0e53eafd9c4145b6bb852667d',
            to_file=str(filepath),
            allow_http=True
        )
        token = filepath.read_text('UTF-8').rstrip().split("\n")[0]
        assert token
        api = api_anonymous
        api.access_token = token
        assert api.account_verify_credentials()

@pytest.mark.vcr()
def test_url_errors(tmpdir):
    clientid_good = tmpdir.join("clientid")
    token_good = tmpdir.join("token")
    clientid_bad = tmpdir.join("clientid_bad")
    token_bad = tmpdir.join("token_bad")
    
    clientid_good.write_text("foo\nbar\nhttps://zombo.com\n", "UTF-8")
    token_good.write_text("foo\nhttps://zombo.com\n", "UTF-8")
    clientid_bad.write_text("foo\nbar\nhttps://evil.org\n", "UTF-8")
    token_bad.write_text("foo\nhttps://evil.org\n", "UTF-8")  
    
    api = Mastodon(client_id = clientid_good, access_token = token_good)
    assert api
    assert api.api_base_url == "https://zombo.com"
    assert Mastodon(client_id = clientid_good, access_token = token_good, api_base_url = "zombo.com")
    
    with pytest.raises(MastodonIllegalArgumentError):
        Mastodon(client_id = clientid_good, access_token = token_bad, api_base_url = "zombo.com")
        
    with pytest.raises(MastodonIllegalArgumentError):
        Mastodon(client_id = clientid_bad, access_token = token_good, api_base_url = "zombo.com")
        
    with pytest.raises(MastodonIllegalArgumentError):
        Mastodon(client_id = clientid_bad, access_token = token_bad, api_base_url = "zombo.com")        

@pytest.mark.skip(reason="Not sure how to test this without setting up selenium or a similar browser automation suite to click on the allow button")
def test_log_in_code(api_anonymous):
    pass

@pytest.mark.skip(reason="Not supported by Mastodon >:@ (yet?)")
def test_log_in_refresh(api_anonymous):
    pass