Fix copyright lines

[folly.git] / folly / test / DemangleTest.cpp

diff --git a/folly/test/DemangleTest.cpp b/folly/test/DemangleTest.cpp
index eab9cff7cbbe6f0a37a20abde2fb341670ea1a67..26978020cb40a33dce5ac819f3ed25e2449a0340 100644 (file)
--- a/folly/test/DemangleTest.cpp
+++ b/folly/test/DemangleTest.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright 2015 Facebook, Inc.
+ * Copyright 2015-present Facebook, Inc.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -16,15 +16,14 @@
 
 #include <folly/Demangle.h>
 
-#include <gflags/gflags.h>
-#include <gtest/gtest.h>
+#include <folly/portability/GTest.h>
 
 using folly::demangle;
 
 namespace folly_test {
 struct ThisIsAVeryLongStructureName {
 };
-}  // namespace folly_test
+} // namespace folly_test
 
 #if FOLLY_HAVE_CPLUS_DEMANGLE_V3_CALLBACK
 TEST(Demangle, demangle) {
@@ -46,10 +45,62 @@ TEST(Demangle, demangle) {
     EXPECT_STREQ("folly_test", buf);
   }
 }
-#endif
 
-int main(int argc, char *argv[]) {
-  testing::InitGoogleTest(&argc, argv);
-  gflags::ParseCommandLineFlags(&argc, &argv, true);
-  return RUN_ALL_TESTS();
+#if defined(FOLLY_DEMANGLE_MAX_SYMBOL_SIZE)
+namespace {
+
+template <int I, class T1, class T2>
+struct Node {};
+
+template <int N, int I = 1>
+struct LongSymbol {
+  using arg1 = typename LongSymbol<N / 2, 2 * I>::type;
+  using arg2 = typename LongSymbol<N / 2, 2 * I + 1>::type;
+  using type = Node<I, arg1, arg2>;
+};
+
+template <int I>
+struct LongSymbol<0, I> {
+  using type = void;
+};
+
+} // namespace
+
+TEST(Demangle, LongSymbolFallback) {
+  // The symbol must be at least FOLLY_DEMANGLE_MAX_SYMBOL_SIZE long.
+  using Symbol = LongSymbol<FOLLY_DEMANGLE_MAX_SYMBOL_SIZE>::type;
+  auto name = typeid(Symbol).name();
+
+  EXPECT_STREQ(name, demangle(name).c_str());
+
+  char buf[16];
+  char expected[16];
+  folly::demangle(name, buf, 16);
+  folly::strlcpy(expected, name, 16);
+  EXPECT_STREQ(expected, buf);
+}
+#endif // defined(FOLLY_DEMANGLE_MAX_SYMBOL_SIZE)
+
+#endif // FOLLY_HAVE_CPLUS_DEMANGLE_V3_CALLBACK
+
+TEST(Demangle, strlcpy) {
+  char buf[6];
+
+  EXPECT_EQ(3, folly::strlcpy(buf, "abc", 6));
+  EXPECT_EQ('\0', buf[3]);
+  EXPECT_EQ("abc", std::string(buf));
+
+  EXPECT_EQ(7, folly::strlcpy(buf, "abcdefg", 3));
+  EXPECT_EQ('\0', buf[2]);
+  EXPECT_EQ("ab", std::string(buf));
+
+  const char* big_string = "abcdefghijklmnop";
+
+  EXPECT_EQ(strlen(big_string), folly::strlcpy(buf, big_string, sizeof(buf)));
+  EXPECT_EQ('\0', buf[5]);
+  EXPECT_EQ("abcde", std::string(buf));
+
+  buf[0] = 'z';
+  EXPECT_EQ(strlen(big_string), folly::strlcpy(buf, big_string, 0));
+  EXPECT_EQ('z', buf[0]);  // unchanged, size = 0
 }