| // Copyright 2020 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <string> |
| |
| #include "find_by_first.h" |
| #include "test_platform.h" |
| |
| namespace v8_crdtp { |
| // ============================================================================= |
| // FindByFirst - Efficient retrieval from a sorted vector. |
| // ============================================================================= |
| TEST(FindByFirst, SpanBySpan) { |
| std::vector<std::pair<span<uint8_t>, span<uint8_t>>> sorted_span_by_span = { |
| {SpanFrom("foo1"), SpanFrom("bar1")}, |
| {SpanFrom("foo2"), SpanFrom("bar2")}, |
| {SpanFrom("foo3"), SpanFrom("bar3")}, |
| }; |
| { |
| auto result = FindByFirst(sorted_span_by_span, SpanFrom("foo1"), |
| SpanFrom("not_found")); |
| EXPECT_EQ("bar1", std::string(result.begin(), result.end())); |
| } |
| { |
| auto result = FindByFirst(sorted_span_by_span, SpanFrom("foo3"), |
| SpanFrom("not_found")); |
| EXPECT_EQ("bar3", std::string(result.begin(), result.end())); |
| } |
| { |
| auto result = FindByFirst(sorted_span_by_span, SpanFrom("baz"), |
| SpanFrom("not_found")); |
| EXPECT_EQ("not_found", std::string(result.begin(), result.end())); |
| } |
| } |
| |
| namespace { |
| class TestObject { |
| public: |
| explicit TestObject(const std::string& message) : message_(message) {} |
| |
| const std::string& message() const { return message_; } |
| |
| private: |
| std::string message_; |
| }; |
| } // namespace |
| |
| TEST(FindByFirst, ObjectBySpan) { |
| std::vector<std::pair<span<uint8_t>, std::unique_ptr<TestObject>>> |
| sorted_object_by_span; |
| sorted_object_by_span.push_back( |
| std::make_pair(SpanFrom("foo1"), std::make_unique<TestObject>("bar1"))); |
| sorted_object_by_span.push_back( |
| std::make_pair(SpanFrom("foo2"), std::make_unique<TestObject>("bar2"))); |
| sorted_object_by_span.push_back( |
| std::make_pair(SpanFrom("foo3"), std::make_unique<TestObject>("bar3"))); |
| { |
| TestObject* result = |
| FindByFirst<TestObject>(sorted_object_by_span, SpanFrom("foo1")); |
| ASSERT_TRUE(result); |
| ASSERT_EQ("bar1", result->message()); |
| } |
| { |
| TestObject* result = |
| FindByFirst<TestObject>(sorted_object_by_span, SpanFrom("foo3")); |
| ASSERT_TRUE(result); |
| ASSERT_EQ("bar3", result->message()); |
| } |
| { |
| TestObject* result = |
| FindByFirst<TestObject>(sorted_object_by_span, SpanFrom("baz")); |
| ASSERT_FALSE(result); |
| } |
| } |
| } // namespace v8_crdtp |