key_value_table_unittest.cc 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
  1. // Copyright 2020 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/sqlite_proto/key_value_table.h"
  5. #include "base/memory/scoped_refptr.h"
  6. #include "components/sqlite_proto/test_proto.pb.h"
  7. #include "sql/database.h"
  8. #include "testing/gmock/include/gmock/gmock.h"
  9. #include "testing/gtest/include/gtest/gtest.h"
  10. using ::testing::ElementsAre;
  11. using ::testing::Pair;
  12. using ::testing::UnorderedElementsAre;
  13. namespace sqlite_proto {
  14. namespace {
  15. MATCHER_P(EqualsProto,
  16. message,
  17. "Match a proto Message equal to the matcher's argument.") {
  18. std::string expected_serialized, actual_serialized;
  19. message.SerializeToString(&expected_serialized);
  20. arg.SerializeToString(&actual_serialized);
  21. return expected_serialized == actual_serialized;
  22. }
  23. bool CreateProtoTable(sql::Database* db) {
  24. const char kCreateProtoTableStatementTemplate[] =
  25. "CREATE TABLE my_table ( "
  26. "key TEXT, "
  27. "proto BLOB, "
  28. "PRIMARY KEY(key))";
  29. return db->Execute(kCreateProtoTableStatementTemplate);
  30. }
  31. } // namespace
  32. class KeyValueTableTest : public ::testing::Test {
  33. public:
  34. KeyValueTableTest() {
  35. CHECK(db_.OpenInMemory());
  36. CHECK(CreateProtoTable(&db_));
  37. }
  38. ~KeyValueTableTest() override = default;
  39. protected:
  40. sql::Database db_;
  41. KeyValueTable<TestProto> table_{"my_table"};
  42. };
  43. TEST_F(KeyValueTableTest, Empty) {
  44. std::map<std::string, TestProto> my_data;
  45. table_.GetAllData(&my_data, &db_);
  46. EXPECT_TRUE(my_data.empty());
  47. }
  48. TEST_F(KeyValueTableTest, PutAndGet) {
  49. TestProto element;
  50. element.set_value(1);
  51. table_.UpdateData("a", element, &db_);
  52. TestProto second_element;
  53. second_element.set_value(2);
  54. table_.UpdateData("b", second_element, &db_);
  55. std::map<std::string, TestProto> my_data;
  56. table_.GetAllData(&my_data, &db_);
  57. EXPECT_THAT(my_data,
  58. UnorderedElementsAre(Pair("a", EqualsProto(element)),
  59. Pair("b", EqualsProto(second_element))));
  60. }
  61. TEST_F(KeyValueTableTest, Update) {
  62. TestProto element;
  63. element.set_value(1);
  64. TestProto superseding_element;
  65. superseding_element.set_value(2);
  66. table_.UpdateData("a", element, &db_);
  67. table_.UpdateData("a", superseding_element, &db_);
  68. std::map<std::string, TestProto> my_data;
  69. table_.GetAllData(&my_data, &db_);
  70. EXPECT_THAT(my_data,
  71. ElementsAre(Pair("a", EqualsProto(superseding_element))));
  72. }
  73. TEST_F(KeyValueTableTest, DeleteNonexistent) {
  74. TestProto element;
  75. element.set_value(1);
  76. table_.UpdateData("a", element, &db_);
  77. // Deleting a nonexistent key should no-op.
  78. table_.DeleteData(std::vector<std::string>{"b"}, &db_);
  79. std::map<std::string, TestProto> my_data;
  80. table_.GetAllData(&my_data, &db_);
  81. EXPECT_THAT(my_data, ElementsAre(Pair("a", EqualsProto(element))));
  82. }
  83. TEST_F(KeyValueTableTest, Delete) {
  84. TestProto element;
  85. element.set_value(1);
  86. table_.UpdateData("a", element, &db_);
  87. table_.DeleteData(std::vector<std::string>{"a"}, &db_);
  88. std::map<std::string, TestProto> my_data;
  89. table_.GetAllData(&my_data, &db_);
  90. EXPECT_TRUE(my_data.empty());
  91. }
  92. TEST_F(KeyValueTableTest, DeleteAll) {
  93. TestProto element;
  94. element.set_value(1);
  95. table_.UpdateData("a", element, &db_);
  96. table_.UpdateData("b", element, &db_);
  97. table_.DeleteAllData(&db_);
  98. std::map<std::string, TestProto> my_data;
  99. table_.GetAllData(&my_data, &db_);
  100. EXPECT_TRUE(my_data.empty());
  101. }
  102. // Storing an element with a default proto value (zero byte size) should work;
  103. // this can be useful since a default value can have different semantics from a
  104. // missing value.
  105. TEST_F(KeyValueTableTest, PutGetDefaultValue) {
  106. TestProto element;
  107. table_.UpdateData("a", element, &db_);
  108. ASSERT_EQ(element.ByteSize(), 0);
  109. std::map<std::string, TestProto> my_data;
  110. table_.GetAllData(&my_data, &db_);
  111. EXPECT_THAT(my_data, ElementsAre(Pair("a", EqualsProto(element))));
  112. }
  113. } // namespace sqlite_proto