tdlight-team/tdlight
tdlight-team
/
tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add fix_formatted_text test. 

GitOrigin-RevId: 90e0f7d0fdcef077f88927311a9c0678711c5da8
This commit is contained in:
levlam 2019-01-16 21:18:26 +03:00
parent 2fb5fbfd4c
commit eaf7e36532
2 changed files with 178 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
td/telegram/MessageEntity.cpp
View File

@ -2160,6 +2160,11 @@ Status fix_formatted_text(string &text, vector<MessageEntity> &entities, bool al
} else {
// rtrim
result.resize(last_non_whitespace_pos);
while (!entities.empty() && entities.back().offset >= last_non_whitespace_utf16_offset) {
CHECK(entities.back().type == MessageEntity::Type::TextUrl ||
entities.back().type == MessageEntity::Type::MentionName);
entities.pop_back();
}
for (auto &entity : entities) {
if (entity.offset + entity.length > last_non_whitespace_utf16_offset) {
entity.length = last_non_whitespace_utf16_offset - entity.offset;

173
test/message_entities.cpp
View File

@ -529,3 +529,176 @@ TEST(MessageEntities, url) {
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(string str, vector<MessageEntity> entities, string expected_str,
vector<MessageEntity> expected_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_ok());
ASSERT_STREQ(expected_str, str);
ASSERT_EQ(expected_entities, entities);
}
static void check_fix_formatted_text(string str, vector<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) {
string str;
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 (int32 i = 33; i <= 35; i++) {
vector<MessageEntity> entities;
entities.emplace_back(MessageEntity::Type::Bold, 0, i);
vector<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);
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++) {
vector<MessageEntity> entities;
entities.emplace_back(MessageEntity::Type::Bold, i, 1);
if (i == 0 || i == 1 || i == 3 || i == 4) {
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, {}, false, false, false, false);
}
}
str = " /test @abaca #ORD $ABC telegram.org ";
for (auto for_draft : {false, true}) {
int32 shift = for_draft ? 2 : 0;
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}) {
vector<MessageEntity> entities;
if (!skip_new_entities) {
if (!skip_bot_commands) {
entities.emplace_back(MessageEntity::Type::BotCommand, shift, 5);
}
entities.emplace_back(MessageEntity::Type::Mention, shift + 6, 6);
entities.emplace_back(MessageEntity::Type::Hashtag, shift + 13, 4);
entities.emplace_back(MessageEntity::Type::Cashtag, shift + 18, 4);
entities.emplace_back(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 (int32 length = 1; length <= 3; length++) {
for (int32 offset = 0; static_cast<size_t>(offset + length) <= str.size(); offset++) {
for (auto type : {MessageEntity::Type::Bold, MessageEntity::Type::Url, MessageEntity::Type::TextUrl,
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--;
}
vector<MessageEntity> entities;
entities.emplace_back(type, offset, length);
vector<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 == MessageEntity::Type::TextUrl ||
type == 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 (int32 length = -10; length <= 10; length++) {
for (int32 offset = -10; offset <= 10; offset++) {
vector<MessageEntity> entities;
entities.emplace_back(MessageEntity::Type::Bold, offset, length);
vector<MessageEntity> fixed_entities;
if (length > 0 && offset >= 0 && static_cast<size_t>(length + offset) <= str.size() &&
(length >= 2 || offset != 3)) {
fixed_entities.emplace_back(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 (int32 length = 1; length <= 7; length++) {
for (int32 offset = 0; offset <= 8 - length; offset++) {
for (int32 length2 = 1; length2 <= 7; length2++) {
for (int32 offset2 = 0; offset2 <= 8 - length2; offset2++) {
if (offset != offset2) {
vector<MessageEntity> entities;
entities.emplace_back(MessageEntity::Type::TextUrl, offset, length);
entities.emplace_back(MessageEntity::Type::TextUrl, offset2, length2);
vector<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);
}
}
}
}
}
}