OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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 | |
5 #include "base/basictypes.h" | |
6 #include "sync/engine/store_timestamps_command.h" | |
7 #include "sync/internal_api/public/base/model_type.h" | |
8 #include "sync/protocol/sync.pb.h" | |
9 #include "sync/test/engine/syncer_command_test.h" | |
10 #include "testing/gtest/include/gtest/gtest.h" | |
11 | |
12 namespace syncer { | |
13 | |
14 namespace { | |
15 | |
16 // Adds a progress marker to |response| for the given field number and | |
17 // token. | |
18 void AddProgressMarkerForFieldNumber( | |
19 sync_pb::GetUpdatesResponse* response, | |
20 int field_number, const std::string& token) { | |
21 sync_pb::DataTypeProgressMarker* marker = | |
22 response->add_new_progress_marker(); | |
23 marker->set_data_type_id(field_number); | |
24 marker->set_token(token); | |
25 } | |
26 | |
27 // Adds a progress marker to |response| for the given model type and | |
28 // token. | |
29 void AddProgressMarkerForModelType( | |
30 sync_pb::GetUpdatesResponse* response, | |
31 ModelType model_type, const std::string& token) { | |
32 AddProgressMarkerForFieldNumber( | |
33 response, GetSpecificsFieldNumberFromModelType(model_type), token); | |
34 } | |
35 | |
36 class StoreTimestampsCommandTest : public SyncerCommandTest { | |
37 protected: | |
38 // Gets the directory's progress marker's token for the given model | |
39 // type. | |
40 std::string GetProgessMarkerToken(ModelType model_type) { | |
41 sync_pb::DataTypeProgressMarker progress_marker; | |
42 session()->context()->directory()->GetDownloadProgress( | |
43 model_type, &progress_marker); | |
44 EXPECT_EQ( | |
45 GetSpecificsFieldNumberFromModelType(model_type), | |
46 progress_marker.data_type_id()); | |
47 return progress_marker.token(); | |
48 } | |
49 }; | |
50 | |
51 // Builds a GetUpdatesResponse with some progress markers, including | |
52 // invalid ones. ProcessNewProgressMarkers() should return the model | |
53 // types for the valid progress markers and fill in the progress | |
54 // markers in the directory. | |
55 TEST_F(StoreTimestampsCommandTest, ProcessNewProgressMarkers) { | |
56 sync_pb::GetUpdatesResponse response; | |
57 AddProgressMarkerForModelType(&response, BOOKMARKS, "token1"); | |
58 AddProgressMarkerForModelType(&response, | |
59 HISTORY_DELETE_DIRECTIVES, "token2"); | |
60 AddProgressMarkerForFieldNumber(&response, -1, "bad token"); | |
61 | |
62 ModelTypeSet forward_progress_types = | |
63 ProcessNewProgressMarkers( | |
64 response, session()->context()->directory()); | |
65 | |
66 EXPECT_TRUE( | |
67 forward_progress_types.Equals( | |
68 ModelTypeSet(BOOKMARKS, HISTORY_DELETE_DIRECTIVES))); | |
69 | |
70 EXPECT_EQ("token1", GetProgessMarkerToken(BOOKMARKS)); | |
71 EXPECT_EQ("token2", GetProgessMarkerToken(HISTORY_DELETE_DIRECTIVES)); | |
72 | |
73 ModelTypeSet non_forward_progress_types = | |
74 Difference(ProtocolTypes(), forward_progress_types); | |
75 for (ModelTypeSet::Iterator it = non_forward_progress_types.First(); | |
76 it.Good(); it.Inc()) { | |
77 EXPECT_TRUE(GetProgessMarkerToken(it.Get()).empty()); | |
78 } | |
79 } | |
80 | |
81 } // namespace | |
82 | |
83 } // namespace syncer | |
OLD | NEW |