image_index_unittest.cc 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131
  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/image_index.h"
  5. #include <stddef.h>
  6. #include <numeric>
  7. #include <vector>
  8. #include "base/test/gtest_util.h"
  9. #include "components/zucchini/image_utils.h"
  10. #include "components/zucchini/test_disassembler.h"
  11. #include "testing/gtest/include/gtest/gtest.h"
  12. namespace zucchini {
  13. class ImageIndexTest : public testing::Test {
  14. protected:
  15. ImageIndexTest()
  16. : buffer_(20),
  17. image_index_(ConstBufferView(buffer_.data(), buffer_.size())) {
  18. std::iota(buffer_.begin(), buffer_.end(), 0);
  19. }
  20. void InitializeWithDefaultTestData() {
  21. TestDisassembler disasm({2, TypeTag(0), PoolTag(0)},
  22. {{1, 0}, {8, 1}, {10, 2}},
  23. {4, TypeTag(1), PoolTag(0)}, {{3, 3}},
  24. {3, TypeTag(2), PoolTag(1)}, {{12, 4}, {17, 5}});
  25. EXPECT_TRUE(image_index_.Initialize(&disasm));
  26. }
  27. std::vector<uint8_t> buffer_;
  28. ImageIndex image_index_;
  29. };
  30. TEST_F(ImageIndexTest, TypeAndPool) {
  31. TestDisassembler disasm({2, TypeTag(0), PoolTag(0)}, {},
  32. {4, TypeTag(1), PoolTag(0)}, {},
  33. {3, TypeTag(2), PoolTag(1)}, {});
  34. EXPECT_TRUE(image_index_.Initialize(&disasm));
  35. EXPECT_EQ(3U, image_index_.TypeCount());
  36. EXPECT_EQ(2U, image_index_.PoolCount());
  37. EXPECT_EQ(TypeTag(0), image_index_.refs(TypeTag(0)).type_tag());
  38. EXPECT_EQ(TypeTag(1), image_index_.refs(TypeTag(1)).type_tag());
  39. EXPECT_EQ(TypeTag(2), image_index_.refs(TypeTag(2)).type_tag());
  40. EXPECT_EQ(PoolTag(0), image_index_.refs(TypeTag(0)).pool_tag());
  41. EXPECT_EQ(PoolTag(0), image_index_.refs(TypeTag(1)).pool_tag());
  42. EXPECT_EQ(PoolTag(1), image_index_.refs(TypeTag(2)).pool_tag());
  43. }
  44. TEST_F(ImageIndexTest, InvalidInitialize1) {
  45. // Overlap within the same group.
  46. TestDisassembler disasm({2, TypeTag(0), PoolTag(0)}, {{1, 0}, {2, 0}},
  47. {4, TypeTag(1), PoolTag(0)}, {},
  48. {3, TypeTag(2), PoolTag(1)}, {});
  49. EXPECT_FALSE(image_index_.Initialize(&disasm));
  50. }
  51. TEST_F(ImageIndexTest, InvalidInitialize2) {
  52. // Overlap across different readers.
  53. TestDisassembler disasm({2, TypeTag(0), PoolTag(0)},
  54. {{1, 0}, {8, 1}, {10, 2}},
  55. {4, TypeTag(1), PoolTag(0)}, {{3, 3}},
  56. {3, TypeTag(2), PoolTag(1)}, {{11, 0}});
  57. EXPECT_FALSE(image_index_.Initialize(&disasm));
  58. }
  59. TEST_F(ImageIndexTest, LookupType) {
  60. InitializeWithDefaultTestData();
  61. std::vector<int> expected = {
  62. -1, // raw
  63. 0, 0, // ref 0
  64. 1, 1, 1, 1, // ref 1
  65. -1, // raw
  66. 0, 0, // ref 0
  67. 0, 0, // ref 0
  68. 2, 2, 2, // ref 2
  69. -1, -1, // raw
  70. 2, 2, 2, // ref 2
  71. };
  72. for (offset_t i = 0; i < image_index_.size(); ++i)
  73. EXPECT_EQ(TypeTag(expected[i]), image_index_.LookupType(i));
  74. }
  75. TEST_F(ImageIndexTest, IsToken) {
  76. InitializeWithDefaultTestData();
  77. std::vector<bool> expected = {
  78. true, // raw
  79. true, false, // ref 0
  80. true, false, false, false, // ref 1
  81. true, // raw
  82. true, false, // ref 0
  83. true, false, // ref 0
  84. true, false, false, // ref 2
  85. true, true, // raw
  86. true, false, false, // ref 2
  87. };
  88. for (offset_t i = 0; i < image_index_.size(); ++i)
  89. EXPECT_EQ(expected[i], image_index_.IsToken(i));
  90. }
  91. TEST_F(ImageIndexTest, IsReference) {
  92. InitializeWithDefaultTestData();
  93. std::vector<bool> expected = {
  94. false, // raw
  95. true, true, // ref 0
  96. true, true, true, true, // ref 1
  97. false, // raw
  98. true, true, // ref 0
  99. true, true, // ref 0
  100. true, true, true, // ref 2
  101. false, false, // raw
  102. true, true, true, // ref 2
  103. };
  104. for (offset_t i = 0; i < image_index_.size(); ++i)
  105. EXPECT_EQ(expected[i], image_index_.IsReference(i));
  106. }
  107. } // namespace zucchini