def test_search_with_spaces(context): search_term = "deer YCH" update = MockTelegramUpdate.with_inline_query(query=search_term) post_id1 = 213231 post_id2 = 84331 submission1 = MockSubmission(post_id1) submission2 = MockSubmission(post_id2) inline = InlineFunctionality(MockExportAPI()) inline.api.with_search_results(search_term, [submission1, submission2]) inline.call(update, context) context.bot.answer_inline_query.assert_called_once() args = context.bot.answer_inline_query.call_args[0] assert context.bot.answer_inline_query.call_args[1]['next_offset'] == 2 assert args[0] == update.inline_query.id assert isinstance(args[1], list) assert len(args[1]) == 2 for result in args[1]: assert isinstance(result, InlineQueryResultPhoto) assert args[1][0].id == str(post_id1) assert args[1][1].id == str(post_id2) assert args[1][0].photo_url == submission1.thumbnail_url assert args[1][1].photo_url == submission2.thumbnail_url assert args[1][0].thumb_url == FASubmission.make_thumbnail_smaller( submission1.thumbnail_url) assert args[1][1].thumb_url == FASubmission.make_thumbnail_smaller( submission2.thumbnail_url) assert args[1][0].caption == submission1.link assert args[1][1].caption == submission2.link
def test_user_favourites(context): post_id1 = 234563 post_id2 = 393282 username = "******" update = MockTelegramUpdate.with_inline_query( query=f"favourites:{username}") submission1 = MockSubmission(post_id1) submission2 = MockSubmission(post_id2) inline = InlineFunctionality(MockExportAPI()) inline.api.with_user_favs(username, [submission1, submission2]) inline.call(update, context) context.bot.answer_inline_query.assert_called_once() args = context.bot.answer_inline_query.call_args[0] assert context.bot.answer_inline_query.call_args[1][ 'next_offset'] == submission2.fav_id assert args[0] == update.inline_query.id assert isinstance(args[1], list) assert len(args[1]) == 2 assert isinstance(args[1][0], InlineQueryResultPhoto) assert isinstance(args[1][1], InlineQueryResultPhoto) assert args[1][0].id == str(post_id1) assert args[1][1].id == str(post_id2) assert args[1][0].photo_url == submission1.thumbnail_url assert args[1][1].photo_url == submission2.thumbnail_url assert args[1][0].thumb_url == FASubmission.make_thumbnail_smaller( submission1.thumbnail_url) assert args[1][1].thumb_url == FASubmission.make_thumbnail_smaller( submission2.thumbnail_url) assert args[1][0].caption == submission1.link assert args[1][1].caption == submission2.link
def test_continue_from_fav_id(context): post_id = 234563 fav_id = "354233" username = "******" update = MockTelegramUpdate.with_inline_query(query=f"favs:{username}", offset=fav_id) submission = MockSubmission(post_id) inline = InlineFunctionality(MockExportAPI()) inline.api.with_user_favs(username, [submission], next_id=fav_id) inline.call(update, context) context.bot.answer_inline_query.assert_called_once() args = context.bot.answer_inline_query.call_args[0] assert context.bot.answer_inline_query.call_args[1][ 'next_offset'] == submission.fav_id assert args[0] == update.inline_query.id assert isinstance(args[1], list) assert len(args[1]) == 1 assert isinstance(args[1][0], InlineQueryResultPhoto) assert args[1][0].id == str(post_id) assert args[1][0].photo_url == submission.thumbnail_url assert args[1][0].thumb_url == FASubmission.make_thumbnail_smaller( submission.thumbnail_url) assert args[1][0].caption == submission.link
def test_over_48_favs(context): username = "******" post_ids = list(range(123456, 123456 + 72)) submissions = [MockSubmission(x) for x in post_ids] inline = InlineFunctionality(MockExportAPI()) inline.api.with_user_favs(username, submissions) update = MockTelegramUpdate.with_inline_query(query=f"favs:{username}") inline.call(update, context) context.bot.answer_inline_query.assert_called_once() args = context.bot.answer_inline_query.call_args[0] assert context.bot.answer_inline_query.call_args[1][ 'next_offset'] == submissions[47].fav_id assert args[0] == update.inline_query.id assert isinstance(args[1], list) assert len(args[1]) == 48 assert isinstance(args[1][0], InlineQueryResultPhoto) assert isinstance(args[1][1], InlineQueryResultPhoto) for x in range(48): assert args[1][x].id == str(post_ids[x]) assert args[1][x].photo_url == submissions[x].thumbnail_url assert args[1][x].thumb_url == FASubmission.make_thumbnail_smaller( submissions[x].thumbnail_url) assert args[1][x].caption == submissions[x].link
def test_to_inline_query_result(self): post_id = "1234" image_id = "5324543" link = f"https://furaffinity.net/view/{post_id}/" thumb_url = f"https://t.facdn.net/{post_id}@1600-{image_id}.jpg" title = "Example post" author = FAUser.from_short_dict({"name": "John", "profile_name": "john"}) submission = FASubmissionShort(post_id, thumb_url, title, author) query_result = submission.to_inline_query_result() assert query_result.id == post_id assert query_result.photo_url == thumb_url assert query_result.thumb_url == FASubmission.make_thumbnail_smaller(thumb_url) assert query_result.caption == link
def test_weird_characters_in_username(context): post_id = 234563 username = "******" update = MockTelegramUpdate.with_inline_query(query=f"gallery:{username}") submission = MockSubmission(post_id) inline = InlineFunctionality(MockExportAPI()) inline.api.with_user_folder(username, "gallery", [submission]) inline.call(update, context) context.bot.answer_inline_query.assert_called_once() args = context.bot.answer_inline_query.call_args[0] assert context.bot.answer_inline_query.call_args[1]['next_offset'] == 2 assert args[0] == update.inline_query.id assert isinstance(args[1], list) assert len(args[1]) == 1 assert isinstance(args[1][0], InlineQueryResultPhoto) assert args[1][0].id == str(post_id) assert args[1][0].photo_url == submission.thumbnail_url assert args[1][0].thumb_url == FASubmission.make_thumbnail_smaller(submission.thumbnail_url) assert args[1][0].caption == submission.link
def test_search_with_field(context): search_term = "@lower citrinelle" update = MockTelegramUpdate.with_inline_query(query=search_term) post_id = 213231 submission = MockSubmission(post_id) inline = InlineFunctionality(MockExportAPI()) inline.api.with_search_results(search_term, [submission]) inline.call(update, context) context.bot.answer_inline_query.assert_called_once() args = context.bot.answer_inline_query.call_args[0] assert context.bot.answer_inline_query.call_args[1]['next_offset'] == 2 assert args[0] == update.inline_query.id assert isinstance(args[1], list) assert len(args[1]) == 1 assert isinstance(args[1][0], InlineQueryResultPhoto) assert args[1][0].id == str(post_id) assert args[1][0].photo_url == submission.thumbnail_url assert args[1][0].thumb_url == FASubmission.make_thumbnail_smaller( submission.thumbnail_url) assert args[1][0].caption == submission.link
def test_simple_search(context): post_id = 234563 search_term = "YCH" update = MockTelegramUpdate.with_inline_query(query=search_term) submission = MockSubmission(post_id) inline = InlineFunctionality(MockExportAPI()) inline.api.with_search_results(search_term, [submission]) inline.call(update, context) context.bot.answer_inline_query.assert_called_once() args = context.bot.answer_inline_query.call_args[0] assert context.bot.answer_inline_query.call_args[1]['next_offset'] == 2 assert args[0] == update.inline_query.id assert isinstance(args[1], list) assert len(args[1]) > 0 for result in args[1]: assert isinstance(result, InlineQueryResultPhoto) assert result.id == str(post_id) assert result.photo_url == submission.thumbnail_url assert result.thumb_url == FASubmission.make_thumbnail_smaller( submission.thumbnail_url) assert result.caption == submission.link