async def test_result_on_third_page(mock_client):
username = "******"
image_id = 1560331512
post_id = 232347
event = MockTelegramEvent.with_message(
text="http://d.furaffinity.net/art/{0}/{1}/{1}.pic_of_me.png".format(
username, image_id),
client=mock_client,
)
neaten = NeatenFunctionality(MockExportAPI())
for page in [1, 2, 3]:
neaten.api.with_user_folder(
username,
"gallery", [
MockSubmission(post_id + 1 + (3 - page) * 5,
image_id=image_id + 16 + (3 - page) * 56),
MockSubmission(post_id + (3 - page) * 5,
image_id=image_id + (3 - page) * 56),
MockSubmission(post_id - 2 + (3 - page) * 5,
image_id=image_id - 27 + (3 - page) * 56),
MockSubmission(post_id - 3 + (3 - page) * 5,
image_id=image_id - 34 + (3 - page) * 56)
],
page=page)
submission = await neaten.api.get_full_submission(str(post_id))
with pytest.raises(StopPropagation):
await neaten.call(event)
submission._send_message.assert_called_once()
args, kwargs = submission._send_message.call_args
assert args[0] == mock_client
assert args[1] == event.input_chat
assert kwargs['reply_to'] == event.message.id
async def test_result_missing_between_pages(mock_client):
username = "******"
image_id = 1560331512
post_id = 232347
event = MockTelegramEvent.with_message(
text="http://d.furaffinity.net/art/{0}/{1}/{1}.pic_of_me.png".format(
username, image_id))
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_user_folder(
username,
"gallery", [
MockSubmission(post_id + 1, image_id=image_id + 16),
MockSubmission(post_id, image_id=image_id + 3)
],
page=1)
neaten.api.with_user_folder(
username,
"gallery", [
MockSubmission(post_id - 2, image_id=image_id - 27),
MockSubmission(post_id - 3, image_id=image_id - 34)
],
page=2)
with pytest.raises(StopPropagation):
await neaten.call(event)
event.reply.assert_called_with(
f"Error finding submission: Could not locate the image by {username} with image id {image_id}.",
)
async def test_submission_link_and_different_direct_link(mock_client):
username = "******"
image_id1 = 1560331512
image_id2 = image_id1 + 300
post_id1 = 232347
post_id2 = 233447
event = MockTelegramEvent.with_message(
text=
"https://furaffinity.net/view/{2}/ http://d.furaffinity.net/art/{0}/{1}/{1}.pic_of_me.png"
.format(username, image_id1, post_id2),
client=mock_client)
submission1 = MockSubmission(post_id1, image_id=image_id1)
submission2 = MockSubmission(post_id2, image_id=image_id2)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_user_folder(username, "gallery", [
submission2, submission1,
MockSubmission(post_id1 - 1, image_id=image_id1 - 15)
])
with pytest.raises(StopPropagation):
await neaten.call(event)
submission1._send_message.assert_called_once()
args1, kwargs1 = submission1._send_message.call_args
assert args1[0] == mock_client
assert args1[1] == event.input_chat
assert kwargs1['reply_to'] == event.message.id
submission2._send_message.assert_called_once()
args2, kwargs2 = submission2._send_message.call_args
assert args2[0] == mock_client
assert args2[1] == event.input_chat
assert kwargs2['reply_to'] == event.message.id
async def test_direct_link_and_matching_submission_link(mock_client):
username = "******"
image_id = 1560331512
post_id = 232347
event = MockTelegramEvent.with_message(
text=
"http://d.furaffinity.net/art/{0}/{1}/{1}.pic_of_me.png https://furaffinity.net/view/{2}/"
.format(username, image_id, post_id),
client=mock_client,
)
submission = MockSubmission(post_id, image_id=image_id)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_user_folder(
username, "gallery",
[submission,
MockSubmission(post_id - 1, image_id=image_id - 15)])
with pytest.raises(StopPropagation):
await neaten.call(event)
submission._send_message.assert_called_once()
args, kwargs = submission._send_message.call_args
assert args[0] == mock_client
assert args[1] == event.input_chat
assert kwargs['reply_to'] == event.message.id
async def test_thumb_and_different_submission_link(mock_client):
post_id1 = 382632
post_id2 = 382672
event = MockTelegramEvent.with_message(
text=
f"https://t.furaffinity.net/{post_id1}@1600-1562445328.jpg\nhttps://furaffinity.net/view/{post_id2}",
client=mock_client,
)
submission1 = MockSubmission(post_id1)
submission2 = MockSubmission(post_id2)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_submissions([submission1, submission2])
with pytest.raises(StopPropagation):
await neaten.call(event)
submission1._send_message.assert_called_once()
args1, kwargs1 = submission1._send_message.call_args
assert args1[0] == mock_client
assert args1[1] == event.input_chat
assert kwargs1['reply_to'] == event.message.id
submission2._send_message.assert_called_once()
args2, kwargs2 = submission2._send_message.call_args
assert args2[0] == mock_client
assert args2[1] == event.input_chat
assert kwargs2['reply_to'] == event.message.id
async def test_result_last_on_page(mock_client):
username = "******"
image_id = 1560331512
post_id = 232347
event = MockTelegramEvent.with_message(
text="http://d.furaffinity.net/art/{0}/{1}/{1}.pic_of_me.png".format(
username, image_id),
client=mock_client,
)
submission = MockSubmission(post_id, image_id=image_id)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_user_folder(username, "gallery", [
MockSubmission(post_id + 4, image_id=image_id + 16),
MockSubmission(post_id + 3, image_id=image_id + 2),
MockSubmission(post_id + 2, image_id=image_id + 1), submission
])
with pytest.raises(StopPropagation):
await neaten.call(event)
submission._send_message.assert_called_once()
args, kwargs = submission._send_message.call_args
assert args[0] == mock_client
assert args[1] == event.input_chat
assert kwargs['reply_to'] == event.message.id
async def test_ignore_message(mock_client):
event = MockTelegramEvent.with_message(text="hello world")
neaten = NeatenFunctionality(MockExportAPI())
await neaten.call(event)
event.reply.assert_not_called()
async def test_ignore_link(mock_client):
event = MockTelegramEvent.with_message(text="http://example.com")
neaten = NeatenFunctionality(MockExportAPI())
await neaten.call(event)
event.reply.assert_not_called()
async def test_two_submission_links(mock_client):
post_id1 = 23636984
post_id2 = 23636996
event = MockTelegramEvent.with_message(
text="furaffinity.net/view/{}\nfuraffinity.net/view/{}".format(
post_id1, post_id2),
client=mock_client,
)
submission1 = MockSubmission(post_id1)
submission2 = MockSubmission(post_id2)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_submissions([submission1, submission2])
with pytest.raises(StopPropagation):
await neaten.call(event)
submission1._send_message.assert_called_once()
args1, kwargs1 = submission1._send_message.call_args
assert args1[0] == mock_client
assert args1[1] == event.input_chat
assert kwargs1['reply_to'] == event.message.id
submission2._send_message.assert_called_once()
args2, kwargs2 = submission2._send_message.call_args
assert args2[0] == mock_client
assert args2[1] == event.input_chat
assert kwargs2['reply_to'] == event.message.id
async def test_ignore_journal_link(mock_client):
event = MockTelegramEvent.with_message(
text="https://www.furaffinity.net/journal/9150534/")
neaten = NeatenFunctionality(MockExportAPI())
await neaten.call(event)
event.reply.assert_not_called()
async def test_doesnt_fire_on_avatar(mock_client):
event = MockTelegramEvent.with_message(
text="https://a.furaffinity.net/1538326752/geordie79.gif",
client=mock_client,
)
neaten = NeatenFunctionality(MockExportAPI())
await neaten.call(event)
event.reply.assert_not_called()
async def test_ignore_channel_post(mock_client):
post_id = 23636984
event = MockTelegramEvent.with_channel_post(
text=f"https://www.furaffinity.net/view/{post_id}/")
submission = MockSubmission(post_id)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_submission(submission)
await neaten.call(event)
submission._send_message.assert_not_called()
async def test_deleted_submission_group_chat(mock_client):
post_id = 23636984
event = MockTelegramEvent.with_message(
text="furaffinity.net/view/{}".format(post_id),
chat_type=ChatType.GROUP)
neaten = NeatenFunctionality(MockExportAPI())
with pytest.raises(StopPropagation):
await neaten.call(event)
event.reply.assert_called_with("⏳ Neatening image link")
event.reply.return_value.delete.assert_called_once()
async def test_deleted_submission(mock_client):
post_id = 23636984
event = MockTelegramEvent.with_message(
text="furaffinity.net/view/{}".format(post_id))
neaten = NeatenFunctionality(MockExportAPI())
with pytest.raises(StopPropagation):
await neaten.call(event)
event.reply.assert_called_with(
f"This doesn't seem to be a valid FA submission: https://www.furaffinity.net/view/{post_id}/"
)
async def test_submission_link_in_group_caption(mock_client):
post_id = 23636984
event = MockTelegramEvent.with_message(
chat_type=ChatType.GROUP,
text=f"https://www.furaffinity.net/view/{post_id}/").with_document()
submission = MockSubmission(post_id)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_submission(submission)
await neaten.call(event)
submission._send_message.assert_not_called()
async def test_unknown_type_submission_groupchat(mock_client):
post_id = 23636984
event = MockTelegramEvent.with_message(
text="https://www.furaffinity.net/view/{}/".format(post_id),
chat_type=ChatType.GROUP)
submission = MockSubmission(post_id, file_ext="zzz")
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_submission(submission)
with pytest.raises(StopPropagation):
await neaten.call(event)
event.reply.assert_called_with("⏳ Neatening image link", )
event.reply.return_value.delete.assert_called_once()
def initialise_functionalities(self):
return [
BeepFunctionality(),
WelcomeFunctionality(),
ImageHashRecommendFunctionality(),
NeatenFunctionality(self.api),
InlineNeatenFunctionality(self.api),
InlineFunctionality(self.api),
InlineEditFunctionality(self.api, self.client),
InlineEditButtonPress(self.api),
SubscriptionFunctionality(self.subscription_watcher),
BlocklistFunctionality(self.subscription_watcher),
SupergroupUpgradeFunctionality(self.subscription_watcher),
UnhandledMessageFunctionality(),
]
async def test_submission_link_in_caption(mock_client):
post_id = 23636984
event = MockTelegramEvent.with_message(client=mock_client).with_photo(
caption=f"https://www.furaffinity.net/view/{post_id}/")
submission = MockSubmission(post_id)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_submission(submission)
with pytest.raises(StopPropagation):
await neaten.call(event)
submission._send_message.assert_called_once()
args, kwargs = submission._send_message.call_args
assert args[0] == mock_client
assert args[1] == event.input_chat
assert kwargs['reply_to'] == event.message.id
async def test_mp3_submission(mock_client):
post_id = 23636984
event = MockTelegramEvent.with_message(
text="https://www.furaffinity.net/view/{}/".format(post_id),
client=mock_client,
)
submission = MockSubmission(post_id, file_ext="mp3")
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_submission(submission)
with pytest.raises(StopPropagation):
await neaten.call(event)
submission._send_message.assert_called_once()
args, kwargs = submission._send_message.call_args
assert args[0] == mock_client
assert args[1] == event.input_chat
assert kwargs['reply_to'] == event.message.id
async def test_thumbnail_link_big(mock_client):
post_id = 382632
event = MockTelegramEvent.with_message(
text=f"https://t.furaffinity.net/{post_id}@1600-1562445328.jpg",
client=mock_client,
)
submission = MockSubmission(post_id)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_submission(submission)
with pytest.raises(StopPropagation):
await neaten.call(event)
submission._send_message.assert_called_once()
args, kwargs = submission._send_message.call_args
assert args[0] == mock_client
assert args[1] == event.input_chat
assert kwargs['reply_to'] == event.message.id
async def test_direct_in_progress_message_groupchat(mock_client):
username = "******"
image_id = 1560331512
post_id = 232347
event = MockTelegramEvent.with_message(
text="http://d.furaffinity.net/art/{0}/{1}/{1}.pic_of_me.png".format(
username, image_id))
goal_submission = MockSubmission(post_id, image_id=image_id)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_user_folder(
username, "gallery",
[goal_submission,
MockSubmission(post_id - 1, image_id=image_id - 15)])
with pytest.raises(StopPropagation):
await neaten.call(event)
event.reply.assert_called_with("⏳ Neatening image link")
event.reply.return_value.delete.assert_called_once()
async def test_cloudflare_error(mock_client):
post_id = 23636984
event = MockTelegramEvent.with_message(
text="https://www.furaffinity.net/view/{}/".format(post_id))
submission = MockSubmission(post_id)
api = MockExportAPI()
neaten = NeatenFunctionality(api)
def raise_cloudflare(*args, **kwargs):
raise CloudflareError()
api.get_full_submission = raise_cloudflare
with pytest.raises(StopPropagation):
await neaten.call(event)
submission._send_message.assert_not_called()
event.reply.assert_called_with(
"Furaffinity returned a cloudflare error, so I cannot neaten links.")
async def test_direct_no_match_groupchat(mock_client):
username = "******"
image_id = 1560331512
post_id = 232347
event = MockTelegramEvent.with_message(
text="http://d.furaffinity.net/art/{0}/{1}/{1}.pic_of_me.png".format(
username, image_id),
chat_type=ChatType.GROUP)
neaten = NeatenFunctionality(MockExportAPI())
for folder in ['gallery', 'scraps']:
neaten.api.with_user_folder(username, folder, [
MockSubmission(post_id, image_id=image_id + 4),
MockSubmission(post_id - 1, image_id=image_id - 15)
])
with pytest.raises(StopPropagation):
await neaten.call(event)
event.reply.assert_called_with("⏳ Neatening image link")
event.reply.return_value.delete.assert_called_once()
async def test_direct_no_match(mock_client):
username = "******"
image_id = 1560331512
post_id = 232347
event = MockTelegramEvent.with_message(
text="http://d.furaffinity.net/art/{0}/{1}/{1}.pic_of_me.png".format(
username, image_id))
neaten = NeatenFunctionality(MockExportAPI())
for folder in ['gallery', 'scraps']:
neaten.api.with_user_folder(username, folder, [
MockSubmission(post_id, image_id=image_id + 4),
MockSubmission(post_id - 1, image_id=image_id - 15)
])
with pytest.raises(StopPropagation):
await neaten.call(event)
event.reply.assert_called()
event.reply.assert_called_with(
f"Error finding submission: Could not locate the image by {username} with image id {image_id}.",
)
async def test_link_in_button(mock_client):
post_id = 23636984
event = MockTelegramEvent.with_message(
text="Hello",
client=mock_client,
).with_buttons([[
MockButton("View on E621", "https://e621.net/post/show/1699284"),
MockButton("View on FA", f"https://www.furaffinity.net/view/{post_id}")
], [MockButton("Visit my website", "https://example.com")]])
submission = MockSubmission(post_id)
neaten = NeatenFunctionality(MockExportAPI())
neaten.api.with_submission(submission)
with pytest.raises(StopPropagation):
await neaten.call(event)
submission._send_message.assert_called_once()
args, kwargs = submission._send_message.call_args
assert args[0] == mock_client
assert args[1] == event.input_chat
assert kwargs['reply_to'] == event.message.id
