bridgy-fed/tests/test_webfinger.py

239 wiersze
8.9 KiB
Python
Czysty Zwykły widok Historia

2017-08-19 15:21:05 +00:00
# coding=utf-8
"""Unit tests for webfinger.py.
to test:
* user URL that redirects
* error handling
2017-08-19 15:21:05 +00:00
"""
import html
from unittest import mock
import urllib.parse
2017-08-19 15:21:05 +00:00
from oauth_dropins.webutil import util
from oauth_dropins.webutil.testutil import requests_response
from oauth_dropins.webutil.util import json_loads
2017-08-19 15:21:05 +00:00
import requests
2017-08-19 20:10:18 +00:00
import common
import models
from . import testutil
2017-08-19 15:21:05 +00:00
class WebfingerTest(testutil.TestCase):
def setUp(self):
2021-08-18 14:59:52 +00:00
super().setUp()
self.html = """
<body class="h-card">
<a class="u-url" rel="me" href="/about-me">
<img class="u-photo" src="/me.jpg" />
Mrs. Foo
</a>
</body>
"""
2022-11-16 06:00:28 +00:00
self.key = models.User.get_or_create('foo.com')
self.expected_webfinger = {
'subject': 'acct:foo.com@foo.com',
'aliases': [
2017-08-19 20:10:18 +00:00
'https://foo.com/about-me',
'https://foo.com/',
],
'magic_keys': [{'value': self.key.href()}],
'links': [{
'rel': 'http://webfinger.net/rel/profile-page',
'type': 'text/html',
'href': 'https://foo.com/about-me'
}, {
2017-08-19 20:10:18 +00:00
'rel': 'http://webfinger.net/rel/profile-page',
'type': 'text/html',
'href': 'https://foo.com/'
2017-08-20 02:46:53 +00:00
}, {
'rel': 'http://webfinger.net/rel/avatar',
'href': 'https://foo.com/me.jpg'
}, {
'rel': 'canonical_uri',
'type': 'text/html',
'href': 'https://foo.com/about-me'
}, {
'rel': 'self',
'type': 'application/activity+json',
'href': 'http://localhost/foo.com'
}, {
'rel': 'inbox',
'type': 'application/activity+json',
'href': 'http://localhost/foo.com/inbox'
}, {
'rel': 'sharedInbox',
'type': 'application/activity+json',
'href': 'http://localhost/inbox'
}, {
'rel': 'http://schemas.google.com/g/2010#updates-from',
'type': 'application/atom+xml',
'href': 'https://granary.io/url?input=html&output=atom&url=https%3A%2F%2Ffoo.com%2F&hub=https%3A%2F%2Ffoo.com%2F',
}, {
'rel': 'hub',
'href': 'https://bridgy-fed.superfeedr.com/'
}, {
'rel': 'magic-public-key',
'href': self.key.href(),
2017-08-20 02:46:53 +00:00
}, {
'rel': 'salmon',
'href': 'http://localhost/foo.com/salmon'
}]
}
def test_host_meta_handler_xrd(self):
2021-08-18 14:59:52 +00:00
got = self.client.get('/.well-known/host-meta')
self.assertEqual(200, got.status_code)
self.assertEqual('application/xrd+xml; charset=utf-8',
got.headers['Content-Type'])
body = got.get_data(as_text=True)
self.assertTrue(body.startswith('<?xml'), body)
def test_host_meta_handler_xrds(self):
2021-08-18 14:59:52 +00:00
got = self.client.get('/.well-known/host-meta.xrds')
self.assertEqual(200, got.status_code)
self.assertEqual('application/xrds+xml', got.headers['Content-Type'])
body = got.get_data(as_text=True)
self.assertTrue(body.startswith('<XRDS'), body)
def test_host_meta_handler_jrd(self):
2021-08-18 14:59:52 +00:00
got = self.client.get('/.well-known/host-meta.json')
self.assertEqual(200, got.status_code)
self.assertEqual('application/jrd+json', got.headers['Content-Type'])
body = got.get_data(as_text=True)
self.assertTrue(body.startswith('{'), body)
@mock.patch('requests.get')
def test_user_handler(self, mock_get):
mock_get.return_value = requests_response(self.html, url='https://foo.com/')
2021-08-18 14:59:52 +00:00
got = self.client.get('/acct:foo.com', headers={'Accept': 'application/json'})
self.assertEqual(200, got.status_code)
self.assertEqual('application/jrd+json', got.headers['Content-Type'])
self.assert_req(mock_get, 'https://foo.com/')
self.assertEqual(self.expected_webfinger, got.json)
# check that magic key is persistent
2021-08-18 14:59:52 +00:00
again = self.client.get('/acct:foo.com',
headers={'Accept': 'application/json'}).json
self.assertEqual(self.key.href(), again['magic_keys'][0]['value'])
2017-08-20 02:46:53 +00:00
links = {l['rel']: l['href'] for l in again['links']}
self.assertEqual(self.key.href(), links['magic-public-key'])
@mock.patch('requests.get')
def test_user_handler_with_atom_feed(self, mock_get):
html = """\
<html>
<head>
<link rel="feed" href="/dont-use">
<link rel="alternate" type="application/rss+xml" href="/dont-use-either">
<link rel="alternate" type="application/atom+xml" href="/use-this">
</head>
""" + self.html
mock_get.return_value = requests_response(html, url = 'https://foo.com/')
2021-08-18 14:59:52 +00:00
got = self.client.get('/acct:foo.com', headers={'Accept': 'application/json'})
self.assertEqual(200, got.status_code)
self.assertIn({
'rel': 'http://schemas.google.com/g/2010#updates-from',
'type': 'application/atom+xml',
'href': 'https://foo.com/use-this',
}, got.json['links'])
@mock.patch('requests.get')
def test_user_handler_with_push_header(self, mock_get):
mock_get.return_value = requests_response(
self.html, url = 'https://foo.com/', headers={
'Link': 'badly formatted, '
"<xyz>; rel='foo',"
'<http://a.custom.hub/>; rel="hub"',
})
2021-08-18 14:59:52 +00:00
got = self.client.get('/acct:foo.com', headers={'Accept': 'application/json'})
self.assertEqual(200, got.status_code)
self.assertIn({
'rel': 'hub',
'href': 'http://a.custom.hub/',
}, got.json['links'])
@mock.patch('requests.get')
def test_user_handler_no_hcard(self, mock_get):
mock_get.return_value = requests_response("""
<body>
<div class="h-entry">
<p class="e-content">foo bar</p>
</div>
</body>
""")
2021-08-18 14:59:52 +00:00
got = self.client.get('/acct:foo.com')
self.assert_req(mock_get, 'https://foo.com/')
self.assertEqual(400, got.status_code)
self.assertIn('representative h-card', got.get_data(as_text=True))
def test_user_handler_bad_tld(self):
2021-08-18 14:59:52 +00:00
got = self.client.get('/acct:foo.json')
self.assertEqual(404, got.status_code)
self.assertIn("doesn't look like a domain",
html.unescape(got.get_data(as_text=True)))
@mock.patch('requests.get')
def test_webfinger_handler(self, mock_get):
mock_get.return_value = requests_response(self.html, url='https://foo.com/')
for resource in ('foo.com@foo.com', 'acct:foo.com@foo.com', 'xyz@foo.com',
'foo.com', 'http://foo.com/', 'https://foo.com/',
'http://localhost/foo.com'):
url = '/.well-known/webfinger?%s' % urllib.parse.urlencode(
{'resource': resource})
2021-08-18 14:59:52 +00:00
got = self.client.get(url, headers={'Accept': 'application/json'})
self.assertEqual(200, got.status_code, got.get_data(as_text=True))
self.assertEqual('application/jrd+json', got.headers['Content-Type'])
self.assertEqual(self.expected_webfinger, got.json)
@mock.patch('requests.get')
def test_webfinger_handler_custom_username(self, mock_get):
self.html = """
<body class="h-card">
<a class="u-url" rel="me" href="/about-me">
<img class="u-photo" src="/me.jpg" />
Mrs. Foo
</a>
<a class="u-url" href="acct:notthisuser@boop.org"></a>
<a class="u-url" href="acct:customuser@foo.com"></a>
</body>
"""
self.expected_webfinger.update({
'subject': 'acct:customuser@foo.com',
'aliases': [
'https://foo.com/about-me',
'acct:notthisuser@boop.org',
'acct:customuser@foo.com',
'https://foo.com/',
],
})
mock_get.return_value = requests_response(self.html, url='https://foo.com/')
for resource in (
'customuser@foo.com',
'acct:customuser@foo.com',
'foo.com',
'http://foo.com/',
'https://foo.com/',
# Mastodon requires this as of 3.3.0
# https://github.com/snarfed/bridgy-fed/issues/73
# 'acct:foo.com@fed.brid.gy',
'acct:foo.com@fed.brid.gy',
'acct:foo.com@bridgy-federated.appspot.com',
'acct:foo.com@localhost',
):
url = '/.well-known/webfinger?%s' % urllib.parse.urlencode(
{'resource': resource})
2021-08-18 14:59:52 +00:00
got = self.client.get(url, headers={'Accept': 'application/json'})
self.assertEqual(200, got.status_code, got.get_data(as_text=True))
self.assertEqual('application/jrd+json', got.headers['Content-Type'])
self.assertEqual(self.expected_webfinger, got.json)