andreacavalli/tdlib-fork
Archived
1
0
Fork 0
Code Pull Requests Releases Activity
This repository has been archived on 2020-05-25. You can view files and clone it, but cannot push or open issues or pull requests.
b2a175463f
tdlib-fork/test/message_entities.cpp
levlam 6417510d51 Return all entities in get_message_entities. 
GitOrigin-RevId: b08c7a9ab0cfa9a78e450d5cbf3567203abd7d67
2020-03-10 03:51:56 +03:00

1521 lines
83 KiB
C++
Raw Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2020
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/telegram/MessageEntity.h"
#include "td/utils/common.h"
#include "td/utils/format.h"
#include "td/utils/logging.h"
#include "td/utils/misc.h"
#include "td/utils/Random.h"
#include "td/utils/Slice.h"
#include "td/utils/tests.h"
#include "td/utils/utf8.h"
#include <algorithm>
REGISTER_TESTS(message_entities);
static void check_mention(const td::string &str, const td::vector<td::string> &expected) {
auto result_slice = td::find_mentions(str);
td::vector<td::string> result;
for (auto &it : result_slice) {
result.push_back(it.str());
}
if (result != expected) {
LOG(FATAL) << td::tag("text", str) << td::tag("got", td::format::as_array(result))
<< td::tag("expected", td::format::as_array(expected));
}
}
TEST(MessageEntities, mention) {
check_mention("@mention", {"@mention"});
check_mention("@mention ", {"@mention"});
check_mention(" @mention", {"@mention"});
check_mention(" @mention ", {"@mention"});
check_mention("@abc @xyz @abc @xyz @xxx@yyy @ttt", {});
check_mention("@abcde @xyzxy @abcde @xyzxy @xxxxx@yyyyy @ttttt",
{"@abcde", "@xyzxy", "@abcde", "@xyzxy", "@xxxxx", "@ttttt"});
check_mention("no@mention", {});
check_mention("@n", {});
check_mention("@abcdefghijklmnopqrstuvwxyz123456", {"@abcdefghijklmnopqrstuvwxyz123456"});
check_mention("@abcdefghijklmnopqrstuvwxyz1234567", {});
check_mention("нет@mention", {});
check_mention("@ya @gif @wiki @vid @bing @pic @bold @imdb @coub @like @vote @giff @cap ya cap @y @yar @bingg @bin",
{"@gif", "@wiki", "@vid", "@bing", "@pic", "@bold", "@imdb", "@coub", "@like", "@vote", "@bingg"});
};
static void check_bot_command(const td::string &str, const td::vector<td::string> &expected) {
auto result_slice = td::find_bot_commands(str);
td::vector<td::string> result;
for (auto &it : result_slice) {
result.push_back(it.str());
}
if (result != expected) {
LOG(FATAL) << td::tag("text", str) << td::tag("got", td::format::as_array(result))
<< td::tag("expected", td::format::as_array(expected));
}
}
TEST(MessageEntities, bot_command) {
// 1..64@3..32
check_bot_command("/abc", {"/abc"});
check_bot_command(" /abc", {"/abc"});
check_bot_command("/abc ", {"/abc"});
check_bot_command(" /abc ", {"/abc"});
check_bot_command("/a@abc", {"/a@abc"});
check_bot_command("/a@b", {});
check_bot_command("/@bfdsa", {});
check_bot_command("/test/", {});
}
static void check_hashtag(const td::string &str, const td::vector<td::string> &expected) {
auto result_slice = td::find_hashtags(str);
td::vector<td::string> result;
for (auto &it : result_slice) {
result.push_back(it.str());
}
if (result != expected) {
LOG(FATAL) << td::tag("text", str) << td::tag("got", td::format::as_array(result))
<< td::tag("expected", td::format::as_array(expected));
}
}
TEST(MessageEntities, hashtag) {
check_hashtag("", {});
check_hashtag("#", {});
check_hashtag("##", {});
check_hashtag("###", {});
check_hashtag("#a", {"#a"});
check_hashtag(" #a", {"#a"});
check_hashtag("#a ", {"#a"});
check_hashtag(" #я ", {"#я"});
check_hashtag(" я#a ", {});
check_hashtag(" #a# ", {});
check_hashtag(" #123 ", {});
check_hashtag(" #123a ", {"#123a"});
check_hashtag(" #a123 ", {"#a123"});
check_hashtag(" #123a# ", {});
check_hashtag(" #" + td::string(300, '1'), {});
check_hashtag(" #" + td::string(256, '1'), {});
check_hashtag(" #" + td::string(256, '1') + "a ", {});
check_hashtag(" #" + td::string(255, '1') + "a", {"#" + td::string(255, '1') + "a"});
check_hashtag(" #" + td::string(255, '1') + "Я", {"#" + td::string(255, '1') + "Я"});
check_hashtag(" #" + td::string(255, '1') + "a" + td::string(255, 'b') + "# ", {});
check_hashtag("#a#b #c #d", {"#c", "#d"});
check_hashtag("#test", {"#test"});
check_hashtag(u8"\U0001F604\U0001F604\U0001F604\U0001F604 \U0001F604\U0001F604\U0001F604#" + td::string(200, '1') +
"ООО" + td::string(200, '2'),
{"#" + td::string(200, '1') + "ООО" + td::string(53, '2')});
check_hashtag(u8"#a\u2122", {"#a"});
}
static void check_cashtag(const td::string &str, const td::vector<td::string> &expected) {
auto result_slice = td::find_cashtags(str);
td::vector<td::string> result;
for (auto &it : result_slice) {
result.push_back(it.str());
}
if (result != expected) {
LOG(FATAL) << td::tag("text", str) << td::tag("got", td::format::as_array(result))
<< td::tag("expected", td::format::as_array(expected));
}
}
TEST(MessageEntities, cashtag) {
check_cashtag("", {});
check_cashtag("$", {});
check_cashtag("$$", {});
check_cashtag("$$$", {});
check_cashtag("$a", {});
check_cashtag(" $a", {});
check_cashtag("$a ", {});
check_cashtag(" $я ", {});
check_cashtag("$ab", {});
check_cashtag("$abc", {});
check_cashtag("$", {});
check_cashtag("$A", {});
check_cashtag("$AB", {});
check_cashtag("$АBC", {});
check_cashtag("$АВС", {});
check_cashtag("$ABC", {"$ABC"});
check_cashtag("$ABCD", {"$ABCD"});
check_cashtag("$ABCDE", {"$ABCDE"});
check_cashtag("$ABCDEF", {"$ABCDEF"});
check_cashtag("$ABCDEFG", {"$ABCDEFG"});
check_cashtag("$ABCDEFGH", {"$ABCDEFGH"});
check_cashtag("$ABCDEFGHJ", {});
check_cashtag("$ABCDEFGH1", {});
check_cashtag(" $XYZ", {"$XYZ"});
check_cashtag("$XYZ ", {"$XYZ"});
check_cashtag(" $XYZ ", {"$XYZ"});
check_cashtag(" $$XYZ ", {});
check_cashtag(" $XYZ$ ", {});
check_cashtag(" $ABC1 ", {});
check_cashtag(" $1ABC ", {});
check_cashtag(" 1$ABC ", {});
check_cashtag(" А$ABC ", {});
check_cashtag("$ABC$DEF $GHI $KLM", {"$GHI", "$KLM"});
check_cashtag("$TEST", {"$TEST"});
check_cashtag(u8"$ABC\u2122", {"$ABC"});
check_cashtag(u8"\u2122$ABC", {"$ABC"});
check_cashtag(u8"\u2122$ABC\u2122", {"$ABC"});
}
static void check_bank_card_number(const td::string &str, const td::vector<td::string> &expected) {
auto result_slice = td::find_bank_card_numbers(str);
td::vector<td::string> result;
for (auto &it : result_slice) {
result.push_back(it.str());
}
if (result != expected) {
LOG(FATAL) << td::tag("text", str) << td::tag("got", td::format::as_array(result))
<< td::tag("expected", td::format::as_array(expected));
}
}
TEST(MessageEntities, bank_card_number) {
check_bank_card_number("", {});
check_bank_card_number("123456789015", {});
check_bank_card_number("1234567890120", {});
check_bank_card_number("1234567890121", {});
check_bank_card_number("1234567890122", {});
check_bank_card_number("1234567890123", {});
check_bank_card_number("1234567890124", {});
check_bank_card_number("1234567890125", {});
check_bank_card_number("1234567890126", {});
check_bank_card_number("1234567890127", {});
check_bank_card_number("1234567890128", {"1234567890128"});
check_bank_card_number("1234567890129", {});
check_bank_card_number("12345678901500", {"12345678901500"});
check_bank_card_number("123456789012800", {"123456789012800"});
check_bank_card_number("1234567890151800", {"1234567890151800"});
check_bank_card_number("12345678901280000", {"12345678901280000"});
check_bank_card_number("123456789015009100", {"123456789015009100"});
check_bank_card_number("1234567890128000000", {"1234567890128000000"});
check_bank_card_number("12345678901500910000", {});
check_bank_card_number(" - - - -1 - -- 2 - - -- 34 - - - 56- - 7890150000 - - - -", {});
check_bank_card_number(" - - - -1 - -- 234 - - 56- - 7890150000 - - - -", {"1 - -- 234 - - 56- - 7890150000"});
check_bank_card_number("4916-3385-0608-2832; 5280 9342 8317 1080 ;345936346788903",
{"4916-3385-0608-2832", "5280 9342 8317 1080", "345936346788903"});
check_bank_card_number("4556728228023269,4916141675244747020,49161416752447470,4556728228023269",
{"4556728228023269", "4916141675244747020", "4556728228023269"});
}
static void check_is_email_address(const td::string &str, bool expected) {
bool result = td::is_email_address(str);
LOG_IF(FATAL, result != expected) << "Expected " << expected << " as result of is_email_address(" << str << ")";
}
TEST(MessageEntities, is_email_address) {
check_is_email_address("telegram.org", false);
check_is_email_address("security@telegram.org", true);
check_is_email_address("security.telegram.org", false);
check_is_email_address("", false);
check_is_email_address("@", false);
check_is_email_address("A@a.a.a.ab", true);
check_is_email_address("A@a.ab", true);
check_is_email_address("Test@aa.aa.aa.aa", true);
check_is_email_address("Test@test.abd", true);
check_is_email_address("a@a.a.a.ab", true);
check_is_email_address("test@test.abd", true);
check_is_email_address("test@test.com", true);
check_is_email_address("test.abd", false);
check_is_email_address("a.ab", false);
check_is_email_address("a.bc@d.ef", true);
td::vector<td::string> bad_userdatas = {"",
"a.a.a.a.a.a.a.a.a.a.a.a",
"+.+.+.+.+.+",
"*.a.a",
"a.*.a",
"a.a.*",
"a.a.",
"a.a.abcdefghijklmnopqrstuvwxyz0123456789",
"a.abcdefghijklmnopqrstuvwxyz0.a",
"abcdefghijklmnopqrstuvwxyz0.a.a"};
td::vector<td::string> good_userdatas = {"a.a.a.a.a.a.a.a.a.a.a",
"a+a+a+a+a+a+a+a+a+a+a",
"+.+.+.+.+._",
"aozAQZ0-5-9_+-aozAQZ0-5-9_.aozAQZ0-5-9_.-._.+-",
"a.a.a",
"a.a.abcdefghijklmnopqrstuvwxyz012345678",
"a.abcdefghijklmnopqrstuvwxyz.a",
"a..a",
"abcdefghijklmnopqrstuvwxyz.a.a",
".a.a"};
td::vector<td::string> bad_domains = {"",
".",
"abc",
"localhost",
"a.a.a.a.a.a.a.ab",
".......",
"a.a.a.a.a.a+ab",
"a+a.a.a.a.a.ab",
"a.a.a.a.a.a.a",
"a.a.a.a.a.a.abcdefg",
"a.a.a.a.a.a.ab0yz",
"a.a.a.a.a.a.ab9yz",
"a.a.a.a.a.a.ab-yz",
"a.a.a.a.a.a.ab_yz",
"a.a.a.a.a.a.ab*yz",
".ab",
".a.ab",
"a..ab",
"a.a.a..a.ab",
".a.a.a.a.ab",
"abcdefghijklmnopqrstuvwxyz01234.ab",
"ab0cd.abd.aA*sd.0.9.0-9.ABOYZ",
"ab*cd.abd.aAasd.0.9.0-9.ABOYZ",
"ab0cd.abd.aAasd.0.9.0*9.ABOYZ",
"*b0cd.ab_d.aA-sd.0.9.0-9.ABOYZ",
"ab0c*.ab_d.aA-sd.0.9.0-9.ABOYZ",
"ab0cd.ab_d.aA-sd.0.9.0-*.ABOYZ",
"ab0cd.ab_d.aA-sd.0.9.*-9.ABOYZ",
"-b0cd.ab_d.aA-sd.0.9.0-9.ABOYZ",
"ab0c-.ab_d.aA-sd.0.9.0-9.ABOYZ",
"ab0cd.ab_d.aA-sd.-.9.0-9.ABOYZ",
"ab0cd.ab_d.aA-sd.0.9.--9.ABOYZ",
"ab0cd.ab_d.aA-sd.0.9.0--.ABOYZ",
"_b0cd.ab_d.aA-sd.0.9.0-9.ABOYZ",
"ab0c_.ab_d.aA-sd.0.9.0-9.ABOYZ",
"ab0cd.ab_d.aA-sd._.9.0-9.ABOYZ",
"ab0cd.ab_d.aA-sd.0.9._-9.ABOYZ",
"ab0cd.ab_d.aA-sd.0.9.0-_.ABOYZ",
"-.ab_d.aA-sd.0.9.0-9.ABOYZ",
"ab0cd.ab_d.-.0.9.0-9.ABOYZ",
"ab0cd.ab_d.aA-sd.0.9.-.ABOYZ",
"_.ab_d.aA-sd.0.9.0-9.ABOYZ",
"ab0cd.ab_d._.0.9.0-9.ABOYZ",
"ab0cd.ab_d.aA-sd.0.9._.ABOYZ"};
td::vector<td::string> good_domains = {"a.a.a.a.a.a.ab",
"a.a.a.a.a.a.abcdef",
"a.a.a.a.a.a.aboyz",
"a.a.a.a.a.a.ABOYZ",
"a.a.a.a.a.a.AbOyZ",
"abcdefghijklmnopqrstuvwxyz0123.ab",
"ab0cd.ab_d.aA-sd.0.9.0-9.ABOYZ",
"A.Z.aA-sd.a.z.0-9.ABOYZ"};
for (auto &userdata : bad_userdatas) {
for (auto &domain : bad_domains) {
check_is_email_address(userdata + '@' + domain, false);
check_is_email_address(userdata + domain, false);
}
for (auto &domain : good_domains) {
check_is_email_address(userdata + '@' + domain, false);
check_is_email_address(userdata + domain, false);
}
}
for (auto &userdata : good_userdatas) {
for (auto &domain : bad_domains) {
check_is_email_address(userdata + '@' + domain, false);
check_is_email_address(userdata + domain, false);
}
for (auto &domain : good_domains) {
check_is_email_address(userdata + '@' + domain, true);
check_is_email_address(userdata + domain, false);
}
}
}
static void check_url(const td::string &str, const td::vector<td::string> &expected_urls,
td::vector<td::string> expected_email_addresses = {}) {
auto result_slice = td::find_urls(str);
td::vector<td::string> result_urls;
td::vector<td::string> result_email_addresses;
for (auto &it : result_slice) {
if (!it.second) {
result_urls.push_back(it.first.str());
} else {
result_email_addresses.push_back(it.first.str());
}
}
if (result_urls != expected_urls) {
LOG(FATAL) << td::tag("text", str) << td::tag("got", td::format::as_array(result_urls))
<< td::tag("expected", td::format::as_array(expected_urls));
}
if (result_email_addresses != expected_email_addresses) {
LOG(FATAL) << td::tag("text", str) << td::tag("got", td::format::as_array(result_email_addresses))
<< td::tag("expected", td::format::as_array(expected_email_addresses));
}
}
TEST(MessageEntities, url) {
check_url("telegram.org", {"telegram.org"});
check_url("(telegram.org)", {"telegram.org"});
check_url("\ntelegram.org)", {"telegram.org"});
check_url(" telegram.org)", {"telegram.org"});
check_url(".telegram.org)", {});
check_url("()telegram.org/?q=()", {"telegram.org/?q=()"});
check_url("\"telegram.org\"", {"telegram.org"});
check_url(" telegram. org. www. com... telegram.org... ...google.com...", {"telegram.org"});
check_url(" telegram.org ", {"telegram.org"});
check_url("Такой сайт: http://www.google.com или такой telegram.org ", {"http://www.google.com", "telegram.org"});
check_url(" telegram.org. ", {"telegram.org"});
check_url("http://google,.com", {});
check_url("http://telegram.org/?asd=123#123.", {"http://telegram.org/?asd=123#123"});
check_url("[http://google.com](test)", {"http://google.com"});
check_url("", {});
check_url(".", {});
check_url("http://@google.com", {});
check_url("http://@goog.com", {}); // TODO: server fix
check_url("http://@@google.com", {"http://@@google.com"});
check_url("http://a@google.com", {"http://a@google.com"});
check_url("http://test@google.com", {"http://test@google.com"});
check_url("google.com:᪉᪉᪉᪉᪉", {"google.com"});
check_url("https://telegram.org", {"https://telegram.org"});
check_url("http://telegram.org", {"http://telegram.org"});
check_url("ftp://telegram.org", {"ftp://telegram.org"});
check_url("ftps://telegram.org", {});
check_url("sftp://telegram.org", {"sftp://telegram.org"});
check_url("hTtPs://telegram.org", {"hTtPs://telegram.org"});
check_url("HTTP://telegram.org", {"HTTP://telegram.org"});
check_url("аHTTP://telegram.org", {"HTTP://telegram.org"});
check_url("sHTTP://telegram.org", {});
check_url("://telegram.org", {});
check_url("google.com:᪀᪀", {"google.com"});
check_url(
"http://"
"abcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkab"
"cdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcd"
"efghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdef"
"ghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefgh"
"ijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghij"
"kabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijka"
"bcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabc"
"defghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijkabcdefghijk.com",
{});
check_url("http://  .com", {});
check_url("URL:     .com", {});
check_url("URL: .com", {});
check_url(".com", {});
check_url("http://  .", {});
check_url("http://.", {});
check_url("http://.com", {});
check_url("http:// .", {});
check_url(",ahttp://google.com", {"http://google.com"});
check_url(".ahttp://google.com", {});
check_url("http://1.0", {});
check_url("http://a.0", {});
check_url("http://a.a", {});
check_url("google.com:1#ab c", {"google.com:1#ab"});
check_url("google.com:1#", {"google.com:1"});
check_url("google.com:1#1", {"google.com:1#1"});
check_url("google.com:00000001/abs", {"google.com:00000001/abs"});
check_url("google.com:000000065535/abs", {"google.com:000000065535/abs"});
check_url("google.com:000000065536/abs", {"google.com"});
check_url("google.com:000000080/abs", {"google.com:000000080/abs"});
check_url("google.com:0000000/abs", {"google.com"});
check_url("google.com:0/abs", {"google.com"});
check_url("google.com:/abs", {"google.com"});
check_url("google.com:65535", {"google.com:65535"});
check_url("google.com:65536", {"google.com"});
check_url("google.com:99999", {"google.com"});
check_url("google.com:100000", {"google.com"});
check_url("127.001", {});
check_url("127.0.0.1", {"127.0.0.1"});
check_url("127.0.0.01", {});
check_url("127.0.0.256", {});
check_url("127.0.0.300", {});
check_url("127.0.0.1000", {});
check_url("127.0.0.260", {});
check_url("1.0", {});
check_url("www.🤙.tk", {"www.🤙.tk"});
check_url("a.ab", {});
check_url("test.abd", {});
check_url("ТеСт.Москва", {});
check_url("ТеСт.МоСкВΑ", {});
check_url("ТеСт.МоСкВа", {"ТеСт.МоСкВа"});
check_url("ТеСт.МоСкВач", {});
check_url("http://ÀТеСт.МоСкВач", {"http://ÀТеСт.МоСкВач"});
check_url("ÀÁ.com. ÀÁ.com.", {"ÀÁ.com", "ÀÁ.com"});
check_url("ÀÁ.com,ÀÁ.com.", {"ÀÁ.com", "ÀÁ.com"});
check_url("teiegram.org", {});
check_url("http://test.google.com/?q=abc()}[]def", {"http://test.google.com/?q=abc()"});
check_url("http://test.google.com/?q=abc([{)]}def", {"http://test.google.com/?q=abc([{)]}def"});
check_url("http://test.google.com/?q=abc(){}]def", {"http://test.google.com/?q=abc(){}"});
check_url("http://test.google.com/?q=abc){}[]def", {"http://test.google.com/?q=abc"});
check_url("http://test.google.com/?q=abc(){}[]def", {"http://test.google.com/?q=abc(){}[]def"});
check_url("http://test-.google.com", {});
check_url("http://test_.google.com", {"http://test_.google.com"});
check_url("http://google_.com", {});
check_url("http://google._com_", {});
check_url("http://[2001:4860:0:2001::68]/", {}); // TODO
check_url("test.abd", {});
check_url("/.b/..a @.....@/. a.ba", {"a.ba"});
check_url("bbbbbbbbbbbbbb.@.@", {});
check_url("http://google.com/", {"http://google.com/"});
check_url("http://google.com?", {"http://google.com"});
check_url("http://google.com#", {"http://google.com"});
check_url("http://google.com##", {"http://google.com##"});
check_url("http://google.com/?", {"http://google.com/"});
check_url("https://www.google.com/ab,", {"https://www.google.com/ab"});
check_url("@.", {});
check_url(
"a.b.google.com dfsknnfs gsdfgsg http://códuia.de/ dffdg,\" 12)(cpia.de/())(\" http://гришка.рф/ sdufhdf "
"http://xn--80afpi2a3c.xn--p1ai/ I have a good time.Thanks, guys!\n\n(hdfughidufhgdis)go#ogle.com гришка.рф "
"hsighsdf gi почта.рф\n\n✪df.ws/123 "
"xn--80afpi2a3c.xn--p1ai\n\nhttp://foo.com/blah_blah\nhttp://foo.com/blah_blah/\n(Something like "
"http://foo.com/blah_blah)\nhttp://foo.com/blah_blah_(wikipedi8989a_Вася)\n(Something like "
"http://foo.com/blah_blah_(Стакан_007))\nhttp://foo.com/blah_blah.\nhttp://foo.com/blah_blah/.\n<http://foo.com/"
"blah_blah>\n<http://fo@@@@@@@@@^%#*@^&@$#*@#%^*&!^#o.com/blah_blah/>\nhttp://foo.com/blah_blah,\nhttp://"
"www.example.com/wpstyle/?p=364.\nhttp://✪df.ws/123\nrdar://1234\nrdar:/1234\nhttp://"
"userid:password@example.com:8080\nhttp://userid@example.com\nhttp://userid@example.com:8080\nhttp://"
"userid:password@example.com\nhttp://example.com:8080 "
"x-yojimbo-item://6303E4C1-xxxx-45A6-AB9D-3A908F59AE0E\nmessage://"
"%3c330e7f8409726r6a4ba78dkf1fd71420c1bf6ff@mail.gmail.com%3e\nhttp://➡️.ws/䨹\nwww.➡️.ws/"
"䨹\n<tag>http://example.com</tag>\nJust a www.example.com "
"link.\n\n➡️.ws/"
"䨹\n\nabcdefghijklmnopqrstuvwxyz0123456789qwe_sdfsdf.aweawe-sdfs.com\nwww.🤙.tk:1\ngoogle.com:"
"᪉᪉᪉᪉\ngoogle."
"com:᪀᪀\nhttp://  .com\nURL:     .com\nURL: "
".com\n\ngoogle.com?qwe\ngoogle.com#qwe\ngoogle.com/?\ngoogle.com/#\ngoogle.com?\ngoogle.com#\n",
{"a.b.google.com",
"http://códuia.de/",
"cpia.de/()",
"http://гришка.рф/",
"http://xn--80afpi2a3c.xn--p1ai/",
"гришка.рф",
"почта.рф",
"✪df.ws/123",
"xn--80afpi2a3c.xn--p1ai",
"http://foo.com/blah_blah",
"http://foo.com/blah_blah/",
"http://foo.com/blah_blah",
"http://foo.com/blah_blah_(wikipedi8989a_Вася)",
"http://foo.com/blah_blah_(Стакан_007)",
"http://foo.com/blah_blah",
"http://foo.com/blah_blah/",
"http://foo.com/blah_blah",
"http://foo.com/blah_blah",
"http://www.example.com/wpstyle/?p=364",
"http://✪df.ws/123",
"http://userid:password@example.com:8080",
"http://userid@example.com",
"http://userid@example.com:8080",
"http://userid:password@example.com",
"http://example.com:8080",
"http://➡️.ws/䨹",
"www.➡️.ws/䨹",
"http://example.com",
"www.example.com",
"➡️.ws/䨹",
"abcdefghijklmnopqrstuvwxyz0123456789qwe_sdfsdf.aweawe-sdfs.com",
"www.🤙.tk:1",
"google.com",
"google.com",
"google.com?qwe",
"google.com#qwe",
"google.com/",
"google.com/#",
"google.com",
"google.com"});
check_url("https://t.…", {});
check_url("('http://telegram.org/a-b/?br=ie&lang=en',)", {"http://telegram.org/a-b/?br=ie&lang=en"});
check_url("https://ai.telegram.org/bot%20bot/test-...", {"https://ai.telegram.org/bot%20bot/test-"});
check_url("a.bc@c.com", {}, {"a.bc@c.com"});
check_url("https://a.bc@c.com", {"https://a.bc@c.com"});
check_url("https://a.de[bc@c.com", {"https://a.de"}, {"bc@c.com"});
check_url("https://a.de/bc@c.com", {"https://a.de/bc@c.com"});
check_url("https://a.de]bc@c.com", {"https://a.de"}, {"bc@c.com"});
check_url("https://a.de{bc@c.com", {"https://a.de"}, {"bc@c.com"});
check_url("https://a.de}bc@c.com", {"https://a.de"}, {"bc@c.com"});
check_url("https://a.de(bc@c.com", {"https://a.de"}, {"bc@c.com"});
check_url("https://a.de)bc@c.com", {"https://a.de"}, {"bc@c.com"});
check_url("https://a.de\\bc@c.com", {"https://a.de\\bc@c.com"});
check_url("https://a.de'bc@c.com", {"https://a.de"}, {"bc@c.com"});
check_url("https://a.de`bc@c.com", {"https://a.de"}, {"bc@c.com"});
check_url("https://a.bc:de.fg@c.com", {"https://a.bc:de.fg@c.com"});
check_url("https://a:h.bc:de.fg@c.com", {"https://a:h.bc:de.fg@c.com"});
check_url("https://abc@c.com", {"https://abc@c.com"});
check_url("https://de[bc@c.com", {}, {"bc@c.com"});
check_url("https://de/bc@c.com", {});
check_url("https://de]bc@c.com", {}, {"bc@c.com"});
check_url("https://de{bc@c.com", {}, {"bc@c.com"});
check_url("https://de}bc@c.com", {}, {"bc@c.com"});
check_url("https://de(bc@c.com", {}, {"bc@c.com"});
check_url("https://de)bc@c.com", {}, {"bc@c.com"});
check_url("https://de\\bc@c.com", {"https://de\\bc@c.com"});
check_url("https://de'bc@c.com", {}, {"bc@c.com"});
check_url("https://de`bc@c.com", {}, {"bc@c.com"});
check_url("https://bc:defg@c.com", {"https://bc:defg@c.com"});
check_url("https://a:hbc:defg@c.com", {"https://a:hbc:defg@c.com"});
check_url("https://a.bc@test.com:cd.com", {"https://a.bc@test.com", "cd.com"});
check_url("telegram.Org", {});
check_url("telegram.ORG", {"telegram.ORG"});
check_url("a.b.c.com.a.b.c", {});
check_url("File '/usr/views.py'", {}); // TODO server fix
check_url("@views.py'", {}); // TODO server fix
check_url("#views.py'", {}); // TODO server fix
check_url("/views.py'", {}); // TODO server fix
check_url(".views.py", {});
check_url("'views.py'", {"views.py"});
check_url("bug.http://test.com/test/+#", {}); // TODO {"http://test.com/test/+#"}
check_url("//AB.C.D.E.F.GH//", {});
check_url("<http://www.ics.uci.edu/pub/ietf/uri/historical.html#WARNING>",
{"http://www.ics.uci.edu/pub/ietf/uri/historical.html#WARNING"});
check_url("Look :test@example.com", {":test@example.com"}, {}); // TODO {}, {"test@example.com"}
check_url("Look mailto:test@example.com", {}, {"test@example.com"});
check_url("http://test.com#a", {"http://test.com#a"});
check_url("http://test.com#", {"http://test.com"});
check_url("http://test.com?#", {"http://test.com?#"});
check_url("http://test.com/?#", {"http://test.com/?#"});
check_url("https://t.me/abcdef…", {"https://t.me/abcdef"});
check_url("https://t.me…", {"https://t.me"});
check_url("https://t.m…", {});
check_url("https://t.…", {});
check_url("https://t…", {});
check_url("👉http://ab.com/cdefgh-1IJ", {"http://ab.com/cdefgh-1IJ"});
check_url("...👉http://ab.com/cdefgh-1IJ", {}); // TODO
}
static void check_fix_formatted_text(td::string str, td::vector<td::MessageEntity> entities,
const td::string &expected_str,
const td::vector<td::MessageEntity> &expected_entities, bool allow_empty = true,
bool skip_new_entities = false, bool skip_bot_commands = false,
bool for_draft = true) {
ASSERT_TRUE(
td::fix_formatted_text(str, entities, allow_empty, skip_new_entities, skip_bot_commands, for_draft).is_ok());
ASSERT_STREQ(expected_str, str);
ASSERT_EQ(expected_entities, entities);
}
static void check_fix_formatted_text(td::string str, td::vector<td::MessageEntity> entities, bool allow_empty,
bool skip_new_entities, bool skip_bot_commands, bool for_draft) {
ASSERT_TRUE(
fix_formatted_text(str, entities, allow_empty, skip_new_entities, skip_bot_commands, for_draft).is_error());
}
TEST(MessageEntities, fix_formatted_text) {
td::string str;
td::string fixed_str;
for (auto i = 0; i <= 32; i++) {
str += static_cast<char>(i);
if (i != 13) {
if (i != 10) {
fixed_str += ' ';
} else {
fixed_str += str.back();
}
}
}
check_fix_formatted_text(str, {}, "", {}, true, true, true, true);
check_fix_formatted_text(str, {}, "", {}, true, true, false, true);
check_fix_formatted_text(str, {}, "", {}, true, false, true, true);
check_fix_formatted_text(str, {}, "", {}, true, false, false, true);
check_fix_formatted_text(str, {}, "", {}, true, false, false, false);
check_fix_formatted_text(str, {}, false, false, false, false);
check_fix_formatted_text(str, {}, false, false, false, true);
str += "a \r\n ";
fixed_str += "a \n ";
for (td::int32 i = 33; i <= 35; i++) {
td::vector<td::MessageEntity> entities;
entities.emplace_back(td::MessageEntity::Type::Bold, 0, i);
td::vector<td::MessageEntity> fixed_entities;
if (i != 33) {
fixed_entities = entities;
fixed_entities.back().length = i - 1;
}
check_fix_formatted_text(str, entities, fixed_str, fixed_entities, true, false, false, true);
td::string expected_str;
if (i != 33) {
fixed_entities = entities;
fixed_entities.back().length = 33;
expected_str = fixed_str.substr(0, 33);
} else {
fixed_entities.clear();
expected_str = "a";
}
check_fix_formatted_text(str, entities, expected_str, fixed_entities, false, false, false, false);
}
str = "👉 👉 ";
for (int i = 0; i < 10; i++) {
td::vector<td::MessageEntity> entities;
entities.emplace_back(td::MessageEntity::Type::Bold, i, 1);
if (i != 2 && i != 5 && i != 6) {
check_fix_formatted_text(str, entities, true, true, true, true);
check_fix_formatted_text(str, entities, false, false, false, false);
} else {
check_fix_formatted_text(str, entities, str, {}, true, true, true, true);
check_fix_formatted_text(str, entities, str.substr(0, str.size() - 2), {}, false, false, false, false);
}
}
str = " /test @abaca #ORD $ABC telegram.org ";
for (auto for_draft : {false, true}) {
td::int32 shift = for_draft ? 2 : 0;
td::string expected_str = for_draft ? str : str.substr(2, str.size() - 3);
for (auto skip_new_entities : {false, true}) {
for (auto skip_bot_commands : {false, true}) {
td::vector<td::MessageEntity> entities;
if (!skip_new_entities) {
if (!skip_bot_commands) {
entities.emplace_back(td::MessageEntity::Type::BotCommand, shift, 5);
}
entities.emplace_back(td::MessageEntity::Type::Mention, shift + 6, 6);
entities.emplace_back(td::MessageEntity::Type::Hashtag, shift + 13, 4);
entities.emplace_back(td::MessageEntity::Type::Cashtag, shift + 18, 4);
entities.emplace_back(td::MessageEntity::Type::Url, shift + 24, 12);
}
check_fix_formatted_text(str, {}, expected_str, entities, true, skip_new_entities, skip_bot_commands,
for_draft);
check_fix_formatted_text(str, {}, expected_str, entities, false, skip_new_entities, skip_bot_commands,
for_draft);
}
}
}
str = "aba \r\n caba ";
for (td::int32 length = 1; length <= 3; length++) {
for (td::int32 offset = 0; static_cast<size_t>(offset + length) <= str.size(); offset++) {
for (auto type : {td::MessageEntity::Type::Bold, td::MessageEntity::Type::Url, td::MessageEntity::Type::TextUrl,
td::MessageEntity::Type::MentionName}) {
for (auto for_draft : {false, true}) {
fixed_str = for_draft ? "aba \n caba " : "aba \n caba";
auto fixed_length = offset <= 4 && offset + length >= 5 ? length - 1 : length;
auto fixed_offset = offset >= 5 ? offset - 1 : offset;
if (static_cast<size_t>(fixed_offset) >= fixed_str.size()) {
fixed_length = 0;
}
while (static_cast<size_t>(fixed_offset + fixed_length) > fixed_str.size()) {
fixed_length--;
}
td::vector<td::MessageEntity> entities;
entities.emplace_back(type, offset, length);
td::vector<td::MessageEntity> fixed_entities;
if (fixed_length > 0) {
for (auto i = 0; i < length; i++) {
if (str[offset + i] != '\r' && str[offset + i] != '\n' &&
(str[offset + i] != ' ' || type == td::MessageEntity::Type::TextUrl ||
type == td::MessageEntity::Type::MentionName)) {
fixed_entities.emplace_back(type, fixed_offset, fixed_length);
break;
}
}
}
check_fix_formatted_text(str, entities, fixed_str, fixed_entities, true, false, false, for_draft);
}
}
}
}
str = "aba caba";
for (td::int32 length = -10; length <= 10; length++) {
for (td::int32 offset = -10; offset <= 10; offset++) {
td::vector<td::MessageEntity> entities;
entities.emplace_back(td::MessageEntity::Type::Bold, offset, length);
td::vector<td::MessageEntity> fixed_entities;
if (length < 0 || offset < 0 || (length > 0 && static_cast<size_t>(length + offset) > str.size())) {
check_fix_formatted_text(str, entities, true, false, false, false);
check_fix_formatted_text(str, entities, false, false, false, true);
continue;
}
if (length > 0 && (length >= 2 || offset != 3)) {
fixed_entities.emplace_back(td::MessageEntity::Type::Bold, offset, length);
}
check_fix_formatted_text(str, entities, str, fixed_entities, true, false, false, false);
check_fix_formatted_text(str, entities, str, fixed_entities, false, false, false, true);
}
}
str = "aba caba";
for (td::int32 length = 1; length <= 7; length++) {
for (td::int32 offset = 0; offset <= 8 - length; offset++) {
for (td::int32 length2 = 1; length2 <= 7; length2++) {
for (td::int32 offset2 = 0; offset2 <= 8 - length2; offset2++) {
if (offset != offset2) {
td::vector<td::MessageEntity> entities;
entities.emplace_back(td::MessageEntity::Type::TextUrl, offset, length);
entities.emplace_back(td::MessageEntity::Type::TextUrl, offset2, length2);
td::vector<td::MessageEntity> fixed_entities = entities;
std::sort(fixed_entities.begin(), fixed_entities.end());
if (fixed_entities[0].offset + fixed_entities[0].length > fixed_entities[1].offset) {
fixed_entities.pop_back();
}
check_fix_formatted_text(str, entities, str, fixed_entities, false, false, false, false);
}
}
}
}
}
for (auto text : {" \n ➡️ ➡️ ➡️ ➡️ \n ", "\n\n\nab cd ef gh "}) {
str = text;
td::vector<td::MessageEntity> entities;
td::vector<td::MessageEntity> fixed_entities;
auto length = td::narrow_cast<int>(td::utf8_utf16_length(str));
for (int i = 0; i < 10; i++) {
if ((i + 1) * 3 + 2 <= length) {
entities.emplace_back(td::MessageEntity::Type::Bold, (i + 1) * 3, 2);
}
if ((i + 2) * 3 <= length) {
entities.emplace_back(td::MessageEntity::Type::Italic, (i + 1) * 3 + 2, 1);
}
if (i < 4) {
fixed_entities.emplace_back(td::MessageEntity::Type::Bold, i * 3, 2);
}
}
check_fix_formatted_text(str, entities, td::utf8_utf16_substr(str, 3, 11), fixed_entities, false, false, false,
false);
}
for (td::string text : {"\t", "\r", "\n", "\t ", "\r ", "\n "}) {
for (auto type : {td::MessageEntity::Type::Bold, td::MessageEntity::Type::TextUrl}) {
check_fix_formatted_text(text, {{type, 0, 1, "http://telegram.org/"}}, "", {}, true, false, false, true);
}
}
check_fix_formatted_text("abc", {{td::MessageEntity::Type::Italic, 1, 1}, {td::MessageEntity::Type::Italic, 0, 1}},
"abc", {{td::MessageEntity::Type::Italic, 0, 2}});
check_fix_formatted_text("abc", {{td::MessageEntity::Type::Italic, 1, 1}, {td::MessageEntity::Type::Italic, 1, 1}},
"abc", {{td::MessageEntity::Type::Italic, 1, 1}});
check_fix_formatted_text("abc", {{td::MessageEntity::Type::Italic, 0, 2}, {td::MessageEntity::Type::Italic, 1, 2}},
"abc", {{td::MessageEntity::Type::Italic, 0, 3}});
check_fix_formatted_text("abc", {{td::MessageEntity::Type::Italic, 0, 2}, {td::MessageEntity::Type::Italic, 2, 1}},
"abc", {{td::MessageEntity::Type::Italic, 0, 3}});
check_fix_formatted_text("abc", {{td::MessageEntity::Type::Italic, 0, 1}, {td::MessageEntity::Type::Italic, 2, 1}},
"abc", {{td::MessageEntity::Type::Italic, 0, 1}, {td::MessageEntity::Type::Italic, 2, 1}});
check_fix_formatted_text("abc", {{td::MessageEntity::Type::Italic, 0, 2}, {td::MessageEntity::Type::Bold, 1, 2}},
"abc",
{{td::MessageEntity::Type::Italic, 0, 1},
{td::MessageEntity::Type::Bold, 1, 2},
{td::MessageEntity::Type::Italic, 1, 1}});
check_fix_formatted_text("abc", {{td::MessageEntity::Type::Italic, 0, 2}, {td::MessageEntity::Type::Bold, 2, 1}},
"abc", {{td::MessageEntity::Type::Italic, 0, 2}, {td::MessageEntity::Type::Bold, 2, 1}});
check_fix_formatted_text("abc", {{td::MessageEntity::Type::Italic, 0, 1}, {td::MessageEntity::Type::Bold, 2, 1}},
"abc", {{td::MessageEntity::Type::Italic, 0, 1}, {td::MessageEntity::Type::Bold, 2, 1}});
// _a*b*_
check_fix_formatted_text(
"ab", {{td::MessageEntity::Type::Underline, 0, 2}, {td::MessageEntity::Type::Strikethrough, 1, 1}}, "ab",
{{td::MessageEntity::Type::Underline, 0, 2}, {td::MessageEntity::Type::Strikethrough, 1, 1}});
check_fix_formatted_text("ab",
{{td::MessageEntity::Type::Underline, 0, 1},
{td::MessageEntity::Type::Underline, 1, 1},
{td::MessageEntity::Type::Strikethrough, 1, 1}},
"ab",
{{td::MessageEntity::Type::Underline, 0, 1},
{td::MessageEntity::Type::Underline, 1, 1},
{td::MessageEntity::Type::Strikethrough, 1, 1}});
check_fix_formatted_text(
"ab", {{td::MessageEntity::Type::Strikethrough, 0, 2}, {td::MessageEntity::Type::Underline, 1, 1}}, "ab",
{{td::MessageEntity::Type::Strikethrough, 0, 2}, {td::MessageEntity::Type::Underline, 1, 1}});
check_fix_formatted_text("ab",
{{td::MessageEntity::Type::Strikethrough, 0, 1},
{td::MessageEntity::Type::Strikethrough, 1, 1},
{td::MessageEntity::Type::Underline, 1, 1}},
"ab",
{{td::MessageEntity::Type::Strikethrough, 0, 1},
{td::MessageEntity::Type::Underline, 1, 1},
{td::MessageEntity::Type::Strikethrough, 1, 1}});
// _*a*b_
check_fix_formatted_text(
"ab", {{td::MessageEntity::Type::Underline, 0, 2}, {td::MessageEntity::Type::Strikethrough, 0, 1}}, "ab",
{{td::MessageEntity::Type::Underline, 0, 2}, {td::MessageEntity::Type::Strikethrough, 0, 1}});
check_fix_formatted_text(
"ab",
{{td::MessageEntity::Type::Underline, 0, 1},
{td::MessageEntity::Type::Underline, 1, 1},
{td::MessageEntity::Type::Strikethrough, 0, 1}},
"ab", {{td::MessageEntity::Type::Underline, 0, 2}, {td::MessageEntity::Type::Strikethrough, 0, 1}});
// _*a*_\r_*b*_
check_fix_formatted_text("a\rb",
{{td::MessageEntity::Type::Bold, 0, 1},
{td::MessageEntity::Type::Strikethrough, 0, 1},
{td::MessageEntity::Type::Bold, 2, 1},
{td::MessageEntity::Type::Strikethrough, 2, 1}},
"ab",
{{td::MessageEntity::Type::Bold, 0, 2}, {td::MessageEntity::Type::Strikethrough, 0, 2}});
check_fix_formatted_text("a\nb",
{{td::MessageEntity::Type::Bold, 0, 1},
{td::MessageEntity::Type::Strikethrough, 0, 1},
{td::MessageEntity::Type::Bold, 2, 1},
{td::MessageEntity::Type::Strikethrough, 2, 1}},
"a\nb",
{{td::MessageEntity::Type::Bold, 0, 1},
{td::MessageEntity::Type::Strikethrough, 0, 1},
{td::MessageEntity::Type::Bold, 2, 1},
{td::MessageEntity::Type::Strikethrough, 2, 1}});
// _`a`_
check_fix_formatted_text("a", {{td::MessageEntity::Type::Pre, 0, 1}, {td::MessageEntity::Type::Strikethrough, 0, 1}},
"a", {{td::MessageEntity::Type::Pre, 0, 1}});
check_fix_formatted_text("a", {{td::MessageEntity::Type::Strikethrough, 0, 1}, {td::MessageEntity::Type::Pre, 0, 1}},
"a", {{td::MessageEntity::Type::Pre, 0, 1}});
check_fix_formatted_text("abc",
{{td::MessageEntity::Type::Pre, 0, 3}, {td::MessageEntity::Type::Strikethrough, 1, 1}},
"abc", {{td::MessageEntity::Type::Pre, 0, 3}});
check_fix_formatted_text(
"abc", {{td::MessageEntity::Type::Pre, 1, 1}, {td::MessageEntity::Type::Strikethrough, 0, 3}}, "abc",
{{td::MessageEntity::Type::Strikethrough, 0, 1},
{td::MessageEntity::Type::Pre, 1, 1},
{td::MessageEntity::Type::Strikethrough, 2, 1}});
check_fix_formatted_text(
"abc", {{td::MessageEntity::Type::Pre, 1, 1}, {td::MessageEntity::Type::Strikethrough, 1, 2}}, "abc",
{{td::MessageEntity::Type::Pre, 1, 1}, {td::MessageEntity::Type::Strikethrough, 2, 1}});
check_fix_formatted_text(
"abc", {{td::MessageEntity::Type::Pre, 1, 1}, {td::MessageEntity::Type::Strikethrough, 0, 2}}, "abc",
{{td::MessageEntity::Type::Strikethrough, 0, 1}, {td::MessageEntity::Type::Pre, 1, 1}});
check_fix_formatted_text("abc", {{td::MessageEntity::Type::Pre, 0, 3}, {td::MessageEntity::Type::BlockQuote, 1, 1}},
"abc", {{td::MessageEntity::Type::BlockQuote, 1, 1}});
check_fix_formatted_text("abc", {{td::MessageEntity::Type::BlockQuote, 0, 3}, {td::MessageEntity::Type::Pre, 1, 1}},
"abc", {{td::MessageEntity::Type::BlockQuote, 0, 3}, {td::MessageEntity::Type::Pre, 1, 1}});
check_fix_formatted_text("example.com", {}, "example.com", {{td::MessageEntity::Type::Url, 0, 11}});
check_fix_formatted_text("example.com", {{td::MessageEntity::Type::Pre, 0, 3}}, "example.com",
{{td::MessageEntity::Type::Pre, 0, 3}});
check_fix_formatted_text("example.com", {{td::MessageEntity::Type::BlockQuote, 0, 3}}, "example.com",
{{td::MessageEntity::Type::BlockQuote, 0, 3}});
check_fix_formatted_text("example.com", {{td::MessageEntity::Type::BlockQuote, 0, 11}}, "example.com",
{{td::MessageEntity::Type::BlockQuote, 0, 11}, {td::MessageEntity::Type::Url, 0, 11}});
check_fix_formatted_text("example.com", {{td::MessageEntity::Type::Italic, 0, 11}}, "example.com",
{{td::MessageEntity::Type::Url, 0, 11}, {td::MessageEntity::Type::Italic, 0, 11}});
check_fix_formatted_text("example.com", {{td::MessageEntity::Type::Italic, 0, 3}}, "example.com",
{{td::MessageEntity::Type::Url, 0, 11}, {td::MessageEntity::Type::Italic, 0, 3}});
check_fix_formatted_text("example.com a", {{td::MessageEntity::Type::Italic, 0, 13}}, "example.com a",
{{td::MessageEntity::Type::Url, 0, 11},
{td::MessageEntity::Type::Italic, 0, 11},
{td::MessageEntity::Type::Italic, 11, 2}});
check_fix_formatted_text("a example.com", {{td::MessageEntity::Type::Italic, 0, 13}}, "a example.com",
{{td::MessageEntity::Type::Italic, 0, 2},
{td::MessageEntity::Type::Url, 2, 11},
{td::MessageEntity::Type::Italic, 2, 11}});
for (size_t test_n = 0; test_n < 100000; test_n++) {
bool is_url = td::Random::fast(0, 1) == 1;
td::int32 url_offset = 0;
td::int32 url_end = 0;
if (is_url) {
str = td::string(td::Random::fast(1, 5), 'a') + ":example.com:" + td::string(td::Random::fast(1, 5), 'a');
url_offset = static_cast<td::int32>(str.find('e'));
url_end = url_offset + 11;
} else {
str = td::string(td::Random::fast(1, 20), 'a');
}
auto n = td::Random::fast(1, 20);
td::vector<td::MessageEntity> entities;
for (int j = 0; j < n; j++) {
td::int32 type = td::Random::fast(4, 16);
td::int32 offset = td::Random::fast(0, static_cast<int>(str.size()) - 1);
auto max_length = static_cast<int>(str.size() - offset);
if ((test_n & 1) != 0 && max_length > 4) {
max_length = 4;
}
td::int32 length = td::Random::fast(0, max_length);
entities.emplace_back(static_cast<td::MessageEntity::Type>(type), offset, length);
}
auto get_type_mask = [](std::size_t length, const td::vector<td::MessageEntity> &entities) {
td::vector<td::int32> result(length);
for (auto &entity : entities) {
for (auto pos = 0; pos < entity.length; pos++) {
result[entity.offset + pos] |= (1 << static_cast<td::int32>(entity.type));
}
}
return result;
};
auto old_type_mask = get_type_mask(str.size(), entities);
ASSERT_TRUE(td::fix_formatted_text(str, entities, false, false, true, false).is_ok());
auto new_type_mask = get_type_mask(str.size(), entities);
auto splittable_mask = (1 << 5) | (1 << 6) | (1 << 14) | (1 << 15);
auto pre_mask = (1 << 7) | (1 << 8) | (1 << 9);
for (std::size_t pos = 0; pos < str.size(); pos++) {
if ((new_type_mask[pos] & pre_mask) != 0) {
ASSERT_EQ(0, new_type_mask[pos] & splittable_mask);
} else {
ASSERT_EQ(old_type_mask[pos] & splittable_mask, new_type_mask[pos] & splittable_mask);
}
}
bool keep_url = is_url;
td::MessageEntity url_entity(td::MessageEntity::Type::Url, url_offset, url_end - url_offset);
for (auto &entity : entities) {
if (entity == url_entity) {
continue;
}
td::int32 offset = entity.offset;
td::int32 end = offset + entity.length;
if (keep_url && ((1 << static_cast<td::int32>(entity.type)) & splittable_mask) == 0 &&
!(end <= url_offset || url_end <= offset)) {
keep_url = (entity.type == td::MessageEntity::Type::BlockQuote && offset <= url_offset && url_end <= end);
}
}
ASSERT_EQ(keep_url, std::count(entities.begin(), entities.end(), url_entity) == 1);
for (size_t i = 0; i < entities.size(); i++) {
auto type_mask = 1 << static_cast<td::int32>(entities[i].type);
for (size_t j = i + 1; j < entities.size(); j++) {
// sorted
ASSERT_TRUE(entities[j].offset > entities[i].offset ||
(entities[j].offset == entities[i].offset && entities[j].length <= entities[i].length));
// not intersecting
ASSERT_TRUE(entities[j].offset >= entities[i].offset + entities[i].length ||
entities[j].offset + entities[j].length <= entities[i].offset + entities[i].length);
if (entities[j].offset < entities[i].offset + entities[i].length) { // if nested
// types are different
ASSERT_TRUE(entities[j].type != entities[i].type);
// pre can't contain other entities
ASSERT_TRUE((type_mask & pre_mask) == 0);
if ((type_mask & splittable_mask) == 0 && entities[i].type != td::MessageEntity::Type::BlockQuote) {
// continuous entities can contain only splittable entities
ASSERT_TRUE(((1 << static_cast<td::int32>(entities[j].type)) & splittable_mask) != 0);
}
}
}
}
}
}
static void check_parse_html(td::string text, const td::string &result, const td::vector<td::MessageEntity> &entities) {
auto r_entities = td::parse_html(text);
ASSERT_TRUE(r_entities.is_ok());
ASSERT_EQ(entities, r_entities.ok());
ASSERT_STREQ(result, text);
}
static void check_parse_html(td::string text, td::Slice error_message) {
auto r_entities = td::parse_html(text);
ASSERT_TRUE(r_entities.is_error());
ASSERT_EQ(400, r_entities.error().code());
ASSERT_STREQ(error_message, r_entities.error().message());
}
TEST(MessageEntities, parse_html) {
td::string invalid_surrogate_pair_error_message =
"Text contains invalid Unicode characters after decoding HTML entities, check for unmatched surrogate code units";
check_parse_html("&#57311;", invalid_surrogate_pair_error_message);
check_parse_html("&#xDFDF;", invalid_surrogate_pair_error_message);
check_parse_html("&#xDFDF", invalid_surrogate_pair_error_message);
check_parse_html("🏟 🏟&lt;<abacaba", "Unclosed start tag at byte offset 13");
check_parse_html("🏟 🏟&lt;<abac aba>", "Unsupported start tag \"abac\" at byte offset 13");
check_parse_html("🏟 🏟&lt;<abac>", "Unsupported start tag \"abac\" at byte offset 13");
check_parse_html("🏟 🏟&lt;<i =aba>", "Empty attribute name in the tag \"i\" at byte offset 13");
check_parse_html("🏟 🏟&lt;<i aba>",
"Expected equal sign in declaration of an attribute of the tag \"i\" at byte offset 13");
check_parse_html("🏟 🏟&lt;<i aba = ", "Unclosed start tag \"i\" at byte offset 13");
check_parse_html("🏟 🏟&lt;<i aba = 190azAz-.,", "Unexpected end of name token at byte offset 27");
check_parse_html("🏟 🏟&lt;<i aba = \"&lt;&gt;&quot;>", "Unclosed start tag at byte offset 13");
check_parse_html("🏟 🏟&lt;<i aba = \'&lt;&gt;&quot;>", "Unclosed start tag at byte offset 13");
check_parse_html("🏟 🏟&lt;</", "Unexpected end tag at byte offset 13");
check_parse_html("🏟 🏟&lt;<b></b></", "Unexpected end tag at byte offset 20");
check_parse_html("🏟 🏟&lt;<i>a</i ", "Unclosed end tag at byte offset 17");
check_parse_html("🏟 🏟&lt;<i>a</em >",
"Unmatched end tag at byte offset 17, expected \"</i>\", found \"</em>\"");
check_parse_html("", "", {});
check_parse_html("➡️ ➡️", "➡️ ➡️", {});
check_parse_html("&lt;&gt;&amp;&quot;&laquo;&raquo;&#12345678;", "<>&\"&laquo;&raquo;&#12345678;", {});
check_parse_html("➡️ ➡️<i>➡️ ➡️</i>", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Italic, 5, 5}});
check_parse_html("➡️ ➡️<em>➡️ ➡️</em>", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Italic, 5, 5}});
check_parse_html("➡️ ➡️<b>➡️ ➡️</b>", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Bold, 5, 5}});
check_parse_html("➡️ ➡️<strong>➡️ ➡️</strong>", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Bold, 5, 5}});
check_parse_html("➡️ ➡️<u>➡️ ➡️</u>", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Underline, 5, 5}});
check_parse_html("➡️ ➡️<ins>➡️ ➡️</ins>", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Underline, 5, 5}});
check_parse_html("➡️ ➡️<s>➡️ ➡️</s>", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Strikethrough, 5, 5}});
check_parse_html("➡️ ➡️<strike>➡️ ➡️</strike>", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Strikethrough, 5, 5}});
check_parse_html("➡️ ➡️<del>➡️ ➡️</del>", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Strikethrough, 5, 5}});
check_parse_html("➡️ ➡️<i>➡️ ➡️</i><b>➡️ ➡️</b>", "➡️ ➡️➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Italic, 5, 5}, {td::MessageEntity::Type::Bold, 10, 5}});
check_parse_html("🏟 🏟<i>🏟 &lt🏟</i>", "🏟 🏟🏟 <🏟", {{td::MessageEntity::Type::Italic, 5, 6}});
check_parse_html("🏟 🏟<i>🏟 &gt;<b aba = caba>&lt🏟</b></i>", "🏟 🏟🏟 ><🏟",
{{td::MessageEntity::Type::Italic, 5, 7}, {td::MessageEntity::Type::Bold, 9, 3}});
check_parse_html("🏟 🏟&lt;<i aba = 190azAz-. >a</i>", "🏟 🏟<a",
{{td::MessageEntity::Type::Italic, 6, 1}});
check_parse_html("🏟 🏟&lt;<i aba = 190azAz-.>a</i>", "🏟 🏟<a",
{{td::MessageEntity::Type::Italic, 6, 1}});
check_parse_html("🏟 🏟&lt;<i aba = \"&lt;&gt;&quot;\">a</i>", "🏟 🏟<a",
{{td::MessageEntity::Type::Italic, 6, 1}});
check_parse_html("🏟 🏟&lt;<i aba = '&lt;&gt;&quot;'>a</i>", "🏟 🏟<a",
{{td::MessageEntity::Type::Italic, 6, 1}});
check_parse_html("🏟 🏟&lt;<i aba = '&lt;&gt;&quot;'>a</>", "🏟 🏟<a",
{{td::MessageEntity::Type::Italic, 6, 1}});
check_parse_html("🏟 🏟&lt;<i>🏟 🏟&lt;</>", "🏟 🏟<🏟 🏟<",
{{td::MessageEntity::Type::Italic, 6, 6}});
check_parse_html("🏟 🏟&lt;<i>a</ >", "🏟 🏟<a", {{td::MessageEntity::Type::Italic, 6, 1}});
check_parse_html("🏟 🏟&lt;<i>a</i >", "🏟 🏟<a", {{td::MessageEntity::Type::Italic, 6, 1}});
check_parse_html("🏟 🏟&lt;<b></b>", "🏟 🏟<", {});
check_parse_html("<i>\t</i>", "\t", {{td::MessageEntity::Type::Italic, 0, 1}});
check_parse_html("<i>\r</i>", "\r", {{td::MessageEntity::Type::Italic, 0, 1}});
check_parse_html("<i>\n</i>", "\n", {{td::MessageEntity::Type::Italic, 0, 1}});
check_parse_html("<a href=telegram.org>\t</a>", "\t",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://telegram.org/"}});
check_parse_html("<a href=telegram.org>\r</a>", "\r",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://telegram.org/"}});
check_parse_html("<a href=telegram.org>\n</a>", "\n",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://telegram.org/"}});
check_parse_html("<code><i><b> </b></i></code><i><b><code> </code></b></i>", " ",
{{td::MessageEntity::Type::Code, 0, 1},
{td::MessageEntity::Type::Bold, 0, 1},
{td::MessageEntity::Type::Italic, 0, 1},
{td::MessageEntity::Type::Code, 1, 1},
{td::MessageEntity::Type::Bold, 1, 1},
{td::MessageEntity::Type::Italic, 1, 1}});
check_parse_html("<i><b> </b> <code> </code></i>", " ",
{{td::MessageEntity::Type::Italic, 0, 3},
{td::MessageEntity::Type::Bold, 0, 1},
{td::MessageEntity::Type::Code, 2, 1}});
check_parse_html("<a href=telegram.org> </a>", " ",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://telegram.org/"}});
check_parse_html("<a href =\"telegram.org\" > </a>", " ",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://telegram.org/"}});
check_parse_html("<a href= 'telegram.org' > </a>", " ",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://telegram.org/"}});
check_parse_html("<a href= 'telegram.org?&lt;' > </a>", " ",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://telegram.org/?<"}});
check_parse_html("<a> </a>", " ", {});
check_parse_html("<a>telegram.org </a>", "telegram.org ", {});
check_parse_html("<a>telegram.org</a>", "telegram.org",
{{td::MessageEntity::Type::TextUrl, 0, 12, "http://telegram.org/"}});
check_parse_html("<a>https://telegram.org/asdsa?asdasdwe#12e3we</a>", "https://telegram.org/asdsa?asdasdwe#12e3we",
{{td::MessageEntity::Type::TextUrl, 0, 42, "https://telegram.org/asdsa?asdasdwe#12e3we"}});
check_parse_html("🏟 🏟&lt;<pre >🏟 🏟&lt;</>", "🏟 🏟<🏟 🏟<",
{{td::MessageEntity::Type::Pre, 6, 6}});
check_parse_html("🏟 🏟&lt;<code >🏟 🏟&lt;</>", "🏟 🏟<🏟 🏟<",
{{td::MessageEntity::Type::Code, 6, 6}});
check_parse_html("🏟 🏟&lt;<pre><code>🏟 🏟&lt;</code></>", "🏟 🏟<🏟 🏟<",
{{td::MessageEntity::Type::Pre, 6, 6}, {td::MessageEntity::Type::Code, 6, 6}});
check_parse_html("🏟 🏟&lt;<pre><code class=\"language-\">🏟 🏟&lt;</code></>", "🏟 🏟<🏟 🏟<",
{{td::MessageEntity::Type::Pre, 6, 6}, {td::MessageEntity::Type::Code, 6, 6}});
check_parse_html("🏟 🏟&lt;<pre><code class=\"language-fift\">🏟 🏟&lt;</></>", "🏟 🏟<🏟 🏟<",
{{td::MessageEntity::Type::PreCode, 6, 6, "fift"}});
check_parse_html("🏟 🏟&lt;<code class=\"language-fift\"><pre>🏟 🏟&lt;</></>", "🏟 🏟<🏟 🏟<",
{{td::MessageEntity::Type::PreCode, 6, 6, "fift"}});
check_parse_html("🏟 🏟&lt;<pre><code class=\"language-fift\">🏟 🏟&lt;</> </>", "🏟 🏟<🏟 🏟< ",
{{td::MessageEntity::Type::Pre, 6, 7}, {td::MessageEntity::Type::Code, 6, 6}});
check_parse_html("🏟 🏟&lt;<pre> <code class=\"language-fift\">🏟 🏟&lt;</></>", "🏟 🏟< 🏟 🏟<",
{{td::MessageEntity::Type::Pre, 6, 7}, {td::MessageEntity::Type::Code, 7, 6}});
}
static void check_parse_markdown(td::string text, const td::string &result,
const td::vector<td::MessageEntity> &entities) {
auto r_entities = td::parse_markdown_v2(text);
ASSERT_TRUE(r_entities.is_ok());
ASSERT_EQ(entities, r_entities.ok());
ASSERT_STREQ(result, text);
}
static void check_parse_markdown(td::string text, td::Slice error_message) {
auto r_entities = td::parse_markdown_v2(text);
ASSERT_TRUE(r_entities.is_error());
ASSERT_EQ(400, r_entities.error().code());
ASSERT_STREQ(error_message, r_entities.error().message());
}
TEST(MessageEntities, parse_markdown) {
td::Slice reserved_characters("]()>#+-=|{}.!");
td::Slice begin_characters("_*[~`");
for (char c = 1; c < 126; c++) {
if (begin_characters.find(c) != td::Slice::npos) {
continue;
}
td::string text(1, c);
if (reserved_characters.find(c) == td::Slice::npos) {
check_parse_markdown(text, text, {});
} else {
check_parse_markdown(
text, PSLICE() << "Character '" << c << "' is reserved and must be escaped with the preceding '\\'");
td::string escaped_text = "\\" + text;
check_parse_markdown(escaped_text, text, {});
}
}
check_parse_markdown("🏟 🏟_abacaba", "Can't find end of Italic entity at byte offset 9");
check_parse_markdown("🏟 🏟_abac * asd ", "Can't find end of Bold entity at byte offset 15");
check_parse_markdown("🏟 🏟_abac * asd _", "Can't find end of Italic entity at byte offset 21");
check_parse_markdown("🏟 🏟`", "Can't find end of Code entity at byte offset 9");
check_parse_markdown("🏟 🏟```", "Can't find end of Pre entity at byte offset 9");
check_parse_markdown("🏟 🏟```a", "Can't find end of Pre entity at byte offset 9");
check_parse_markdown("🏟 🏟```a ", "Can't find end of PreCode entity at byte offset 9");
check_parse_markdown("🏟 🏟__🏟 🏟_", "Can't find end of Italic entity at byte offset 20");
check_parse_markdown("🏟 🏟_🏟 🏟__", "Can't find end of Underline entity at byte offset 19");
check_parse_markdown("🏟 🏟```🏟 🏟`", "Can't find end of Code entity at byte offset 21");
check_parse_markdown("🏟 🏟```🏟 🏟_", "Can't find end of PreCode entity at byte offset 9");
check_parse_markdown("🏟 🏟```🏟 🏟\\`", "Can't find end of PreCode entity at byte offset 9");
check_parse_markdown("[telegram\\.org](asd\\)", "Can't find end of a URL at byte offset 16");
check_parse_markdown("[telegram\\.org](", "Can't find end of a URL at byte offset 16");
check_parse_markdown("[telegram\\.org](asd", "Can't find end of a URL at byte offset 16");
check_parse_markdown("🏟 🏟__🏟 _🏟___", "Can't find end of Italic entity at byte offset 23");
check_parse_markdown("🏟 🏟__", "Can't find end of Underline entity at byte offset 9");
check_parse_markdown("", "", {});
check_parse_markdown("\\\\", "\\", {});
check_parse_markdown("\\\\\\", "\\\\", {});
check_parse_markdown("\\\\\\\\\\_\\*\\`", "\\\\_*`", {});
check_parse_markdown("➡️ ➡️", "➡️ ➡️", {});
check_parse_markdown("🏟 🏟``", "🏟 🏟", {});
check_parse_markdown("🏟 🏟_abac \\* asd _", "🏟 🏟abac * asd ", {{td::MessageEntity::Type::Italic, 5, 11}});
check_parse_markdown("🏟 \\.🏟_🏟\\. 🏟_", "🏟 .🏟🏟. 🏟", {{td::MessageEntity::Type::Italic, 6, 6}});
check_parse_markdown("\\\\\\a\\b\\c\\d\\e\\f\\1\\2\\3\\4\\➡️\\", "\\abcdef1234\\➡️\\", {});
check_parse_markdown("➡️ ➡️_➡️ ➡️_", "➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Italic, 5, 5}});
check_parse_markdown("➡️ ➡️_➡️ ➡️_*➡️ ➡️*", "➡️ ➡️➡️ ➡️➡️ ➡️",
{{td::MessageEntity::Type::Italic, 5, 5}, {td::MessageEntity::Type::Bold, 10, 5}});
check_parse_markdown("🏟 🏟_🏟 \\.🏟_", "🏟 🏟🏟 .🏟", {{td::MessageEntity::Type::Italic, 5, 6}});
check_parse_markdown("🏟 🏟_🏟 *🏟*_", "🏟 🏟🏟 🏟",
{{td::MessageEntity::Type::Italic, 5, 5}, {td::MessageEntity::Type::Bold, 8, 2}});
check_parse_markdown("🏟 🏟_🏟 __🏟___", "🏟 🏟🏟 🏟",
{{td::MessageEntity::Type::Italic, 5, 5}, {td::MessageEntity::Type::Underline, 8, 2}});
check_parse_markdown("🏟 🏟__🏟 _🏟_ __", "🏟 🏟🏟 🏟 ",
{{td::MessageEntity::Type::Underline, 5, 6}, {td::MessageEntity::Type::Italic, 8, 2}});
check_parse_markdown("🏟 🏟__🏟 _🏟_\\___", "🏟 🏟🏟 🏟_",
{{td::MessageEntity::Type::Underline, 5, 6}, {td::MessageEntity::Type::Italic, 8, 2}});
check_parse_markdown("🏟 🏟`🏟 🏟```", "🏟 🏟🏟 🏟", {{td::MessageEntity::Type::Code, 5, 5}});
check_parse_markdown("🏟 🏟```🏟 🏟```", "🏟 🏟 🏟",
{{td::MessageEntity::Type::PreCode, 5, 3, "🏟"}});
check_parse_markdown("🏟 🏟```🏟\n🏟```", "🏟 🏟🏟",
{{td::MessageEntity::Type::PreCode, 5, 2, "🏟"}});
check_parse_markdown("🏟 🏟```🏟\r🏟```", "🏟 🏟🏟",
{{td::MessageEntity::Type::PreCode, 5, 2, "🏟"}});
check_parse_markdown("🏟 🏟```🏟\n\r🏟```", "🏟 🏟🏟",
{{td::MessageEntity::Type::PreCode, 5, 2, "🏟"}});
check_parse_markdown("🏟 🏟```🏟\r\n🏟```", "🏟 🏟🏟",
{{td::MessageEntity::Type::PreCode, 5, 2, "🏟"}});
check_parse_markdown("🏟 🏟```🏟\n\n🏟```", "🏟 🏟\n🏟",
{{td::MessageEntity::Type::PreCode, 5, 3, "🏟"}});
check_parse_markdown("🏟 🏟```🏟\r\r🏟```", "🏟 🏟\r🏟",
{{td::MessageEntity::Type::PreCode, 5, 3, "🏟"}});
check_parse_markdown("🏟 🏟```🏟 \\\\\\`🏟```", "🏟 🏟 \\`🏟",
{{td::MessageEntity::Type::PreCode, 5, 5, "🏟"}});
check_parse_markdown("🏟 🏟**", "🏟 🏟", {});
check_parse_markdown("🏟 🏟``", "🏟 🏟", {});
check_parse_markdown("🏟 🏟``````", "🏟 🏟", {});
check_parse_markdown("🏟 🏟____", "🏟 🏟", {});
check_parse_markdown("`_* *_`__*` `*__", "_* *_ ",
{{td::MessageEntity::Type::Code, 0, 5},
{td::MessageEntity::Type::Code, 5, 1},
{td::MessageEntity::Type::Bold, 5, 1},
{td::MessageEntity::Type::Underline, 5, 1}});
check_parse_markdown("_* * ` `_", " ",
{{td::MessageEntity::Type::Italic, 0, 3},
{td::MessageEntity::Type::Bold, 0, 1},
{td::MessageEntity::Type::Code, 2, 1}});
check_parse_markdown("[](telegram.org)", "", {});
check_parse_markdown("[ ](telegram.org)", " ", {{td::MessageEntity::Type::TextUrl, 0, 1, "http://telegram.org/"}});
check_parse_markdown("[ ](as)", " ", {});
check_parse_markdown("[telegram\\.org]", "telegram.org",
{{td::MessageEntity::Type::TextUrl, 0, 12, "http://telegram.org/"}});
check_parse_markdown("[telegram\\.org]a", "telegram.orga",
{{td::MessageEntity::Type::TextUrl, 0, 12, "http://telegram.org/"}});
check_parse_markdown("[telegram\\.org](telegram.dog)", "telegram.org",
{{td::MessageEntity::Type::TextUrl, 0, 12, "http://telegram.dog/"}});
check_parse_markdown("[telegram\\.org](https://telegram.dog?)", "telegram.org",
{{td::MessageEntity::Type::TextUrl, 0, 12, "https://telegram.dog/?"}});
check_parse_markdown("[telegram\\.org](https://telegram.dog?\\\\\\()", "telegram.org",
{{td::MessageEntity::Type::TextUrl, 0, 12, "https://telegram.dog/?\\("}});
check_parse_markdown("[telegram\\.org]()", "telegram.org", {});
check_parse_markdown("[telegram\\.org](asdasd)", "telegram.org", {});
check_parse_markdown("[telegram\\.org](tg:user?id=123456)", "telegram.org", {{0, 12, td::UserId(123456)}});
}
static void check_parse_markdown_v3(td::string text, td::vector<td::MessageEntity> entities, const td::string &result,
const td::vector<td::MessageEntity> &result_entities, bool fix = false) {
auto parsed_text = td::parse_markdown_v3({std::move(text), std::move(entities)});
if (fix) {
ASSERT_TRUE(fix_formatted_text(parsed_text.text, parsed_text.entities, true, true, true, true).is_ok());
}
ASSERT_STREQ(result, parsed_text.text);
ASSERT_EQ(result_entities, parsed_text.entities);
}
static void check_parse_markdown_v3(td::string text, const td::string &result,
const td::vector<td::MessageEntity> &result_entities, bool fix = false) {
check_parse_markdown_v3(std::move(text), td::vector<td::MessageEntity>(), result, result_entities, fix);
}
TEST(MessageEntities, parse_markdown_v3) {
check_parse_markdown_v3("🏟````🏟``🏟`aba🏟```c🏟`aba🏟 daba🏟```c🏟`aba🏟```🏟 `🏟``🏟```",
"🏟````🏟``🏟aba🏟```c🏟aba🏟 daba🏟c🏟`aba🏟🏟 `🏟``🏟```",
{{td::MessageEntity::Type::Code, 12, 11}, {td::MessageEntity::Type::Pre, 35, 9}});
check_parse_markdown_v3(
"🏟````🏟``🏟`aba🏟```c🏟`aba🏟 daba🏟```c🏟`aba🏟🏟```🏟 `🏟``🏟```",
{{td::MessageEntity::Type::Italic, 12, 1},
{td::MessageEntity::Type::Italic, 44, 1},
{td::MessageEntity::Type::Bold, 45, 1},
{td::MessageEntity::Type::Bold, 49, 2}},
"🏟````🏟``🏟`aba🏟c🏟`aba🏟 daba🏟c🏟`aba🏟🏟🏟 `🏟``🏟",
{{td::MessageEntity::Type::Italic, 12, 1},
{td::MessageEntity::Type::Pre, 18, 16},
{td::MessageEntity::Type::Italic, 38, 1},
{td::MessageEntity::Type::Bold, 39, 1},
{td::MessageEntity::Type::Bold, 43, 2},
{td::MessageEntity::Type::Pre, 45, 10}});
check_parse_markdown_v3("` `", " ", {{td::MessageEntity::Type::Code, 0, 1}});
check_parse_markdown_v3("`\n`", "\n", {{td::MessageEntity::Type::Code, 0, 1}});
check_parse_markdown_v3("` `a", " a", {{td::MessageEntity::Type::Code, 0, 1}}, true);
check_parse_markdown_v3("`\n`a", "\na", {}, true);
check_parse_markdown_v3("``", "``", {});
check_parse_markdown_v3("[a](b[c](t.me)", "[a](b[c](t.me)", {});
check_parse_markdown_v3("[](t.me)", "[](t.me)", {});
check_parse_markdown_v3("[ ](t.me)", " ", {{td::MessageEntity::Type::TextUrl, 0, 1, "http://t.me/"}});
check_parse_markdown_v3("[ ](t.me)", "", {}, true);
check_parse_markdown_v3("[ ](t.me)a", " a", {{td::MessageEntity::Type::TextUrl, 0, 1, "http://t.me/"}}, true);
check_parse_markdown_v3(
"[ ](t.me) [ ](t.me)", {{td::MessageEntity::Type::TextUrl, 8, 1, "http://t.me/"}, {10, 1, td::UserId(1)}},
"[ ](t.me) [ ](t.me)", {{td::MessageEntity::Type::TextUrl, 8, 1, "http://t.me/"}, {10, 1, td::UserId(1)}});
check_parse_markdown_v3("[\n](t.me)", "\n", {{td::MessageEntity::Type::TextUrl, 0, 1, "http://t.me/"}});
check_parse_markdown_v3("[\n](t.me)a", "\na", {}, true);
check_parse_markdown_v3("asd[abcd](google.com)", {{td::MessageEntity::Type::Italic, 0, 5}}, "asdabcd",
{{td::MessageEntity::Type::Italic, 0, 3},
{td::MessageEntity::Type::TextUrl, 3, 4, "http://google.com/"},
{td::MessageEntity::Type::Italic, 3, 1}});
check_parse_markdown_v3("asd[abcd](google.com)efg[hi](https://t.me?t=1#h)e",
{{td::MessageEntity::Type::Italic, 0, 5}, {td::MessageEntity::Type::Italic, 18, 31}},
"asdabcdefghie",
{{td::MessageEntity::Type::Italic, 0, 3},
{td::MessageEntity::Type::TextUrl, 3, 4, "http://google.com/"},
{td::MessageEntity::Type::Italic, 3, 1},
{td::MessageEntity::Type::Italic, 7, 3},
{td::MessageEntity::Type::TextUrl, 10, 2, "https://t.me/?t=1#h"},
{td::MessageEntity::Type::Italic, 10, 2},
{td::MessageEntity::Type::Italic, 12, 1}});
check_parse_markdown_v3(
"🏟🏟🏟[🏟🏟🏟🏟🏟](www.🤙.tk#1)🤙🤙🤙[🏟🏟🏟🏟](www.🤙.tk#2)🤙🤙🤙["
"🏟🏟🏟🏟](www.🤙.tk#3)🏟🏟🏟[🏟🏟🏟🏟](www.🤙.tk#4)🤙🤙",
"🏟🏟🏟🏟🏟🏟🏟🏟🤙🤙🤙🏟🏟🏟🏟🤙🤙🤙🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟"
"🏟🤙🤙",
{{td::MessageEntity::Type::TextUrl, 6, 10, "http://www.🤙.tk/#1"},
{td::MessageEntity::Type::TextUrl, 22, 8, "http://www.🤙.tk/#2"},
{td::MessageEntity::Type::TextUrl, 36, 8, "http://www.🤙.tk/#3"},
{td::MessageEntity::Type::TextUrl, 50, 8, "http://www.🤙.tk/#4"}});
check_parse_markdown_v3(
"[🏟🏟🏟🏟🏟](www.🤙.tk#1)[🏟🏟🏟🏟](www.🤙.tk#2)[🏟🏟🏟🏟](www.🤙.tk#3)["
"🏟🏟🏟🏟](www.🤙.tk#4)",
"🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟",
{{td::MessageEntity::Type::TextUrl, 0, 10, "http://www.🤙.tk/#1"},
{td::MessageEntity::Type::TextUrl, 10, 8, "http://www.🤙.tk/#2"},
{td::MessageEntity::Type::TextUrl, 18, 8, "http://www.🤙.tk/#3"},
{td::MessageEntity::Type::TextUrl, 26, 8, "http://www.🤙.tk/#4"}});
check_parse_markdown_v3(
"🏟🏟🏟[🏟🏟🏟🏟🏟](www.🤙.tk)🤙🤙🤙[🏟🏟🏟🏟](www.🤙.tk)🤙🤙🤙["
"🏟🏟🏟🏟](www.🤙.tk)🏟🏟🏟[🏟🏟🏟🏟](www.🤙.tk)🤙🤙",
{{td::MessageEntity::Type::Bold, 0, 2},
{td::MessageEntity::Type::Bold, 4, 2},
{td::MessageEntity::Type::Bold, 7, 2},
{td::MessageEntity::Type::Bold, 11, 2},
{td::MessageEntity::Type::Bold, 15, 2},
{td::MessageEntity::Type::Bold, 18, 2},
{td::MessageEntity::Type::Bold, 26, 2},
{31, 2, td::UserId(1)},
{td::MessageEntity::Type::Bold, 35, 1},
{td::MessageEntity::Type::Bold, 44, 2},
{td::MessageEntity::Type::Bold, 50, 2},
{td::MessageEntity::Type::Bold, 54, 2},
{56, 2, td::UserId(2)},
{td::MessageEntity::Type::Bold, 58, 7},
{60, 2, td::UserId(3)},
{td::MessageEntity::Type::Bold, 67, 7},
{td::MessageEntity::Type::Bold, 80, 7},
{td::MessageEntity::Type::Bold, 89, 25}},
"🏟🏟🏟🏟🏟🏟🏟🏟🤙🤙🤙🏟🏟🏟🏟🤙🤙🤙🏟🏟🏟🏟🏟🏟🏟🏟🏟🏟"
"🏟🤙🤙",
{{td::MessageEntity::Type::Bold, 0, 2},
{td::MessageEntity::Type::Bold, 4, 2},
{td::MessageEntity::Type::TextUrl, 6, 10, "http://www.🤙.tk/"},
{td::MessageEntity::Type::Bold, 6, 2},
{td::MessageEntity::Type::Bold, 10, 2},
{td::MessageEntity::Type::Bold, 14, 2},
{18, 2, td::UserId(1)},
{td::MessageEntity::Type::TextUrl, 22, 8, "http://www.🤙.tk/"},
{30, 2, td::UserId(2)},
{td::MessageEntity::Type::Bold, 32, 2},
{34, 2, td::UserId(3)},
{td::MessageEntity::Type::Bold, 34, 2},
{td::MessageEntity::Type::TextUrl, 36, 8, "http://www.🤙.tk/"},
{td::MessageEntity::Type::Bold, 36, 2},
{td::MessageEntity::Type::Bold, 40, 4},
{td::MessageEntity::Type::Bold, 44, 4},
{td::MessageEntity::Type::TextUrl, 50, 8, "http://www.🤙.tk/"},
{td::MessageEntity::Type::Bold, 50, 8},
{td::MessageEntity::Type::Bold, 58, 4}});
check_parse_markdown_v3("[`a`](t.me) [b](t.me)", {{td::MessageEntity::Type::Code, 13, 1}}, "[a](t.me) [b](t.me)",
{{td::MessageEntity::Type::Code, 1, 1}, {td::MessageEntity::Type::Code, 11, 1}});
check_parse_markdown_v3(
"[text](example.com)",
{{td::MessageEntity::Type::Strikethrough, 0, 1}, {td::MessageEntity::Type::Strikethrough, 5, 14}}, "text",
{{td::MessageEntity::Type::TextUrl, 0, 4, "http://example.com/"}});
check_parse_markdown_v3("🏟[🏟](t.me) `🏟` [🏟](t.me) `a`", "🏟🏟 🏟 🏟 a",
{{td::MessageEntity::Type::TextUrl, 2, 2, "http://t.me/"},
{td::MessageEntity::Type::Code, 5, 2},
{td::MessageEntity::Type::TextUrl, 8, 2, "http://t.me/"},
{td::MessageEntity::Type::Code, 11, 1}});
check_parse_markdown_v3("__ __", " ", {{td::MessageEntity::Type::Italic, 0, 1}});
check_parse_markdown_v3("__\n__", "\n", {{td::MessageEntity::Type::Italic, 0, 1}});
check_parse_markdown_v3("__ __a", " a", {}, true);
check_parse_markdown_v3("__\n__a", "\na", {}, true);
check_parse_markdown_v3("**** __a__ **b** ~~c~~", "**** a b c",
{{td::MessageEntity::Type::Italic, 5, 1},
{td::MessageEntity::Type::Bold, 7, 1},
{td::MessageEntity::Type::Strikethrough, 9, 1}});
check_parse_markdown_v3("тест __аааа__ **бббб** ~~вввв~~", "тест аааа бббб вввв",
{{td::MessageEntity::Type::Italic, 5, 4},
{td::MessageEntity::Type::Bold, 10, 4},
{td::MessageEntity::Type::Strikethrough, 15, 4}});
check_parse_markdown_v3("___a___ ***b** ~c~~", "___a___ ***b** ~c~~", {});
check_parse_markdown_v3(
"__asd[ab__cd](t.me)", "asdabcd",
{{td::MessageEntity::Type::Italic, 0, 5}, {td::MessageEntity::Type::TextUrl, 3, 4, "http://t.me/"}});
check_parse_markdown_v3("__asd[ab__cd](t.me)", "asdabcd",
{{td::MessageEntity::Type::Italic, 0, 3},
{td::MessageEntity::Type::TextUrl, 3, 4, "http://t.me/"},
{td::MessageEntity::Type::Italic, 3, 2}},
true);
check_parse_markdown_v3("__[ab_](t.me)_", "__ab__", {{td::MessageEntity::Type::TextUrl, 2, 3, "http://t.me/"}});
check_parse_markdown_v3(
"__[ab__](t.me)_", "ab_",
{{td::MessageEntity::Type::TextUrl, 0, 2, "http://t.me/"}, {td::MessageEntity::Type::Italic, 0, 2}});
check_parse_markdown_v3("__[__ab__](t.me)__", "____ab____",
{{td::MessageEntity::Type::TextUrl, 2, 6, "http://t.me/"}});
check_parse_markdown_v3(
"__[__ab__](t.me)a__", "____aba",
{{td::MessageEntity::Type::TextUrl, 2, 4, "http://t.me/"}, {td::MessageEntity::Type::Italic, 6, 1}});
check_parse_markdown_v3("`a` __ab__", {{td::MessageEntity::Type::Bold, 6, 3}}, "a __ab__",
{{td::MessageEntity::Type::Code, 0, 1}, {td::MessageEntity::Type::Bold, 4, 3}});
check_parse_markdown_v3("`a` __ab__", {{td::MessageEntity::Type::Underline, 5, 1}}, "a __ab__",
{{td::MessageEntity::Type::Code, 0, 1}, {td::MessageEntity::Type::Underline, 3, 1}});
check_parse_markdown_v3("`a` @test__test__test", "a @test__test__test", {{td::MessageEntity::Type::Code, 0, 1}});
check_parse_markdown_v3("`a` #test__test__test", "a #test__test__test", {{td::MessageEntity::Type::Code, 0, 1}});
check_parse_markdown_v3("`a` __@test_test_test__", "a @test_test_test",
{{td::MessageEntity::Type::Code, 0, 1}, {td::MessageEntity::Type::Italic, 2, 15}});
check_parse_markdown_v3("`a` __#test_test_test__", "a #test_test_test",
{{td::MessageEntity::Type::Code, 0, 1}, {td::MessageEntity::Type::Italic, 2, 15}});
check_parse_markdown_v3("[a](t.me) __@test**test**test__", "a @testtesttest",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://t.me/"},
{td::MessageEntity::Type::Italic, 2, 13},
{td::MessageEntity::Type::Bold, 7, 4}});
check_parse_markdown_v3("[a](t.me) __#test~~test~~test__", "a #testtesttest",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://t.me/"},
{td::MessageEntity::Type::Italic, 2, 13},
{td::MessageEntity::Type::Strikethrough, 7, 4}});
check_parse_markdown_v3("[a](t.me) __@test__test__test__", "a @testtesttest",
{{td::MessageEntity::Type::TextUrl, 0, 1, "http://t.me/"},
{td::MessageEntity::Type::Italic, 2, 5},
{td::MessageEntity::Type::Italic, 11, 4}});
check_parse_markdown_v3("__**~~__gh**~~", "gh",
{{td::MessageEntity::Type::Bold, 0, 2}, {td::MessageEntity::Type::Strikethrough, 0, 2}});
check_parse_markdown_v3("__ab**cd~~ef__gh**ij~~", "abcdefghij",
{{td::MessageEntity::Type::Italic, 0, 6},
{td::MessageEntity::Type::Bold, 2, 6},
{td::MessageEntity::Type::Strikethrough, 4, 6}});
check_parse_markdown_v3("__ab**cd~~ef__gh**ij~~", "abcdefghij",
{{td::MessageEntity::Type::Italic, 0, 2},
{td::MessageEntity::Type::Bold, 2, 2},
{td::MessageEntity::Type::Italic, 2, 2},
{td::MessageEntity::Type::Strikethrough, 4, 6},
{td::MessageEntity::Type::Bold, 4, 4},
{td::MessageEntity::Type::Italic, 4, 2}},
true);
check_parse_markdown_v3("__ab**[cd~~ef__](t.me)gh**ij~~", "abcdefghij",
{{td::MessageEntity::Type::Italic, 0, 6},
{td::MessageEntity::Type::Bold, 2, 6},
{td::MessageEntity::Type::TextUrl, 2, 4, "http://t.me/"},
{td::MessageEntity::Type::Strikethrough, 4, 6}});
check_parse_markdown_v3("__ab**[cd~~e](t.me)f__gh**ij~~", "abcdefghij",
{{td::MessageEntity::Type::Italic, 0, 6},
{td::MessageEntity::Type::Bold, 2, 6},
{td::MessageEntity::Type::TextUrl, 2, 3, "http://t.me/"},
{td::MessageEntity::Type::Strikethrough, 4, 6}});
check_parse_markdown_v3("__ab**[cd~~](t.me)ef__gh**ij~~", "abcdefghij",
{{td::MessageEntity::Type::Italic, 0, 6},
{td::MessageEntity::Type::Bold, 2, 6},
{td::MessageEntity::Type::TextUrl, 2, 2, "http://t.me/"},
{td::MessageEntity::Type::Strikethrough, 4, 6}});
check_parse_markdown_v3("[__**bold italic link**__](example.com)", "bold italic link",
{{td::MessageEntity::Type::TextUrl, 0, 16, "http://example.com/"},
{td::MessageEntity::Type::Bold, 0, 16},
{td::MessageEntity::Type::Italic, 0, 16}});
check_parse_markdown_v3(
"__italic__ ~~strikethrough~~ **bold** `code` ```pre``` __[italic__ text_url](telegram.org) __italic**bold "
"italic__bold**__italic__ ~~strikethrough~~ **bold** `code` ```pre``` __[italic__ text_url](telegram.org) "
"__italic**bold italic__bold**",
"italic strikethrough bold code pre italic text_url italicbold italicbolditalic strikethrough bold code pre "
"italic text_url italicbold italicbold",
{{td::MessageEntity::Type::Italic, 0, 6},
{td::MessageEntity::Type::Strikethrough, 7, 13},
{td::MessageEntity::Type::Bold, 21, 4},
{td::MessageEntity::Type::Code, 26, 4},
{td::MessageEntity::Type::Pre, 31, 3},
{td::MessageEntity::Type::TextUrl, 35, 15, "http://telegram.org/"},
{td::MessageEntity::Type::Italic, 35, 6},
{td::MessageEntity::Type::Italic, 51, 17},
{td::MessageEntity::Type::Bold, 57, 15},
{td::MessageEntity::Type::Italic, 72, 6},
{td::MessageEntity::Type::Strikethrough, 79, 13},
{td::MessageEntity::Type::Bold, 93, 4},
{td::MessageEntity::Type::Code, 98, 4},
{td::MessageEntity::Type::Pre, 103, 3},
{td::MessageEntity::Type::TextUrl, 107, 15, "http://telegram.org/"},
{td::MessageEntity::Type::Italic, 107, 6},
{td::MessageEntity::Type::Italic, 123, 17},
{td::MessageEntity::Type::Bold, 129, 15}});
td::vector<td::string> parts{"a", " #test ", "__", "**", "~~", "[", "](t.me)", "`"};
td::vector<td::MessageEntity::Type> types{
td::MessageEntity::Type::Bold, td::MessageEntity::Type::Italic, td::MessageEntity::Type::Underline,
td::MessageEntity::Type::Strikethrough, td::MessageEntity::Type::Code, td::MessageEntity::Type::Pre,
td::MessageEntity::Type::PreCode, td::MessageEntity::Type::TextUrl, td::MessageEntity::Type::MentionName,
td::MessageEntity::Type::Cashtag};
for (size_t test_n = 0; test_n < 1000; test_n++) {
td::string str;
int part_n = td::Random::fast(1, 200);
for (int i = 0; i < part_n; i++) {
str += parts[td::Random::fast(0, static_cast<int>(parts.size()) - 1)];
}
td::vector<td::MessageEntity> entities;
int entity_n = td::Random::fast(1, 20);
for (int i = 0; i < entity_n; i++) {
auto type = types[td::Random::fast(0, static_cast<int>(types.size()) - 1)];
td::int32 offset = td::Random::fast(0, static_cast<int>(str.size()) - 1);
auto max_length = static_cast<int>(str.size() - offset);
if ((test_n & 1) != 0 && max_length > 4) {
max_length = 4;
}
td::int32 length = td::Random::fast(0, max_length);
entities.emplace_back(type, offset, length);
}
ASSERT_TRUE(fix_formatted_text(str, entities, true, true, true, true).is_ok());
td::parse_markdown_v3({std::move(str), std::move(entities)});
}
}
View Git Blame Copy Permalink