Home | History | Annotate | Download | only in engine 
      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
     84