kopia lustrzana https://github.com/snarfed/bridgy-fed
212 wiersze
7.1 KiB
Python
212 wiersze
7.1 KiB
Python
# coding=utf-8
|
|
"""Unit tests for webmention.py.
|
|
|
|
TODO: test error handling
|
|
"""
|
|
from __future__ import unicode_literals
|
|
import copy
|
|
import json
|
|
import logging
|
|
import urllib
|
|
import urllib2
|
|
|
|
from django_salmon import magicsigs, utils
|
|
import feedparser
|
|
from granary import atom, microformats2
|
|
import mf2py
|
|
import mock
|
|
from mock import call
|
|
from oauth_dropins.webutil import util
|
|
from oauth_dropins.webutil.testutil import requests_response
|
|
import requests
|
|
|
|
import activitypub
|
|
import common
|
|
from models import MagicKey
|
|
import testutil
|
|
import webmention
|
|
from webmention import app
|
|
|
|
|
|
@mock.patch('requests.post')
|
|
@mock.patch('requests.get')
|
|
class WebmentionTest(testutil.TestCase):
|
|
|
|
def setUp(self):
|
|
super(WebmentionTest, self).setUp()
|
|
self.orig = requests_response("""\
|
|
<html>
|
|
<meta>
|
|
<link href='http://orig/atom' rel='alternate' type='application/atom+xml'>
|
|
</meta>
|
|
</html>
|
|
""", url='http://orig/post', content_type='text/html; charset=utf-8')
|
|
|
|
self.reply_html = """\
|
|
<html>
|
|
<body>
|
|
<div class="h-entry">
|
|
<a class="u-url" href="http://a/reply"></a>
|
|
<p class="e-content">
|
|
<a class="u-in-reply-to" href="http://orig/post">foo ☕ bar</a>
|
|
</p>
|
|
</div>
|
|
</body>
|
|
</html>
|
|
"""
|
|
self.reply = requests_response(
|
|
self.reply_html, content_type='text/html; charset=utf-8')
|
|
mf2 = mf2py.parse(self.reply_html, url='http://a/reply')
|
|
self.reply_obj = microformats2.json_to_object(mf2['items'][0])
|
|
|
|
def test_activitypub(self, mock_get, mock_post):
|
|
article = requests_response({
|
|
'@context': ['https://www.w3.org/ns/activitystreams'],
|
|
'type': 'Article',
|
|
'content': 'Lots of ☕ words...',
|
|
'actor': {
|
|
'url': 'http://orig/author',
|
|
},
|
|
})
|
|
actor = requests_response({
|
|
'objectType' : 'person',
|
|
'displayName': 'Mrs. ☕ Foo',
|
|
'url': 'https://foo.com/about-me',
|
|
'inbox': 'https://foo.com/inbox',
|
|
})
|
|
|
|
mock_get.side_effect = [self.reply, article, actor]
|
|
|
|
got = app.get_response(
|
|
'/webmention', method='POST', body=urllib.urlencode({
|
|
'source': 'http://a/reply',
|
|
'target': 'http://orig/post',
|
|
}))
|
|
self.assertEquals(200, got.status_int)
|
|
|
|
mock_get.assert_has_calls((
|
|
call('http://a/reply', headers=common.HEADERS, timeout=util.HTTP_TIMEOUT),
|
|
call('http://orig/post', headers=activitypub.CONNEG_HEADER,
|
|
timeout=util.HTTP_TIMEOUT),
|
|
call('http://orig/author', headers=activitypub.CONNEG_HEADER,
|
|
timeout=util.HTTP_TIMEOUT),))
|
|
|
|
args, kwargs = mock_post.call_args
|
|
self.assertEqual(('https://foo.com/inbox',), args)
|
|
self.assertEqual({
|
|
'objectType': 'comment',
|
|
'url': 'http://a/reply',
|
|
'displayName': 'foo ☕ bar',
|
|
'content': ' <a class="u-in-reply-to" href="http://orig/post">foo ☕ bar</a> ',
|
|
'inReplyTo': 'http://orig/post',
|
|
'cc': [
|
|
activitypub.PUBLIC_AUDIENCE,
|
|
'http://orig/post'
|
|
],
|
|
}, kwargs['json'])
|
|
|
|
expected_headers = copy.copy(common.HEADERS)
|
|
expected_headers['Content-Type'] = activitypub.CONTENT_TYPE_AS
|
|
self.assertEqual(expected_headers, kwargs['headers'])
|
|
|
|
def test_salmon(self, mock_get, mock_post):
|
|
orig_atom = requests_response("""\
|
|
<?xml version="1.0"?>
|
|
<entry xmlns="http://www.w3.org/2005/Atom">
|
|
<id>tag:fed.brid.gy,2017-08-22:orig-post</id>
|
|
<link rel="salmon" href="http://orig/salmon"/>
|
|
<content type="html">baz ☕ baj</content>
|
|
</entry>
|
|
""")
|
|
mock_get.side_effect = [self.reply, self.orig, orig_atom]
|
|
|
|
got = app.get_response(
|
|
'/webmention', method='POST', body=urllib.urlencode({
|
|
'source': 'http://a/reply',
|
|
'target': 'http://orig/post',
|
|
}))
|
|
self.assertEquals(200, got.status_int)
|
|
|
|
mock_get.assert_has_calls((
|
|
call('http://a/reply', headers=common.HEADERS, timeout=util.HTTP_TIMEOUT),
|
|
call('http://orig/post', headers=activitypub.CONNEG_HEADER,
|
|
timeout=util.HTTP_TIMEOUT),
|
|
call('http://orig/atom', headers=common.HEADERS, timeout=util.HTTP_TIMEOUT),
|
|
))
|
|
|
|
args, kwargs = mock_post.call_args
|
|
self.assertEqual(('http://orig/salmon',), args)
|
|
self.assertEqual(common.MAGIC_ENVELOPE_CONTENT_TYPE,
|
|
kwargs['headers']['Content-Type'])
|
|
|
|
env = utils.parse_magic_envelope(kwargs['data'])
|
|
self.reply_obj['inReplyTo'][0]['id'] = 'tag:fed.brid.gy,2017-08-22:orig-post'
|
|
reply_atom = atom.activity_to_atom(
|
|
{'object': self.reply_obj}, xml_base='http://a/reply')
|
|
key = MagicKey.get_by_id('a')
|
|
assert magicsigs.verify(None, reply_atom, env['sig'], key=key)
|
|
|
|
data = utils.decode(env['data'])
|
|
parsed = feedparser.parse(data)
|
|
entry = parsed.entries[0]
|
|
|
|
self.assertEquals('http://a/reply', entry['id'])
|
|
self.assertIn({
|
|
'rel': 'alternate',
|
|
'href': 'http://a/reply',
|
|
'type': 'text/html',
|
|
}, entry['links'])
|
|
self.assertEquals({
|
|
'type': 'text/html',
|
|
'href': 'http://orig/post',
|
|
'ref': 'tag:fed.brid.gy,2017-08-22:orig-post'
|
|
}, entry['thr_in-reply-to'])
|
|
self.assertEquals(
|
|
'<a class="u-in-reply-to" href="http://orig/post">foo ☕ bar</a>',
|
|
entry.content[0]['value'])
|
|
|
|
def test_salmon_get_salmon_from_webfinger(self, mock_get, mock_post):
|
|
orig_atom = requests_response("""\
|
|
<?xml version="1.0"?>
|
|
<entry xmlns="http://www.w3.org/2005/Atom">
|
|
<author>
|
|
<name>ryan</name>
|
|
<email>ryan@orig</email>
|
|
</author>
|
|
<id>tag:fed.brid.gy,2017-08-22:orig-post</id>
|
|
</entry>
|
|
""")
|
|
webfinger = requests_response({
|
|
'subject': 'acct:ryan@orig',
|
|
'links': [{
|
|
'rel': 'salmon',
|
|
'href': 'http://orig/@ryan/salmon',
|
|
}],
|
|
})
|
|
mock_get.side_effect = [self.reply, self.orig, orig_atom, webfinger]
|
|
|
|
got = app.get_response('/webmention', method='POST', body=urllib.urlencode({
|
|
'source': 'http://a/reply',
|
|
'target': 'http://orig/post',
|
|
}))
|
|
self.assertEquals(200, got.status_int)
|
|
|
|
mock_get.assert_any_call(
|
|
'http://orig/.well-known/webfinger?resource=ryan@orig',
|
|
headers=common.HEADERS, timeout=util.HTTP_TIMEOUT)
|
|
self.assertEqual(('http://orig/@ryan/salmon',), mock_post.call_args[0])
|
|
|
|
def test_salmon_no_target_atom(self, mock_get, mock_post):
|
|
orig_no_atom = requests_response("""\
|
|
<html>
|
|
<body>foo</body>
|
|
</html>""", 'http://orig/url')
|
|
mock_get.side_effect = [self.reply, orig_no_atom]
|
|
|
|
got = app.get_response('/webmention', method='POST', body=urllib.urlencode({
|
|
'source': 'http://a/reply',
|
|
'target': 'http://orig/post',
|
|
}))
|
|
self.assertEquals(400, got.status_int)
|
|
self.assertIn('Target post http://orig/url has no Atom link', got.body)
|