utility_test.cc 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376
  1. // Copyright 2017 The Abseil Authors.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // https://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #include "absl/utility/utility.h"
  15. #include <sstream>
  16. #include <string>
  17. #include <tuple>
  18. #include <type_traits>
  19. #include <vector>
  20. #include "gmock/gmock.h"
  21. #include "gtest/gtest.h"
  22. #include "absl/base/attributes.h"
  23. #include "absl/memory/memory.h"
  24. #include "absl/strings/str_cat.h"
  25. namespace {
  26. #ifdef _MSC_VER
  27. // Warnings for unused variables in this test are false positives. On other
  28. // platforms, they are suppressed by ABSL_ATTRIBUTE_UNUSED, but that doesn't
  29. // work on MSVC.
  30. // Both the unused variables and the name length warnings are due to calls
  31. // to absl::make_index_sequence with very large values, creating very long type
  32. // names. The resulting warnings are so long they make build output unreadable.
  33. #pragma warning( push )
  34. #pragma warning( disable : 4503 ) // decorated name length exceeded
  35. #pragma warning( disable : 4101 ) // unreferenced local variable
  36. #endif // _MSC_VER
  37. using ::testing::ElementsAre;
  38. using ::testing::Pointee;
  39. using ::testing::StaticAssertTypeEq;
  40. TEST(IntegerSequenceTest, ValueType) {
  41. StaticAssertTypeEq<int, absl::integer_sequence<int>::value_type>();
  42. StaticAssertTypeEq<char, absl::integer_sequence<char>::value_type>();
  43. }
  44. TEST(IntegerSequenceTest, Size) {
  45. EXPECT_EQ(0, (absl::integer_sequence<int>::size()));
  46. EXPECT_EQ(1, (absl::integer_sequence<int, 0>::size()));
  47. EXPECT_EQ(1, (absl::integer_sequence<int, 1>::size()));
  48. EXPECT_EQ(2, (absl::integer_sequence<int, 1, 2>::size()));
  49. EXPECT_EQ(3, (absl::integer_sequence<int, 0, 1, 2>::size()));
  50. EXPECT_EQ(3, (absl::integer_sequence<int, -123, 123, 456>::size()));
  51. constexpr size_t sz = absl::integer_sequence<int, 0, 1>::size();
  52. EXPECT_EQ(2, sz);
  53. }
  54. TEST(IntegerSequenceTest, MakeIndexSequence) {
  55. StaticAssertTypeEq<absl::index_sequence<>, absl::make_index_sequence<0>>();
  56. StaticAssertTypeEq<absl::index_sequence<0>, absl::make_index_sequence<1>>();
  57. StaticAssertTypeEq<absl::index_sequence<0, 1>,
  58. absl::make_index_sequence<2>>();
  59. StaticAssertTypeEq<absl::index_sequence<0, 1, 2>,
  60. absl::make_index_sequence<3>>();
  61. }
  62. TEST(IntegerSequenceTest, MakeIntegerSequence) {
  63. StaticAssertTypeEq<absl::integer_sequence<int>,
  64. absl::make_integer_sequence<int, 0>>();
  65. StaticAssertTypeEq<absl::integer_sequence<int, 0>,
  66. absl::make_integer_sequence<int, 1>>();
  67. StaticAssertTypeEq<absl::integer_sequence<int, 0, 1>,
  68. absl::make_integer_sequence<int, 2>>();
  69. StaticAssertTypeEq<absl::integer_sequence<int, 0, 1, 2>,
  70. absl::make_integer_sequence<int, 3>>();
  71. }
  72. template <typename... Ts>
  73. class Counter {};
  74. template <size_t... Is>
  75. void CountAll(absl::index_sequence<Is...>) {
  76. // We only need an alias here, but instantiate a variable to silence warnings
  77. // for unused typedefs in some compilers.
  78. ABSL_ATTRIBUTE_UNUSED Counter<absl::make_index_sequence<Is>...> seq;
  79. }
  80. // This test verifies that absl::make_index_sequence can handle large arguments
  81. // without blowing up template instantiation stack, going OOM or taking forever
  82. // to compile (there is hard 15 minutes limit imposed by forge).
  83. TEST(IntegerSequenceTest, MakeIndexSequencePerformance) {
  84. // O(log N) template instantiations.
  85. // We only need an alias here, but instantiate a variable to silence warnings
  86. // for unused typedefs in some compilers.
  87. ABSL_ATTRIBUTE_UNUSED absl::make_index_sequence<(1 << 16) - 1> seq;
  88. // O(N) template instantiations.
  89. CountAll(absl::make_index_sequence<(1 << 8) - 1>());
  90. }
  91. template <typename F, typename Tup, size_t... Is>
  92. auto ApplyFromTupleImpl(F f, const Tup& tup, absl::index_sequence<Is...>)
  93. -> decltype(f(std::get<Is>(tup)...)) {
  94. return f(std::get<Is>(tup)...);
  95. }
  96. template <typename Tup>
  97. using TupIdxSeq = absl::make_index_sequence<std::tuple_size<Tup>::value>;
  98. template <typename F, typename Tup>
  99. auto ApplyFromTuple(F f, const Tup& tup)
  100. -> decltype(ApplyFromTupleImpl(f, tup, TupIdxSeq<Tup>{})) {
  101. return ApplyFromTupleImpl(f, tup, TupIdxSeq<Tup>{});
  102. }
  103. template <typename T>
  104. std::string Fmt(const T& x) {
  105. std::ostringstream os;
  106. os << x;
  107. return os.str();
  108. }
  109. struct PoorStrCat {
  110. template <typename... Args>
  111. std::string operator()(const Args&... args) const {
  112. std::string r;
  113. for (const auto& e : {Fmt(args)...}) r += e;
  114. return r;
  115. }
  116. };
  117. template <typename Tup, size_t... Is>
  118. std::vector<std::string> TupStringVecImpl(const Tup& tup,
  119. absl::index_sequence<Is...>) {
  120. return {Fmt(std::get<Is>(tup))...};
  121. }
  122. template <typename... Ts>
  123. std::vector<std::string> TupStringVec(const std::tuple<Ts...>& tup) {
  124. return TupStringVecImpl(tup, absl::index_sequence_for<Ts...>());
  125. }
  126. TEST(MakeIndexSequenceTest, ApplyFromTupleExample) {
  127. PoorStrCat f{};
  128. EXPECT_EQ("12abc3.14", f(12, "abc", 3.14));
  129. EXPECT_EQ("12abc3.14", ApplyFromTuple(f, std::make_tuple(12, "abc", 3.14)));
  130. }
  131. TEST(IndexSequenceForTest, Basic) {
  132. StaticAssertTypeEq<absl::index_sequence<>, absl::index_sequence_for<>>();
  133. StaticAssertTypeEq<absl::index_sequence<0>, absl::index_sequence_for<int>>();
  134. StaticAssertTypeEq<absl::index_sequence<0, 1, 2, 3>,
  135. absl::index_sequence_for<int, void, char, int>>();
  136. }
  137. TEST(IndexSequenceForTest, Example) {
  138. EXPECT_THAT(TupStringVec(std::make_tuple(12, "abc", 3.14)),
  139. ElementsAre("12", "abc", "3.14"));
  140. }
  141. int Function(int a, int b) { return a - b; }
  142. int Sink(std::unique_ptr<int> p) { return *p; }
  143. std::unique_ptr<int> Factory(int n) { return absl::make_unique<int>(n); }
  144. void NoOp() {}
  145. struct ConstFunctor {
  146. int operator()(int a, int b) const { return a - b; }
  147. };
  148. struct MutableFunctor {
  149. int operator()(int a, int b) { return a - b; }
  150. };
  151. struct EphemeralFunctor {
  152. EphemeralFunctor() {}
  153. EphemeralFunctor(const EphemeralFunctor&) {}
  154. EphemeralFunctor(EphemeralFunctor&&) {}
  155. int operator()(int a, int b) && { return a - b; }
  156. };
  157. struct OverloadedFunctor {
  158. OverloadedFunctor() {}
  159. OverloadedFunctor(const OverloadedFunctor&) {}
  160. OverloadedFunctor(OverloadedFunctor&&) {}
  161. template <typename... Args>
  162. std::string operator()(const Args&... args) & {
  163. return absl::StrCat("&", args...);
  164. }
  165. template <typename... Args>
  166. std::string operator()(const Args&... args) const& {
  167. return absl::StrCat("const&", args...);
  168. }
  169. template <typename... Args>
  170. std::string operator()(const Args&... args) && {
  171. return absl::StrCat("&&", args...);
  172. }
  173. };
  174. struct Class {
  175. int Method(int a, int b) { return a - b; }
  176. int ConstMethod(int a, int b) const { return a - b; }
  177. int member;
  178. };
  179. struct FlipFlop {
  180. int ConstMethod() const { return member; }
  181. FlipFlop operator*() const { return {-member}; }
  182. int member;
  183. };
  184. TEST(ApplyTest, Function) {
  185. EXPECT_EQ(1, absl::apply(Function, std::make_tuple(3, 2)));
  186. EXPECT_EQ(1, absl::apply(&Function, std::make_tuple(3, 2)));
  187. }
  188. TEST(ApplyTest, NonCopyableArgument) {
  189. EXPECT_EQ(42, absl::apply(Sink, std::make_tuple(absl::make_unique<int>(42))));
  190. }
  191. TEST(ApplyTest, NonCopyableResult) {
  192. EXPECT_THAT(absl::apply(Factory, std::make_tuple(42)),
  193. ::testing::Pointee(42));
  194. }
  195. TEST(ApplyTest, VoidResult) { absl::apply(NoOp, std::tuple<>()); }
  196. TEST(ApplyTest, ConstFunctor) {
  197. EXPECT_EQ(1, absl::apply(ConstFunctor(), std::make_tuple(3, 2)));
  198. }
  199. TEST(ApplyTest, MutableFunctor) {
  200. MutableFunctor f;
  201. EXPECT_EQ(1, absl::apply(f, std::make_tuple(3, 2)));
  202. EXPECT_EQ(1, absl::apply(MutableFunctor(), std::make_tuple(3, 2)));
  203. }
  204. TEST(ApplyTest, EphemeralFunctor) {
  205. EphemeralFunctor f;
  206. EXPECT_EQ(1, absl::apply(std::move(f), std::make_tuple(3, 2)));
  207. EXPECT_EQ(1, absl::apply(EphemeralFunctor(), std::make_tuple(3, 2)));
  208. }
  209. TEST(ApplyTest, OverloadedFunctor) {
  210. OverloadedFunctor f;
  211. const OverloadedFunctor& cf = f;
  212. EXPECT_EQ("&", absl::apply(f, std::tuple<>{}));
  213. EXPECT_EQ("& 42", absl::apply(f, std::make_tuple(" 42")));
  214. EXPECT_EQ("const&", absl::apply(cf, std::tuple<>{}));
  215. EXPECT_EQ("const& 42", absl::apply(cf, std::make_tuple(" 42")));
  216. EXPECT_EQ("&&", absl::apply(std::move(f), std::tuple<>{}));
  217. OverloadedFunctor f2;
  218. EXPECT_EQ("&& 42", absl::apply(std::move(f2), std::make_tuple(" 42")));
  219. }
  220. TEST(ApplyTest, ReferenceWrapper) {
  221. ConstFunctor cf;
  222. MutableFunctor mf;
  223. EXPECT_EQ(1, absl::apply(std::cref(cf), std::make_tuple(3, 2)));
  224. EXPECT_EQ(1, absl::apply(std::ref(cf), std::make_tuple(3, 2)));
  225. EXPECT_EQ(1, absl::apply(std::ref(mf), std::make_tuple(3, 2)));
  226. }
  227. TEST(ApplyTest, MemberFunction) {
  228. std::unique_ptr<Class> p(new Class);
  229. std::unique_ptr<const Class> cp(new Class);
  230. EXPECT_EQ(
  231. 1, absl::apply(&Class::Method,
  232. std::tuple<std::unique_ptr<Class>&, int, int>(p, 3, 2)));
  233. EXPECT_EQ(1, absl::apply(&Class::Method,
  234. std::tuple<Class*, int, int>(p.get(), 3, 2)));
  235. EXPECT_EQ(
  236. 1, absl::apply(&Class::Method, std::tuple<Class&, int, int>(*p, 3, 2)));
  237. EXPECT_EQ(
  238. 1, absl::apply(&Class::ConstMethod,
  239. std::tuple<std::unique_ptr<Class>&, int, int>(p, 3, 2)));
  240. EXPECT_EQ(1, absl::apply(&Class::ConstMethod,
  241. std::tuple<Class*, int, int>(p.get(), 3, 2)));
  242. EXPECT_EQ(1, absl::apply(&Class::ConstMethod,
  243. std::tuple<Class&, int, int>(*p, 3, 2)));
  244. EXPECT_EQ(1, absl::apply(&Class::ConstMethod,
  245. std::tuple<std::unique_ptr<const Class>&, int, int>(
  246. cp, 3, 2)));
  247. EXPECT_EQ(1, absl::apply(&Class::ConstMethod,
  248. std::tuple<const Class*, int, int>(cp.get(), 3, 2)));
  249. EXPECT_EQ(1, absl::apply(&Class::ConstMethod,
  250. std::tuple<const Class&, int, int>(*cp, 3, 2)));
  251. EXPECT_EQ(1, absl::apply(&Class::Method,
  252. std::make_tuple(absl::make_unique<Class>(), 3, 2)));
  253. EXPECT_EQ(1, absl::apply(&Class::ConstMethod,
  254. std::make_tuple(absl::make_unique<Class>(), 3, 2)));
  255. EXPECT_EQ(
  256. 1, absl::apply(&Class::ConstMethod,
  257. std::make_tuple(absl::make_unique<const Class>(), 3, 2)));
  258. }
  259. TEST(ApplyTest, DataMember) {
  260. std::unique_ptr<Class> p(new Class{42});
  261. std::unique_ptr<const Class> cp(new Class{42});
  262. EXPECT_EQ(
  263. 42, absl::apply(&Class::member, std::tuple<std::unique_ptr<Class>&>(p)));
  264. EXPECT_EQ(42, absl::apply(&Class::member, std::tuple<Class&>(*p)));
  265. EXPECT_EQ(42, absl::apply(&Class::member, std::tuple<Class*>(p.get())));
  266. absl::apply(&Class::member, std::tuple<std::unique_ptr<Class>&>(p)) = 42;
  267. absl::apply(&Class::member, std::tuple<Class*>(p.get())) = 42;
  268. absl::apply(&Class::member, std::tuple<Class&>(*p)) = 42;
  269. EXPECT_EQ(42, absl::apply(&Class::member,
  270. std::tuple<std::unique_ptr<const Class>&>(cp)));
  271. EXPECT_EQ(42, absl::apply(&Class::member, std::tuple<const Class&>(*cp)));
  272. EXPECT_EQ(42,
  273. absl::apply(&Class::member, std::tuple<const Class*>(cp.get())));
  274. }
  275. TEST(ApplyTest, FlipFlop) {
  276. FlipFlop obj = {42};
  277. // This call could resolve to (obj.*&FlipFlop::ConstMethod)() or
  278. // ((*obj).*&FlipFlop::ConstMethod)(). We verify that it's the former.
  279. EXPECT_EQ(42, absl::apply(&FlipFlop::ConstMethod, std::make_tuple(obj)));
  280. EXPECT_EQ(42, absl::apply(&FlipFlop::member, std::make_tuple(obj)));
  281. }
  282. TEST(ExchangeTest, MoveOnly) {
  283. auto a = Factory(1);
  284. EXPECT_EQ(1, *a);
  285. auto b = absl::exchange(a, Factory(2));
  286. EXPECT_EQ(2, *a);
  287. EXPECT_EQ(1, *b);
  288. }
  289. TEST(MakeFromTupleTest, String) {
  290. EXPECT_EQ(
  291. absl::make_from_tuple<std::string>(std::make_tuple("hello world", 5)),
  292. "hello");
  293. }
  294. TEST(MakeFromTupleTest, MoveOnlyParameter) {
  295. struct S {
  296. S(std::unique_ptr<int> n, std::unique_ptr<int> m) : value(*n + *m) {}
  297. int value = 0;
  298. };
  299. auto tup =
  300. std::make_tuple(absl::make_unique<int>(3), absl::make_unique<int>(4));
  301. auto s = absl::make_from_tuple<S>(std::move(tup));
  302. EXPECT_EQ(s.value, 7);
  303. }
  304. TEST(MakeFromTupleTest, NoParameters) {
  305. struct S {
  306. S() : value(1) {}
  307. int value = 2;
  308. };
  309. EXPECT_EQ(absl::make_from_tuple<S>(std::make_tuple()).value, 1);
  310. }
  311. TEST(MakeFromTupleTest, Pair) {
  312. EXPECT_EQ(
  313. (absl::make_from_tuple<std::pair<bool, int>>(std::make_tuple(true, 17))),
  314. std::make_pair(true, 17));
  315. }
  316. } // namespace