kopia lustrzana https://github.com/snarfed/bridgy-fed
474 wiersze
18 KiB
Python
474 wiersze
18 KiB
Python
"""Integration tests."""
|
|
import copy
|
|
from unittest.mock import patch
|
|
|
|
from arroba.datastore_storage import DatastoreStorage
|
|
from arroba.repo import Repo
|
|
from dns.resolver import NXDOMAIN
|
|
from granary import as2
|
|
from granary.tests.test_bluesky import ACTOR_PROFILE_BSKY, POST_BSKY
|
|
from oauth_dropins.webutil.flask_util import NoContent
|
|
from oauth_dropins.webutil.testutil import requests_response
|
|
from oauth_dropins.webutil import util
|
|
from oauth_dropins.webutil.util import json_dumps, json_loads
|
|
|
|
from activitypub import ActivityPub
|
|
import app
|
|
from atproto import ATProto
|
|
import hub
|
|
from models import Object, Target
|
|
from web import Web
|
|
|
|
from .testutil import ATPROTO_KEY, TestCase
|
|
from .test_activitypub import ACTOR
|
|
from . import test_atproto
|
|
from . import test_web
|
|
|
|
DID_DOC = {
|
|
**test_atproto.DID_DOC,
|
|
'id': 'did:plc:alice',
|
|
'alsoKnownAs': ['at://alice.com'],
|
|
}
|
|
PROFILE_GETRECORD = {
|
|
'uri': 'at://did:plc:alice/app.bsky.actor.profile/self',
|
|
'cid': 'alice sidd',
|
|
'value': test_atproto.ACTOR_PROFILE_BSKY,
|
|
}
|
|
|
|
|
|
@patch('ids.COPIES_PROTOCOLS', ['atproto'])
|
|
class IntegrationTests(TestCase):
|
|
|
|
@patch('requests.post')
|
|
@patch('requests.get')
|
|
def test_atproto_notify_reply_to_activitypub(self, mock_get, mock_post):
|
|
"""ATProto poll notifications, deliver reply to ActivityPub.
|
|
|
|
ActivityPub original post http://inst/post by bob
|
|
ATProto reply 123 by alice.com (did:plc:alice)
|
|
|
|
https://github.com/snarfed/bridgy-fed/issues/720
|
|
"""
|
|
# setup
|
|
self.store_object(id='did:plc:alice', raw=DID_DOC)
|
|
alice = self.make_user(id='did:plc:alice', cls=ATProto)
|
|
|
|
storage = DatastoreStorage()
|
|
Repo.create(storage, 'did:plc:bob', signing_key=ATPROTO_KEY)
|
|
bob = self.make_user(
|
|
id='http://inst/bob',
|
|
cls=ActivityPub,
|
|
copies=[Target(uri='did:plc:bob', protocol='atproto')],
|
|
enabled_protocols=['atproto'],
|
|
obj_as2={
|
|
'id': 'http://inst/bob',
|
|
'inbox': 'http://inst/bob/inbox',
|
|
})
|
|
|
|
self.store_object(id='http://inst/post', source_protocol='activitypub',
|
|
our_as1={
|
|
'objectType': 'note',
|
|
'author': 'http://inst/bob',
|
|
},
|
|
copies=[
|
|
Target(uri='at://did:plc:bob/app.bsky.feed.post/123', protocol='atproto'),
|
|
])
|
|
|
|
mock_get.side_effect = [
|
|
# ATProto listNotifications
|
|
requests_response({
|
|
'cursor': '...',
|
|
'notifications': [{
|
|
'uri': 'at://did:plc:alice/app.bsky.feed.post/456',
|
|
'cid': '...',
|
|
'author': {
|
|
'$type': 'app.bsky.actor.defs#profileView',
|
|
'did': 'did:plc:alice',
|
|
'handle': 'alice.com',
|
|
},
|
|
'reason': 'reply',
|
|
'record': {
|
|
'$type': 'app.bsky.feed.post',
|
|
'text': 'I hereby reply',
|
|
'reply': {
|
|
'root': {
|
|
'cid': '...',
|
|
'uri': 'at://did:plc:bob/app.bsky.feed.post/123',
|
|
},
|
|
'parent': {
|
|
'cid': '...',
|
|
'uri': 'at://did:plc:bob/app.bsky.feed.post/123',
|
|
}
|
|
},
|
|
},
|
|
}],
|
|
}),
|
|
# ATProto getRecord of alice's profile
|
|
requests_response(PROFILE_GETRECORD),
|
|
]
|
|
|
|
resp = self.post('/queue/atproto-poll-notifs', client=hub.app.test_client())
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
web_test = test_web.WebTest()
|
|
web_test.user = alice
|
|
web_test.assert_deliveries(mock_post, ['http://inst/bob/inbox'], data={
|
|
'@context': 'https://www.w3.org/ns/activitystreams',
|
|
'type': 'Create',
|
|
'id': 'https://bsky.brid.gy/convert/ap/at://did:plc:alice/app.bsky.feed.post/456#bridgy-fed-create',
|
|
'actor': 'https://bsky.brid.gy/ap/did:plc:alice',
|
|
'published': '2022-01-02T03:04:05+00:00',
|
|
'object': {
|
|
'type': 'Note',
|
|
'id': 'https://bsky.brid.gy/convert/ap/at://did:plc:alice/app.bsky.feed.post/456',
|
|
'url': 'http://localhost/r/https://bsky.app/profile/did:plc:alice/post/456',
|
|
'attributedTo': 'https://bsky.brid.gy/ap/did:plc:alice',
|
|
'content': 'I hereby reply',
|
|
'contentMap': {'en': 'I hereby reply'},
|
|
'inReplyTo': 'http://inst/post',
|
|
'tag': [{'type': 'Mention', 'href': 'http://inst/bob'}],
|
|
'to': ['https://www.w3.org/ns/activitystreams#Public'],
|
|
'cc': ['http://inst/bob'],
|
|
},
|
|
'to': ['https://www.w3.org/ns/activitystreams#Public'],
|
|
})
|
|
|
|
|
|
@patch('requests.post', return_value=requests_response(''))
|
|
@patch('requests.get')
|
|
def test_atproto_follow_to_web(self, mock_get, mock_post):
|
|
"""ATProto poll notifications, deliver follow to Web.
|
|
|
|
ATProto user alice.com (did:plc:alice)
|
|
ATProto follow at://did:plc:alice/app.bsky.graph.follow/123
|
|
Web user bob.com
|
|
"""
|
|
# setup
|
|
self.store_object(id='did:plc:alice', raw=DID_DOC)
|
|
alice = self.make_user(id='did:plc:alice', cls=ATProto)
|
|
|
|
storage = DatastoreStorage()
|
|
Repo.create(storage, 'did:plc:bob', signing_key=ATPROTO_KEY)
|
|
bob = self.make_user(id='bob.com', cls=Web,
|
|
copies=[Target(uri='did:plc:bob', protocol='atproto')],
|
|
enabled_protocols=['atproto'])
|
|
|
|
mock_get.side_effect = [
|
|
# ATProto listNotifications
|
|
requests_response({
|
|
'cursor': '...',
|
|
'notifications': [{
|
|
'uri': 'at://did:plc:alice/app.bsky.graph.follow/123',
|
|
'cid': '...',
|
|
'author': {
|
|
'$type': 'app.bsky.actor.defs#profileView',
|
|
'did': 'did:plc:alice',
|
|
'handle': 'alice.com',
|
|
},
|
|
'reason': 'follow',
|
|
'record': {
|
|
'$type': 'app.bsky.graph.follow',
|
|
'subject': 'did:plc:bob',
|
|
'createdAt': '2022-01-02T03:04:05.000Z',
|
|
},
|
|
}],
|
|
}),
|
|
# ATProto getRecord of alice's profile
|
|
requests_response(PROFILE_GETRECORD),
|
|
# webmention discovery
|
|
test_web.WEBMENTION_REL_LINK,
|
|
]
|
|
|
|
resp = self.post('/queue/atproto-poll-notifs', client=hub.app.test_client())
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
self.assert_req(mock_get, 'https://bob.com/')
|
|
self.assert_req(mock_post, 'https://bob.com/webmention', data={
|
|
'source': 'https://bsky.brid.gy/convert/web/at://did:plc:alice/app.bsky.graph.follow/123',
|
|
'target': 'https://bob.com/',
|
|
}, allow_redirects=False, headers={'Accept': '*/*'})
|
|
|
|
|
|
@patch('dns.resolver.resolve', side_effect=NXDOMAIN())
|
|
@patch('oauth_dropins.webutil.appengine_config.tasks_client.create_task')
|
|
@patch('requests.post', side_effect=[
|
|
requests_response('OK'), # create DID
|
|
])
|
|
@patch('requests.get', side_effect = [
|
|
# webmention source page, follow HTML
|
|
requests_response("""\
|
|
<html>
|
|
<body class="h-entry">
|
|
<a class="u-url" href="https://bob.com/follow"></a>
|
|
<a class="u-follow-of" href="https://bsky.app/profile/alice.com"></a>
|
|
<a href="http://localhost/"></a>
|
|
</body>
|
|
</html>
|
|
"""),
|
|
# https://bob.com/ , for authorship
|
|
requests_response("""\
|
|
<html>
|
|
<body class="h-card">
|
|
<a class="p-name u-url" rel="me" href="https://bob.com/">Bob</a>
|
|
</body>
|
|
</html>
|
|
"""),
|
|
# alice.com handle resolution, HTTPS method
|
|
requests_response('did:plc:alice', content_type='text/plain'),
|
|
# alice profile
|
|
requests_response(PROFILE_GETRECORD),
|
|
# alice DID
|
|
requests_response(DID_DOC),
|
|
# alice profile
|
|
requests_response(PROFILE_GETRECORD),
|
|
])
|
|
def test_web_follow_of_atproto(self, mock_get, mock_post, _, __):
|
|
"""Incoming webmention for a web follow of an ATProto bsky.app profile URL.
|
|
|
|
Web user bob.com
|
|
ATProto user alice.com (did:plc:alice)
|
|
Follow is HTML with mf2 u-follow-of of https://bsky.app/profile/alice.com
|
|
"""
|
|
bob = self.make_user(id='bob.com', cls=Web, enabled_protocols=['atproto'],
|
|
obj_mf2={
|
|
'type': ['h-card'],
|
|
'properties': {
|
|
'url': ['https://bob.com/'],
|
|
'name': ['Bob'],
|
|
},
|
|
})
|
|
|
|
# send webmention
|
|
resp = self.post('/webmention', data={
|
|
'source': 'https://bob.com/follow',
|
|
'target': 'http://localhost/',
|
|
})
|
|
self.assertEqual(202, resp.status_code)
|
|
|
|
# check results
|
|
bob = bob.key.get()
|
|
self.assertEqual(1, len(bob.copies))
|
|
self.assertEqual('atproto', bob.copies[0].protocol)
|
|
bob_did = bob.copies[0].uri
|
|
|
|
self.assertEqual({
|
|
'type': ['h-entry'],
|
|
'properties': {
|
|
'url': ['https://bob.com/follow'],
|
|
'follow-of': ['https://bsky.app/profile/alice.com'],
|
|
'name': [''],
|
|
'author': [{
|
|
'type': ['h-card'],
|
|
'properties': {
|
|
'name': ['Bob'],
|
|
'url': ['https://bob.com/'],
|
|
},
|
|
}],
|
|
},
|
|
}, Object.get_by_id('https://bob.com/follow').mf2)
|
|
|
|
storage = DatastoreStorage()
|
|
repo = storage.load_repo('bob.com.web.brid.gy')
|
|
self.assertEqual(bob_did, repo.did)
|
|
|
|
records = repo.get_contents()
|
|
self.assertEqual(['app.bsky.actor.profile', 'app.bsky.graph.follow'],
|
|
list(records.keys()))
|
|
self.assertEqual(['self'], list(records['app.bsky.actor.profile'].keys()))
|
|
self.assertEqual([{
|
|
'$type': 'app.bsky.graph.follow',
|
|
'subject': 'did:plc:alice',
|
|
'createdAt': '2022-01-02T03:04:05.000Z',
|
|
}], list(records['app.bsky.graph.follow'].values()))
|
|
|
|
|
|
@patch('oauth_dropins.webutil.appengine_config.tasks_client.create_task')
|
|
@patch('requests.get', side_effect=[
|
|
# getRecord of original post
|
|
# alice profile
|
|
requests_response({
|
|
'uri': 'at://did:plc:alice/app.bsky.feed.post/123',
|
|
'cid': 'sydd',
|
|
'value': POST_BSKY,
|
|
}),
|
|
])
|
|
def test_activitypub_like_of_atproto(self, mock_get, _):
|
|
"""AP inbox delivery of a Like of an ATProto bsky.app profile URL.
|
|
|
|
ActivityPub user @bob@inst , https://inst/bob
|
|
ATProto user alice.com (did:plc:alice)
|
|
Like is https://inst/like
|
|
"""
|
|
self.store_object(id='did:plc:alice', raw=DID_DOC)
|
|
alice = self.make_user(id='did:plc:alice', cls=ATProto)
|
|
|
|
storage = DatastoreStorage()
|
|
Repo.create(storage, 'did:plc:bob', signing_key=ATPROTO_KEY)
|
|
bob = self.make_user(id='https://inst/bob', cls=ActivityPub,
|
|
copies=[Target(uri='did:plc:bob', protocol='atproto')],
|
|
obj_as2={
|
|
'type': 'Person',
|
|
'id': 'https://inst/bob',
|
|
'name': 'Bob',
|
|
})
|
|
|
|
bob_did_doc = copy.deepcopy(test_atproto.DID_DOC)
|
|
bob_did_doc['service'][0]['serviceEndpoint'] = ATProto.PDS_URL
|
|
bob_did_doc.update({
|
|
'id': 'did:plc:bob',
|
|
'alsoKnownAs': ['at://bob.inst.ap.brid.gy'],
|
|
})
|
|
self.store_object(id='did:plc:bob', raw=bob_did_doc)
|
|
|
|
# existing Object with original post, *without* cid. we should refetch.
|
|
Object(id='at://did:plc:alice/app.bsky.feed.post/123', bsky=POST_BSKY).put()
|
|
|
|
# inbox delivery
|
|
like = {
|
|
'type': 'Like',
|
|
'id': 'http://inst/like',
|
|
'actor': 'https://inst/bob',
|
|
'object': 'https://bsky.brid.gy/convert/ap/at://did:plc:alice/app.bsky.feed.post/123',
|
|
}
|
|
resp = self.post('/ap/atproto/did:plc:alice/inbox', json=like)
|
|
self.assertEqual(202, resp.status_code)
|
|
|
|
# check results
|
|
self.assertEqual({
|
|
**like,
|
|
# TODO: stop normalizing this in the original protocol's data
|
|
'object': 'at://did:plc:alice/app.bsky.feed.post/123',
|
|
}, Object.get_by_id('http://inst/like').as2)
|
|
|
|
repo = storage.load_repo('did:plc:bob')
|
|
|
|
records = repo.get_contents()
|
|
self.assertEqual(['app.bsky.feed.like'], list(records.keys()))
|
|
self.assertEqual([{
|
|
'$type': 'app.bsky.feed.like',
|
|
'subject': {
|
|
'uri': 'at://did:plc:alice/app.bsky.feed.post/123',
|
|
'cid': 'sydd',
|
|
},
|
|
'createdAt': '2022-01-02T03:04:05.000Z',
|
|
}], list(records['app.bsky.feed.like'].values()))
|
|
|
|
# we needed to refetch the original post
|
|
self.assert_object(id='at://did:plc:alice/app.bsky.feed.post/123',
|
|
source_protocol='atproto', bsky={
|
|
**POST_BSKY,
|
|
'cid': 'sydd',
|
|
})
|
|
|
|
|
|
@patch('requests.post', return_value=requests_response('OK')) # create DID
|
|
@patch('requests.get')
|
|
def test_activitypub_follow_bsky_bot_user_enables_protocol(self, mock_get, mock_post):
|
|
"""AP follow of @bsky.brid.gy@bsky.brid.gy bridges the account into BLuesky.
|
|
|
|
ActivityPub user @alice@inst , https://inst/alice
|
|
ATProto bot user bsky.brid.gy (did:plc:bsky)
|
|
Follow is https://inst/follow
|
|
"""
|
|
mock_get.return_value = self.as2_resp({
|
|
'type': 'Person',
|
|
'id': 'https://inst/alice',
|
|
'name': 'Mrs. ☕ Alice',
|
|
'preferredUsername': 'alice',
|
|
'inbox': 'http://inst/inbox',
|
|
})
|
|
self.make_user(id='bsky.brid.gy', cls=Web, ap_subdomain='bsky')
|
|
|
|
# deliver follow
|
|
resp = self.post('/bsky.brid.gy/inbox', json={
|
|
'type': 'Follow',
|
|
'id': 'http://inst/follow',
|
|
'actor': 'https://inst/alice',
|
|
'object': 'https://bsky.brid.gy/bsky.brid.gy',
|
|
})
|
|
self.assertEqual(204, resp.status_code)
|
|
|
|
# check results
|
|
user = ActivityPub.get_by_id('https://inst/alice')
|
|
self.assertTrue(ActivityPub.is_enabled_to(ATProto, user=user))
|
|
|
|
self.assertEqual(1, len(user.copies))
|
|
self.assertEqual('atproto', user.copies[0].protocol)
|
|
did = user.copies[0].uri
|
|
|
|
storage = DatastoreStorage()
|
|
repo = storage.load_repo('alice.inst.ap.brid.gy')
|
|
self.assertEqual(did, repo.did)
|
|
|
|
records = repo.get_contents()
|
|
self.assertEqual(['app.bsky.actor.profile'], list(records.keys()))
|
|
self.assertEqual(['self'], list(records['app.bsky.actor.profile'].keys()))
|
|
|
|
args, kwargs = mock_post.call_args_list[1]
|
|
self.assert_equals(('http://inst/inbox',), args)
|
|
self.assert_equals({
|
|
'type': 'Accept',
|
|
'id': 'http://localhost/r/bsky.brid.gy/followers#accept-http://inst/follow',
|
|
'actor': 'https://bsky.brid.gy/bsky.brid.gy',
|
|
'object': {
|
|
'actor': 'https://inst/alice',
|
|
'id': 'http://inst/follow',
|
|
'url': 'https://inst/alice#followed-bsky.brid.gy',
|
|
'type': 'Follow',
|
|
'object': 'https://bsky.brid.gy/bsky.brid.gy',
|
|
},
|
|
}, json_loads(kwargs['data']), ignore=['to', '@context'])
|
|
|
|
@patch('requests.post')
|
|
@patch('requests.get')
|
|
def test_atproto_follow_ap_bot_user_enables_protocol(self, mock_get, mock_post):
|
|
"""Bluesky follow of @ap.brid.gy enables the ActivityPub protocol.
|
|
|
|
ATProto user alice.com, did:plc:alice
|
|
ActivityPub bot user @ap.brid.gy, did:plc:ap
|
|
"""
|
|
self.make_user(id='ap.brid.gy', cls=Web, ap_subdomain='ap',
|
|
enabled_protocols=['atproto'],
|
|
copies=[Target(uri='did:plc:ap', protocol='atproto')])
|
|
self.store_object(id='did:plc:ap', raw={
|
|
**DID_DOC,
|
|
'id': 'did:plc:ap',
|
|
'alsoKnownAs': ['at://ap.brid.gy'],
|
|
})
|
|
storage = DatastoreStorage()
|
|
Repo.create(storage, 'did:plc:ap', signing_key=ATPROTO_KEY)
|
|
|
|
mock_get.side_effect = [
|
|
# ATProto listNotifications
|
|
requests_response({
|
|
'cursor': '...',
|
|
'notifications': [{
|
|
'uri': 'at://did:plc:alice/app.bsky.graph.follow/456',
|
|
'cid': '...',
|
|
'author': {
|
|
'$type': 'app.bsky.actor.defs#profileView',
|
|
'did': 'did:plc:alice',
|
|
'handle': 'alice.com',
|
|
},
|
|
'reason': 'follow',
|
|
'record': {
|
|
'$type': 'app.bsky.graph.follow',
|
|
'subject': 'did:plc:ap',
|
|
},
|
|
}],
|
|
}),
|
|
# alice DID
|
|
requests_response(DID_DOC),
|
|
# alice profile
|
|
requests_response(PROFILE_GETRECORD),
|
|
# alice.com handle resolution, HTTPS method
|
|
# requests_response('did:plc:alice', content_type='text/plain'),
|
|
# # alice profile
|
|
# requests_response(PROFILE_GETRECORD),
|
|
]
|
|
resp = self.post('/queue/atproto-poll-notifs', client=hub.app.test_client())
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
user = ATProto.get_by_id('did:plc:alice')
|
|
self.assertTrue(ATProto.is_enabled_to(ActivityPub, user=user))
|