reference_set_unittest.cc 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. // Copyright 2017 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/zucchini/reference_set.h"
  5. #include <vector>
  6. #include "components/zucchini/image_utils.h"
  7. #include "components/zucchini/target_pool.h"
  8. #include "components/zucchini/test_reference_reader.h"
  9. #include "testing/gtest/include/gtest/gtest.h"
  10. namespace zucchini {
  11. namespace {
  12. constexpr offset_t kWidth = 2U;
  13. } // namespace
  14. class ReferenceSetTest : public testing::Test {
  15. protected:
  16. // For simplicity, |target_pool_| has no type info (not needed here).
  17. TargetPool target_pool_ = TargetPool{{0, 2, 3, 5}};
  18. ReferenceSet reference_set_ =
  19. ReferenceSet{{kWidth, TypeTag(0), PoolTag(0)}, target_pool_};
  20. };
  21. TEST_F(ReferenceSetTest, InitReferencesFromReader) {
  22. EXPECT_EQ(std::vector<Reference>(), reference_set_.references());
  23. EXPECT_EQ(0U, reference_set_.size());
  24. std::vector<Reference> references = {{10, 0}, {12, 2}, {14, 5}};
  25. reference_set_.InitReferences(TestReferenceReader(references));
  26. EXPECT_EQ(references, reference_set_.references());
  27. }
  28. TEST_F(ReferenceSetTest, At) {
  29. reference_set_.InitReferences({{10, 0}, {12, 2}, {15, 5}});
  30. // Each references has kWidth = 2, so check all bytes covered.
  31. EXPECT_EQ(Reference({10, 0}), reference_set_.at(10));
  32. EXPECT_EQ(Reference({10, 0}), reference_set_.at(11));
  33. EXPECT_EQ(Reference({12, 2}), reference_set_.at(12));
  34. EXPECT_EQ(Reference({12, 2}), reference_set_.at(13));
  35. EXPECT_EQ(Reference({15, 5}), reference_set_.at(15));
  36. EXPECT_EQ(Reference({15, 5}), reference_set_.at(16));
  37. }
  38. } // namespace zucchini