1 // Copyright 2023 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #ifndef BASE_CONTAINERS_MAP_UTIL_UNITTEST_CC_
6 #define BASE_CONTAINERS_MAP_UTIL_UNITTEST_CC_
7
8 #include "base/containers/map_util.h"
9
10 #include <memory>
11 #include <string>
12
13 #include "base/containers/flat_map.h"
14 #include "testing/gmock/include/gmock/gmock.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16
17 namespace base {
18
19 namespace {
20
21 using testing::AllOf;
22 using testing::Eq;
23 using testing::Pointee;
24
25 constexpr char kKey[] = "key";
26 constexpr char kValue[] = "value";
27 constexpr char kMissingKey[] = "missing_key";
28
29 using StringToStringMap = base::flat_map<std::string, std::string>;
30 using StringToStringPtrMap = base::flat_map<std::string, std::string*>;
31 using StringToStringUniquePtrMap =
32 base::flat_map<std::string, std::unique_ptr<std::string>>;
33
TEST(MapUtilTest,FindOrNull)34 TEST(MapUtilTest, FindOrNull) {
35 StringToStringMap mapping({{kKey, kValue}});
36
37 EXPECT_THAT(FindOrNull(mapping, kKey), Pointee(Eq(kValue)));
38 EXPECT_EQ(FindOrNull(mapping, kMissingKey), nullptr);
39 }
40
TEST(MapUtilTest,FindPtrOrNullForPointers)41 TEST(MapUtilTest, FindPtrOrNullForPointers) {
42 auto val = std::make_unique<std::string>(kValue);
43
44 StringToStringPtrMap mapping({{kKey, val.get()}});
45
46 EXPECT_THAT(FindPtrOrNull(mapping, kKey),
47 AllOf(Eq(val.get()), Pointee(Eq(kValue))));
48 EXPECT_EQ(FindPtrOrNull(mapping, kMissingKey), nullptr);
49 }
50
TEST(MapUtilTest,FindPtrOrNullForPointerLikeValues)51 TEST(MapUtilTest, FindPtrOrNullForPointerLikeValues) {
52 StringToStringUniquePtrMap mapping;
53 mapping.insert({kKey, std::make_unique<std::string>(kValue)});
54
55 EXPECT_THAT(FindPtrOrNull(mapping, kKey), Pointee(Eq(kValue)));
56 EXPECT_EQ(FindPtrOrNull(mapping, kMissingKey), nullptr);
57 }
58
59 } // namespace
60
61 } // namespace base
62
63 #endif // BASE_CONTAINERS_MAP_UTIL_UNITTEST_CC_
64