kopia lustrzana https://github.com/snarfed/bridgy-fed
451 wiersze
17 KiB
Python
451 wiersze
17 KiB
Python
# coding=utf-8
|
|
"""Unit tests for activitypub.py.
|
|
|
|
TODO: test error handling
|
|
"""
|
|
import copy
|
|
from unittest.mock import ANY, call, patch
|
|
|
|
from oauth_dropins.webutil import util
|
|
from oauth_dropins.webutil.testutil import requests_response
|
|
from oauth_dropins.webutil.util import json_dumps, json_loads
|
|
import requests
|
|
from urllib3.exceptions import ReadTimeoutError
|
|
|
|
import activitypub
|
|
import common
|
|
from models import Follower, Domain, Activity
|
|
from . import testutil
|
|
|
|
REPLY_OBJECT = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'type': 'Note',
|
|
'content': 'A ☕ reply',
|
|
'id': 'http://this/reply/id',
|
|
'url': 'http://this/reply',
|
|
'inReplyTo': 'http://orig/post',
|
|
'cc': ['https://www.w3.org/ns/activitystreams#Public'],
|
|
}
|
|
REPLY_OBJECT_WRAPPED = copy.deepcopy(REPLY_OBJECT)
|
|
REPLY_OBJECT_WRAPPED['inReplyTo'] = 'http://localhost/r/orig/post'
|
|
REPLY = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'type': 'Create',
|
|
'id': 'http://this/reply/as2',
|
|
'object': REPLY_OBJECT,
|
|
}
|
|
MENTION_OBJECT = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'type': 'Note',
|
|
'content': '☕ mentions of @other @target@target',
|
|
'id': 'http://this/mention/id',
|
|
'url': 'http://this/mention',
|
|
'to': ['https://www.w3.org/ns/activitystreams#Public'],
|
|
'cc': [
|
|
'https://this/author/followers',
|
|
'https://masto.foo/@other',
|
|
'http://localhost/target', # redirect-wrapped
|
|
],
|
|
'tag': [{
|
|
'type': 'Mention',
|
|
'href': 'https://masto.foo/@other',
|
|
'name': '@other@masto.foo',
|
|
}, {
|
|
'type': 'Mention',
|
|
'href': 'http://localhost/target', # redirect-wrapped
|
|
'name': '@target@target',
|
|
}],
|
|
}
|
|
MENTION = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'type': 'Create',
|
|
'id': 'http://this/mention/as2',
|
|
'object': MENTION_OBJECT,
|
|
}
|
|
# based on example Mastodon like:
|
|
# https://github.com/snarfed/bridgy-fed/issues/4#issuecomment-334212362
|
|
# (reposts are very similar)
|
|
LIKE = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'id': 'http://this/like#ok',
|
|
'type': 'Like',
|
|
'object': 'http://orig/post',
|
|
'actor': 'http://orig/actor',
|
|
}
|
|
LIKE_WRAPPED = copy.deepcopy(LIKE)
|
|
LIKE_WRAPPED['object'] = 'http://localhost/r/http://orig/post'
|
|
LIKE_WITH_ACTOR = copy.deepcopy(LIKE)
|
|
LIKE_WITH_ACTOR['actor'] = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'id': 'http://orig/actor',
|
|
'type': 'Person',
|
|
'name': 'Ms. Actor',
|
|
'preferredUsername': 'msactor',
|
|
'image': {'type': 'Image', 'url': 'http://orig/pic.jpg'},
|
|
}
|
|
|
|
FOLLOW = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'id': 'https://mastodon.social/6d1a',
|
|
'type': 'Follow',
|
|
'actor': 'https://mastodon.social/users/swentel',
|
|
'object': 'https://www.realize.be/',
|
|
}
|
|
FOLLOW_WRAPPED = copy.deepcopy(FOLLOW)
|
|
FOLLOW_WRAPPED['object'] = 'http://localhost/www.realize.be'
|
|
ACTOR = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'id': FOLLOW['actor'],
|
|
'type': 'Person',
|
|
'inbox': 'http://follower/inbox',
|
|
}
|
|
FOLLOW_WITH_ACTOR = copy.deepcopy(FOLLOW)
|
|
FOLLOW_WITH_ACTOR['actor'] = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'id': FOLLOW['actor'],
|
|
'type': 'Person',
|
|
'inbox': 'http://follower/inbox',
|
|
}
|
|
FOLLOW_WRAPPED_WITH_ACTOR = copy.deepcopy(FOLLOW_WRAPPED)
|
|
FOLLOW_WRAPPED_WITH_ACTOR['actor'] = FOLLOW_WITH_ACTOR['actor']
|
|
|
|
ACCEPT = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'type': 'Accept',
|
|
'id': 'tag:localhost:accept/www.realize.be/https://mastodon.social/6d1a',
|
|
'actor': 'http://localhost/www.realize.be',
|
|
'object': {
|
|
'type': 'Follow',
|
|
'actor': 'https://mastodon.social/users/swentel',
|
|
'object': 'http://localhost/www.realize.be',
|
|
}
|
|
}
|
|
|
|
UNDO_FOLLOW_WRAPPED = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'id': 'https://mastodon.social/6d1b',
|
|
'type': 'Undo',
|
|
'actor': 'https://mastodon.social/users/swentel',
|
|
'object': FOLLOW_WRAPPED,
|
|
}
|
|
|
|
DELETE = {
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'id': 'https://mastodon.social/users/swentel#delete',
|
|
'type': 'Delete',
|
|
'actor': 'https://mastodon.social/users/swentel',
|
|
'object': 'https://mastodon.social/users/swentel',
|
|
}
|
|
|
|
|
|
@patch('requests.post')
|
|
@patch('requests.get')
|
|
@patch('requests.head')
|
|
class ActivityPubTest(testutil.TestCase):
|
|
|
|
def test_actor(self, _, mock_get, __):
|
|
mock_get.return_value = requests_response("""
|
|
<body>
|
|
<a class="h-card u-url" rel="me" href="/about-me">Mrs. ☕ Foo</a>
|
|
</body>
|
|
""", url='https://foo.com/', content_type=common.CONTENT_TYPE_HTML)
|
|
|
|
got = self.client.get('/foo.com')
|
|
self.assert_req(mock_get, 'http://foo.com/')
|
|
self.assertEqual(200, got.status_code)
|
|
type = got.headers['Content-Type']
|
|
self.assertTrue(type.startswith(common.CONTENT_TYPE_AS2), type)
|
|
self.assertEqual({
|
|
'@context': [
|
|
'https://www.w3.org/ns/activitystreams',
|
|
'https://w3id.org/security/v1',
|
|
],
|
|
'type' : 'Person',
|
|
'name': 'Mrs. ☕ Foo',
|
|
'summary': '',
|
|
'preferredUsername': 'foo.com',
|
|
'id': 'http://localhost/foo.com',
|
|
'url': 'http://localhost/r/https://foo.com/about-me',
|
|
'inbox': 'http://localhost/foo.com/inbox',
|
|
'outbox': 'http://localhost/foo.com/outbox',
|
|
'following': 'http://localhost/foo.com/following',
|
|
'followers': 'http://localhost/foo.com/followers',
|
|
'publicKey': {
|
|
'id': 'http://localhost/foo.com',
|
|
'owner': 'http://localhost/foo.com',
|
|
'publicKeyPem': Domain.get_by_id('foo.com').public_pem().decode(),
|
|
},
|
|
}, got.json)
|
|
|
|
def test_actor_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>
|
|
""")
|
|
|
|
got = self.client.get('/foo.com')
|
|
self.assert_req(mock_get, 'http://foo.com/')
|
|
self.assertEqual(400, got.status_code)
|
|
self.assertIn('representative h-card', got.get_data(as_text=True))
|
|
|
|
def test_actor_override_preferredUsername(self, _, mock_get, __):
|
|
mock_get.return_value = requests_response("""
|
|
<body>
|
|
<a class="h-card u-url" rel="me" href="/about-me">
|
|
<span class="p-nickname">Nick</span>
|
|
</a>
|
|
</body>
|
|
""", url='https://foo.com/', content_type=common.CONTENT_TYPE_HTML)
|
|
|
|
got = self.client.get('/foo.com')
|
|
self.assertEqual(200, got.status_code)
|
|
self.assertEqual('foo.com', got.json['preferredUsername'])
|
|
|
|
def test_actor_blocked_tld(self, _, __, ___):
|
|
got = self.client.get('/foo.json')
|
|
self.assertEqual(404, got.status_code)
|
|
|
|
def test_inbox_reply_object(self, *mocks):
|
|
self._test_inbox_reply(REPLY_OBJECT, REPLY_OBJECT, *mocks)
|
|
|
|
def test_inbox_reply_object_wrapped(self, *mocks):
|
|
self._test_inbox_reply(REPLY_OBJECT_WRAPPED, REPLY_OBJECT, *mocks)
|
|
|
|
def test_inbox_reply_create_activity(self, *mocks):
|
|
self._test_inbox_reply(REPLY, REPLY, *mocks)
|
|
|
|
def _test_inbox_reply(self, as2, expected_as2, mock_head, mock_get, mock_post):
|
|
mock_head.return_value = requests_response(url='http://orig/post')
|
|
mock_get.return_value = requests_response(
|
|
'<html><head><link rel="webmention" href="/webmention"></html>')
|
|
mock_post.return_value = requests_response()
|
|
|
|
got = self.client.post('/foo.com/inbox', json=as2)
|
|
self.assertEqual(200, got.status_code, got.get_data(as_text=True))
|
|
self.assert_req(mock_get, 'http://orig/post')
|
|
self.assert_req(
|
|
mock_post,
|
|
'http://orig/webmention',
|
|
headers={'Accept': '*/*'},
|
|
allow_redirects=False,
|
|
data={
|
|
'source': 'http://localhost/render?source=http%3A%2F%2Fthis%2Freply&target=http%3A%2F%2Forig%2Fpost',
|
|
'target': 'http://orig/post',
|
|
},
|
|
)
|
|
|
|
resp = Activity.get_by_id('http://this/reply http://orig/post')
|
|
self.assertEqual('orig', resp.domain)
|
|
self.assertEqual('in', resp.direction)
|
|
self.assertEqual('activitypub', resp.protocol)
|
|
self.assertEqual('complete', resp.status)
|
|
self.assertEqual(expected_as2, json_loads(resp.source_as2))
|
|
|
|
def test_inbox_reply_drop_self_domain_target(self, mock_head, mock_get, mock_post):
|
|
reply = copy.deepcopy(REPLY_OBJECT)
|
|
# same domain as source; should drop
|
|
reply['inReplyTo'] = 'http://localhost/this',
|
|
|
|
mock_head.return_value = requests_response(url='http://this/')
|
|
|
|
got = self.client.post('/foo.com/inbox', json=reply)
|
|
self.assertEqual(200, got.status_code, got.get_data(as_text=True))
|
|
|
|
self.assert_req(mock_head, 'http://this', allow_redirects=True)
|
|
mock_get.assert_not_called()
|
|
mock_post.assert_not_called()
|
|
self.assertEqual(0, Activity.query().count())
|
|
|
|
def test_inbox_mention_object(self, *mocks):
|
|
self._test_inbox_mention(MENTION_OBJECT, *mocks)
|
|
|
|
def test_inbox_mention_create_activity(self, *mocks):
|
|
self._test_inbox_mention(MENTION, *mocks)
|
|
|
|
def _test_inbox_mention(self, as2, mock_head, mock_get, mock_post):
|
|
mock_head.return_value = requests_response(url='http://target')
|
|
mock_get.return_value = requests_response(
|
|
'<html><head><link rel="webmention" href="/webmention"></html>')
|
|
mock_post.return_value = requests_response()
|
|
|
|
with self.client:
|
|
got = self.client.post('/foo.com/inbox', json=as2)
|
|
self.assertEqual(200, got.status_code, got.get_data(as_text=True))
|
|
self.assert_req(mock_get, 'http://target/')
|
|
self.assert_req(
|
|
mock_post,
|
|
'http://target/webmention',
|
|
headers={'Accept': '*/*'},
|
|
allow_redirects=False,
|
|
data={
|
|
'source': 'http://localhost/render?source=http%3A%2F%2Fthis%2Fmention&target=http%3A%2F%2Ftarget%2F',
|
|
'target': 'http://target/',
|
|
},
|
|
)
|
|
|
|
resp = Activity.get_by_id('http://this/mention http://target/')
|
|
self.assertEqual('target', resp.domain)
|
|
self.assertEqual('in', resp.direction)
|
|
self.assertEqual('activitypub', resp.protocol)
|
|
self.assertEqual('complete', resp.status)
|
|
self.assertEqual(common.redirect_unwrap(as2), json_loads(resp.source_as2))
|
|
|
|
def test_inbox_like(self, mock_head, mock_get, mock_post):
|
|
mock_head.return_value = requests_response(url='http://orig/post')
|
|
mock_get.side_effect = [
|
|
# source actor
|
|
requests_response(LIKE_WITH_ACTOR['actor'], headers={'Content-Type': common.CONTENT_TYPE_AS2}),
|
|
# target post webmention discovery
|
|
requests_response(
|
|
'<html><head><link rel="webmention" href="/webmention"></html>'),
|
|
]
|
|
mock_post.return_value = requests_response()
|
|
|
|
got = self.client.post('/foo.com/inbox', json=LIKE)
|
|
self.assertEqual(200, got.status_code)
|
|
|
|
self.assert_req(mock_get, 'http://orig/actor',
|
|
headers=common.CONNEG_HEADERS_AS2_HTML)
|
|
self.assert_req(mock_get, 'http://orig/post')
|
|
|
|
args, kwargs = mock_post.call_args
|
|
self.assertEqual(('http://orig/webmention',), args)
|
|
self.assertEqual({
|
|
# TODO
|
|
'source': 'http://localhost/render?source=http%3A%2F%2Fthis%2Flike__ok&target=http%3A%2F%2Forig%2Fpost',
|
|
'target': 'http://orig/post',
|
|
}, kwargs['data'])
|
|
|
|
resp = Activity.get_by_id('http://this/like__ok http://orig/post')
|
|
self.assertEqual('orig', resp.domain)
|
|
self.assertEqual('in', resp.direction)
|
|
self.assertEqual('activitypub', resp.protocol)
|
|
self.assertEqual('complete', resp.status)
|
|
self.assertEqual(LIKE_WITH_ACTOR, json_loads(resp.source_as2))
|
|
|
|
def test_inbox_follow_accept(self, mock_head, mock_get, mock_post):
|
|
mock_head.return_value = requests_response(url='https://www.realize.be/')
|
|
mock_get.side_effect = [
|
|
# source actor
|
|
requests_response(FOLLOW_WITH_ACTOR['actor'],
|
|
content_type=common.CONTENT_TYPE_AS2),
|
|
# target post webmention discovery
|
|
requests_response(
|
|
'<html><head><link rel="webmention" href="/webmention"></html>'),
|
|
]
|
|
mock_post.return_value = requests_response()
|
|
|
|
got = self.client.post('/foo.com/inbox', json=FOLLOW_WRAPPED)
|
|
self.assertEqual(200, got.status_code)
|
|
|
|
self.assert_req(mock_get, FOLLOW['actor'],
|
|
headers=common.CONNEG_HEADERS_AS2_HTML)
|
|
|
|
# check AP Accept
|
|
self.assertEqual(2, len(mock_post.call_args_list))
|
|
args, kwargs = mock_post.call_args_list[0]
|
|
self.assertEqual(('http://follower/inbox',), args)
|
|
self.assertEqual(ACCEPT, json_loads(kwargs['data']))
|
|
|
|
# check webmention
|
|
args, kwargs = mock_post.call_args_list[1]
|
|
self.assertEqual(('https://www.realize.be/webmention',), args)
|
|
self.assertEqual({
|
|
'source': 'http://localhost/render?source=https%3A%2F%2Fmastodon.social%2F6d1a&target=https%3A%2F%2Fwww.realize.be%2F',
|
|
'target': 'https://www.realize.be/',
|
|
}, kwargs['data'])
|
|
|
|
resp = Activity.get_by_id('https://mastodon.social/6d1a https://www.realize.be/')
|
|
self.assertEqual('www.realize.be', resp.domain)
|
|
self.assertEqual('in', resp.direction)
|
|
self.assertEqual('activitypub', resp.protocol)
|
|
self.assertEqual('complete', resp.status)
|
|
self.assertEqual(FOLLOW_WITH_ACTOR, json_loads(resp.source_as2))
|
|
|
|
# check that we stored a Follower object
|
|
follower = Follower.get_by_id(f'www.realize.be {FOLLOW["actor"]}')
|
|
self.assertEqual('active', follower.status)
|
|
self.assertEqual(FOLLOW_WRAPPED_WITH_ACTOR, json_loads(follower.last_follow))
|
|
|
|
def test_inbox_undo_follow(self, mock_head, mock_get, mock_post):
|
|
mock_head.return_value = requests_response(url='https://www.realize.be/')
|
|
|
|
Follower(id=Follower._id('www.realize.be', FOLLOW['actor'])).put()
|
|
|
|
got = self.client.post('/foo.com/inbox', json=UNDO_FOLLOW_WRAPPED)
|
|
self.assertEqual(200, got.status_code)
|
|
|
|
follower = Follower.get_by_id(f'www.realize.be {FOLLOW["actor"]}')
|
|
self.assertEqual('inactive', follower.status)
|
|
|
|
def test_inbox_undo_follow_doesnt_exist(self, mock_head, mock_get, mock_post):
|
|
mock_head.return_value = requests_response(url='https://realize.be/')
|
|
|
|
got = self.client.post('/foo.com/inbox', json=UNDO_FOLLOW_WRAPPED)
|
|
self.assertEqual(200, got.status_code)
|
|
|
|
def test_inbox_undo_follow_inactive(self, mock_head, mock_get, mock_post):
|
|
mock_head.return_value = requests_response(url='https://realize.be/')
|
|
Follower(id=Follower._id('realize.be', 'https://mastodon.social/users/swentel'),
|
|
status='inactive').put()
|
|
|
|
got = self.client.post('/foo.com/inbox', json=UNDO_FOLLOW_WRAPPED)
|
|
self.assertEqual(200, got.status_code)
|
|
|
|
def test_inbox_unsupported_type(self, *_):
|
|
got = self.client.post('/foo.com/inbox', json={
|
|
'@context': ['https://www.w3.org/ns/activitystreams'],
|
|
'id': 'https://xoxo.zone/users/aaronpk#follows/40',
|
|
'type': 'Block',
|
|
'actor': 'https://xoxo.zone/users/aaronpk',
|
|
'object': 'http://snarfed.org/',
|
|
})
|
|
self.assertEqual(501, got.status_code)
|
|
|
|
def test_inbox_delete_actor(self, mock_head, mock_get, mock_post):
|
|
follower = Follower.get_or_create('realize.be', DELETE['actor'])
|
|
Follower.get_or_create('snarfed.org', DELETE['actor'])
|
|
# other unrelated follower
|
|
other = Follower.get_or_create('realize.be', 'https://mas.to/users/other')
|
|
self.assertEqual(3, Follower.query().count())
|
|
|
|
got = self.client.post('/realize.be/inbox', json=DELETE)
|
|
self.assertEqual(200, got.status_code)
|
|
|
|
# TODO: bring back once we actually delete followers
|
|
# self.assertEqual([other], Follower.query().fetch())
|
|
|
|
def test_inbox_webmention_discovery_connection_fails(self, mock_head,
|
|
mock_get, mock_post):
|
|
mock_get.side_effect = [
|
|
# source actor
|
|
requests_response(LIKE_WITH_ACTOR['actor'],
|
|
headers={'Content-Type': common.CONTENT_TYPE_AS2}),
|
|
# target post webmention discovery
|
|
ReadTimeoutError(None, None, None),
|
|
]
|
|
|
|
got = self.client.post('/foo.com/inbox', json=LIKE)
|
|
self.assertEqual(504, got.status_code)
|
|
|
|
def test_inbox_no_webmention_endpoint(self, mock_head, mock_get, mock_post):
|
|
mock_get.side_effect = [
|
|
# source actor
|
|
requests_response(LIKE_WITH_ACTOR['actor'],
|
|
headers={'Content-Type': common.CONTENT_TYPE_AS2}),
|
|
# target post webmention discovery
|
|
requests_response('<html><body>foo</body></html>'),
|
|
]
|
|
|
|
got = self.client.post('/foo.com/inbox', json=LIKE)
|
|
self.assertEqual(200, got.status_code)
|
|
|
|
resp = Activity.get_by_id('http://this/like__ok http://orig/post')
|
|
self.assertEqual('orig', resp.domain)
|
|
self.assertEqual('in', resp.direction)
|
|
self.assertEqual('activitypub', resp.protocol)
|
|
self.assertEqual('ignored', resp.status)
|