Add test for CountryInfoManager::get_phone_number_info_sync.

test/CMakeLists.txt

@@ -4,6 +4,7 @@ endif()
 
 #SOURCE SETS
 set(TD_TEST_SOURCE
+  ${CMAKE_CURRENT_SOURCE_DIR}/country_info.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/db.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/http.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/link.cpp

test/country_info.cpp

@@ -0,0 +1,88 @@
+//
+// Copyright Aliaksei Levin ([email protected]), Arseny Smirnov ([email protected]) 2014-2021
+//
+// 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/CountryInfoManager.h"
+
+#include "td/utils/common.h"
+#include "td/utils/tests.h"
+
+static void check_phone_number_info(const td::string &phone_number_prefix, const td::string &country_code,
+                                    const td::string &calling_code, const td::string &formatted_phone_number) {
+  auto result = td::CountryInfoManager::get_phone_number_info_sync(td::string(), phone_number_prefix);
+  CHECK(result != nullptr);
+  // LOG(ERROR) << phone_number_prefix << ' ' << to_string(result);
+  if (result->country_ == nullptr) {
+    CHECK(country_code.empty());
+  } else {
+    CHECK(result->country_->country_code_ == country_code);
+  }
+  CHECK(result->country_calling_code_ == calling_code);
+  CHECK(result->formatted_phone_number_ == formatted_phone_number);
+}
+
+TEST(CountryInfo, phone_number_info) {
+  check_phone_number_info("", "", "", "");
+  check_phone_number_info("aba c aba", "", "", "");
+
+  td::string str;
+  td::string reverse_str;
+  for (auto i = 0; i < 256; i++) {
+    str += static_cast<char>(i);
+    reverse_str += static_cast<char>(255 - i);
+  }
+  check_phone_number_info(str, "", "", "0123456789");
+  check_phone_number_info(reverse_str, "IR", "98", "765 432 10--");
+
+  check_phone_number_info("1", "US", "1", "--- --- ----");
+  check_phone_number_info("12", "US", "1", "2-- --- ----");
+  check_phone_number_info("126", "US", "1", "26- --- ----");
+  check_phone_number_info("128", "US", "1", "28- --- ----");
+  check_phone_number_info("1289", "CA", "1", "289 --- ----");
+  check_phone_number_info("1289123123", "CA", "1", "289 123 123-");
+  check_phone_number_info("128912312345", "CA", "1", "289 123 12345");
+  check_phone_number_info("1268", "AG", "1268", "--- ----");
+  check_phone_number_info("126801", "AG", "1268", "01- ----");
+  check_phone_number_info("12680123", "AG", "1268", "012 3---");
+  check_phone_number_info("12680123456", "AG", "1268", "012 3456");
+  check_phone_number_info("1268012345678", "AG", "1268", "012 345678");
+  check_phone_number_info("7", "RU", "7", "--- --- ----");
+  check_phone_number_info("71234567", "RU", "7", "123 456 7---");
+  check_phone_number_info("77654321", "KZ", "7", "765 432 1- --");
+  check_phone_number_info("3", "", "3", "");
+  check_phone_number_info("37", "", "37", "");
+  check_phone_number_info("372", "EE", "372", "---- ----");
+  check_phone_number_info("42", "", "42", "");
+  check_phone_number_info("420", "CZ", "420", "--- --- ---");
+  check_phone_number_info("421", "SK", "421", "--- --- ---");
+  check_phone_number_info("422", "", "", "422");
+  check_phone_number_info("423", "LI", "423", "--- ----");
+  check_phone_number_info("424", "YL", "42", "4");
+  check_phone_number_info("4241234567890", "YL", "42", "41234567890");
+  check_phone_number_info("4", "", "4", "");
+  check_phone_number_info("49", "DE", "49", "---- -------");
+  check_phone_number_info("491", "DE", "49", "1--- -------");
+  check_phone_number_info("492", "DE", "49", "2--- -------");
+  check_phone_number_info("4915", "DE", "49", "15-- -------");
+  check_phone_number_info("4916", "DE", "49", "16- -------");
+  check_phone_number_info("4917", "DE", "49", "17- -------");
+  check_phone_number_info("4918", "DE", "49", "18-- -------");
+  check_phone_number_info("493", "DE", "49", "3--- -------");
+  check_phone_number_info("4936", "DE", "49", "36-- -------");
+  check_phone_number_info("49360", "DE", "49", "360- -------");
+  check_phone_number_info("493601", "DE", "49", "3601 -------");
+  check_phone_number_info("4936014", "DE", "49", "3601 4------");
+  check_phone_number_info("4936015", "DE", "49", "3601 5------");
+  check_phone_number_info("493601419", "DE", "49", "3601 419----");
+  check_phone_number_info("4936014198", "DE", "49", "3601 4198--");
+  check_phone_number_info("49360141980", "DE", "49", "3601 41980-");
+  check_phone_number_info("841234567890", "VN", "84", "1234567890");
+  check_phone_number_info("31", "NL", "31", "- -- -- -- --");
+  check_phone_number_info("318", "NL", "31", "8 -- -- -- --");
+  check_phone_number_info("319", "NL", "31", "9 -- -- -- --");
+  check_phone_number_info("3196", "NL", "31", "9 6- -- -- --");
+  check_phone_number_info("3197", "NL", "31", "97 ---- -----");
+  check_phone_number_info("3198", "NL", "31", "9 8- -- -- --");
+}