OLD | NEW |
---|---|
1 // Copyright 2017 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/history/core/browser/typed_url_sync_metadata_database.h" | 5 #include "components/history/core/browser/typed_url_sync_metadata_database.h" |
6 | 6 |
7 #include "base/big_endian.h" | |
7 #include "base/files/file_path.h" | 8 #include "base/files/file_path.h" |
8 #include "base/files/scoped_temp_dir.h" | 9 #include "base/files/scoped_temp_dir.h" |
10 #include "components/history/core/browser/url_row.h" | |
9 #include "components/sync/protocol/model_type_state.pb.h" | 11 #include "components/sync/protocol/model_type_state.pb.h" |
10 #include "sql/statement.h" | 12 #include "sql/statement.h" |
11 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
12 | 14 |
13 using sync_pb::EntityMetadata; | 15 using sync_pb::EntityMetadata; |
14 using sync_pb::ModelTypeState; | 16 using sync_pb::ModelTypeState; |
15 using syncer::EntityMetadataMap; | 17 using syncer::EntityMetadataMap; |
16 using syncer::MetadataBatch; | 18 using syncer::MetadataBatch; |
17 | 19 |
18 namespace history { | 20 namespace history { |
19 | 21 |
22 namespace { | |
23 | |
24 std::string IntToStorageKey(int i) { | |
25 std::string storage_key(8, 0); | |
26 base::WriteBigEndian(&storage_key[0], static_cast<URLID>(i)); | |
pavely
2017/06/01 20:49:18
nit: I think you can explicitly specify which vari
Gang Wu
2017/06/02 00:00:24
Done.
| |
27 return storage_key; | |
28 } | |
29 | |
30 } // namespace | |
31 | |
20 class TypedURLSyncMetadataDatabaseTest : public testing::Test, | 32 class TypedURLSyncMetadataDatabaseTest : public testing::Test, |
21 public TypedURLSyncMetadataDatabase { | 33 public TypedURLSyncMetadataDatabase { |
22 public: | 34 public: |
23 TypedURLSyncMetadataDatabaseTest() {} | 35 TypedURLSyncMetadataDatabaseTest() {} |
24 ~TypedURLSyncMetadataDatabaseTest() override {} | 36 ~TypedURLSyncMetadataDatabaseTest() override {} |
25 | 37 |
26 protected: | 38 protected: |
27 sql::Connection& GetDB() override { return db_; } | 39 sql::Connection& GetDB() override { return db_; } |
28 | 40 |
29 void SetUp() override { | 41 void SetUp() override { |
(...skipping 23 matching lines...) Expand all Loading... | |
53 TEST_F(TypedURLSyncMetadataDatabaseTest, TypedURLNoMetadata) { | 65 TEST_F(TypedURLSyncMetadataDatabaseTest, TypedURLNoMetadata) { |
54 MetadataBatch metadata_batch; | 66 MetadataBatch metadata_batch; |
55 EXPECT_TRUE(GetAllSyncMetadata(&metadata_batch)); | 67 EXPECT_TRUE(GetAllSyncMetadata(&metadata_batch)); |
56 EXPECT_EQ(0u, metadata_batch.TakeAllMetadata().size()); | 68 EXPECT_EQ(0u, metadata_batch.TakeAllMetadata().size()); |
57 EXPECT_EQ(ModelTypeState().SerializeAsString(), | 69 EXPECT_EQ(ModelTypeState().SerializeAsString(), |
58 metadata_batch.GetModelTypeState().SerializeAsString()); | 70 metadata_batch.GetModelTypeState().SerializeAsString()); |
59 } | 71 } |
60 | 72 |
61 TEST_F(TypedURLSyncMetadataDatabaseTest, TypedURLGetAllSyncMetadata) { | 73 TEST_F(TypedURLSyncMetadataDatabaseTest, TypedURLGetAllSyncMetadata) { |
62 EntityMetadata metadata; | 74 EntityMetadata metadata; |
63 std::string storage_key = "1"; | 75 std::string storage_key = IntToStorageKey(0); |
64 std::string storage_key2 = "2"; | 76 std::string storage_key2 = IntToStorageKey(1); |
65 metadata.set_sequence_number(1); | 77 metadata.set_sequence_number(1); |
66 | 78 |
67 EXPECT_TRUE(UpdateSyncMetadata(syncer::TYPED_URLS, storage_key, metadata)); | 79 EXPECT_TRUE(UpdateSyncMetadata(syncer::TYPED_URLS, storage_key, metadata)); |
68 | 80 |
69 ModelTypeState model_type_state; | 81 ModelTypeState model_type_state; |
70 model_type_state.set_initial_sync_done(true); | 82 model_type_state.set_initial_sync_done(true); |
71 | 83 |
72 EXPECT_TRUE(UpdateModelTypeState(syncer::TYPED_URLS, model_type_state)); | 84 EXPECT_TRUE(UpdateModelTypeState(syncer::TYPED_URLS, model_type_state)); |
73 | 85 |
74 metadata.set_sequence_number(2); | 86 metadata.set_sequence_number(2); |
(...skipping 14 matching lines...) Expand all Loading... | |
89 model_type_state.set_initial_sync_done(false); | 101 model_type_state.set_initial_sync_done(false); |
90 EXPECT_TRUE(UpdateModelTypeState(syncer::TYPED_URLS, model_type_state)); | 102 EXPECT_TRUE(UpdateModelTypeState(syncer::TYPED_URLS, model_type_state)); |
91 | 103 |
92 EXPECT_TRUE(GetAllSyncMetadata(&metadata_batch)); | 104 EXPECT_TRUE(GetAllSyncMetadata(&metadata_batch)); |
93 EXPECT_FALSE(metadata_batch.GetModelTypeState().initial_sync_done()); | 105 EXPECT_FALSE(metadata_batch.GetModelTypeState().initial_sync_done()); |
94 } | 106 } |
95 | 107 |
96 TEST_F(TypedURLSyncMetadataDatabaseTest, TypedURLWriteThenDeleteSyncMetadata) { | 108 TEST_F(TypedURLSyncMetadataDatabaseTest, TypedURLWriteThenDeleteSyncMetadata) { |
97 EntityMetadata metadata; | 109 EntityMetadata metadata; |
98 MetadataBatch metadata_batch; | 110 MetadataBatch metadata_batch; |
99 std::string storage_key = "1"; | 111 std::string storage_key = IntToStorageKey(1); |
100 ModelTypeState model_type_state; | 112 ModelTypeState model_type_state; |
101 | 113 |
102 model_type_state.set_initial_sync_done(true); | 114 model_type_state.set_initial_sync_done(true); |
103 | 115 |
104 metadata.set_client_tag_hash("client_hash"); | 116 metadata.set_client_tag_hash("client_hash"); |
105 | 117 |
106 // Write the data into the store. | 118 // Write the data into the store. |
107 EXPECT_TRUE(UpdateSyncMetadata(syncer::TYPED_URLS, storage_key, metadata)); | 119 EXPECT_TRUE(UpdateSyncMetadata(syncer::TYPED_URLS, storage_key, metadata)); |
108 EXPECT_TRUE(UpdateModelTypeState(syncer::TYPED_URLS, model_type_state)); | 120 EXPECT_TRUE(UpdateModelTypeState(syncer::TYPED_URLS, model_type_state)); |
109 // Delete the data we just wrote. | 121 // Delete the data we just wrote. |
(...skipping 24 matching lines...) Expand all Loading... | |
134 } | 146 } |
135 | 147 |
136 TEST_F(TypedURLSyncMetadataDatabaseTest, TypedURLCorruptModelTypeState) { | 148 TEST_F(TypedURLSyncMetadataDatabaseTest, TypedURLCorruptModelTypeState) { |
137 MetadataBatch metadata_batch; | 149 MetadataBatch metadata_batch; |
138 GetMetaTable().SetValue("typed_url_model_type_state", "unparseable"); | 150 GetMetaTable().SetValue("typed_url_model_type_state", "unparseable"); |
139 | 151 |
140 EXPECT_FALSE(GetAllSyncMetadata(&metadata_batch)); | 152 EXPECT_FALSE(GetAllSyncMetadata(&metadata_batch)); |
141 } | 153 } |
142 | 154 |
143 } // namespace history | 155 } // namespace history |
OLD | NEW |