2023-03-14 00:25:10 +00:00
|
|
|
"""Unit tests for protocol.py."""
|
|
|
|
|
from unittest.mock import patch
|
|
|
|
|
|
2023-03-20 21:28:14 +00:00
|
|
|
from flask import g
|
2023-03-14 00:25:10 +00:00
|
|
|
from oauth_dropins.webutil.testutil import requests_response
|
|
|
|
|
import requests
|
|
|
|
|
|
|
|
|
|
from protocol import Protocol
|
|
|
|
|
from app import app
|
|
|
|
|
from models import Follower, Object, User
|
|
|
|
|
|
|
|
|
|
from .test_activitypub import ACTOR, REPLY
|
|
|
|
|
from . import testutil
|
2023-03-27 21:12:06 +00:00
|
|
|
from .testutil import FakeProtocol
|
2023-03-14 00:25:10 +00:00
|
|
|
|
2023-03-14 03:38:17 +00:00
|
|
|
REPLY = {
|
|
|
|
|
**REPLY,
|
|
|
|
|
'actor': ACTOR,
|
|
|
|
|
'object': {
|
|
|
|
|
**REPLY['object'],
|
|
|
|
|
'author': ACTOR,
|
|
|
|
|
},
|
|
|
|
|
}
|
|
|
|
|
|
2023-03-14 00:25:10 +00:00
|
|
|
|
|
|
|
|
class ProtocolTest(testutil.TestCase):
|
|
|
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
|
super().setUp()
|
|
|
|
|
self.user = self.make_user('foo.com', has_hcard=True)
|
|
|
|
|
self.app_context = app.test_request_context('/')
|
|
|
|
|
self.app_context.__enter__()
|
2023-03-20 21:28:14 +00:00
|
|
|
g.user = None
|
2023-03-14 00:25:10 +00:00
|
|
|
|
|
|
|
|
def tearDown(self):
|
|
|
|
|
self.app_context.__exit__(None, None, None)
|
|
|
|
|
super().tearDown()
|
|
|
|
|
|
|
|
|
|
@patch('requests.get')
|
2023-03-14 03:38:17 +00:00
|
|
|
def test_receive_reply_not_feed_not_notification(self, mock_get):
|
|
|
|
|
Follower.get_or_create(ACTOR['id'], 'foo.com')
|
2023-03-14 00:25:10 +00:00
|
|
|
|
2023-03-19 22:43:55 +00:00
|
|
|
# user.com webmention discovery
|
2023-03-14 00:25:10 +00:00
|
|
|
mock_get.return_value = requests_response('<html></html>')
|
|
|
|
|
|
2023-03-20 21:28:14 +00:00
|
|
|
Protocol.receive(REPLY['id'], as2=REPLY)
|
2023-03-14 03:38:17 +00:00
|
|
|
|
|
|
|
|
self.assert_object(REPLY['id'],
|
|
|
|
|
as2=REPLY,
|
2023-03-14 00:25:10 +00:00
|
|
|
type='post',
|
2023-03-19 22:43:55 +00:00
|
|
|
domains=['user.com'],
|
2023-03-14 03:38:17 +00:00
|
|
|
# not feed since it's a reply
|
|
|
|
|
# not notification since it doesn't involve the user
|
|
|
|
|
labels=['activity'],
|
2023-03-14 00:25:10 +00:00
|
|
|
status='complete',
|
|
|
|
|
)
|
2023-03-14 03:38:17 +00:00
|
|
|
self.assert_object(REPLY['object']['id'],
|
|
|
|
|
as2=REPLY['object'],
|
2023-03-14 00:25:10 +00:00
|
|
|
type='comment',
|
|
|
|
|
)
|
2023-03-27 21:12:06 +00:00
|
|
|
|
2023-03-29 20:13:32 +00:00
|
|
|
def test_load(self):
|
2023-03-27 21:12:06 +00:00
|
|
|
obj = Object(id='foo', our_as1={'x': 'y'})
|
|
|
|
|
FakeProtocol.objects = {'foo': obj}
|
2023-03-29 20:13:32 +00:00
|
|
|
self.assert_entities_equal(obj, FakeProtocol.load('foo'))
|
2023-03-27 21:12:06 +00:00
|
|
|
self.assertIsNotNone(Object.get_by_id('foo'))
|
|
|
|
|
self.assertEqual(['foo'], FakeProtocol.fetched)
|
|
|
|
|
|
2023-03-29 20:13:32 +00:00
|
|
|
def test_load_already_stored(self):
|
2023-03-27 21:12:06 +00:00
|
|
|
stored = Object(id='foo', our_as1={'x': 'y'})
|
|
|
|
|
stored.put()
|
2023-03-29 20:13:32 +00:00
|
|
|
self.assert_entities_equal(stored, FakeProtocol.load('foo'))
|
2023-03-27 21:12:06 +00:00
|
|
|
self.assertEqual([], FakeProtocol.fetched)
|
2023-03-29 19:48:50 +00:00
|
|
|
|
|
|
|
|
@patch('requests.get')
|
2023-03-29 20:13:32 +00:00
|
|
|
def test_load_empty_deleted(self, mock_get):
|
2023-03-29 19:48:50 +00:00
|
|
|
stored = Object(id='foo', deleted=True)
|
|
|
|
|
stored.put()
|
|
|
|
|
|
2023-03-29 20:13:32 +00:00
|
|
|
self.assert_entities_equal(stored, FakeProtocol.load('foo'))
|
2023-03-29 19:48:50 +00:00
|
|
|
mock_get.assert_not_called()
|
