[Sync] Add support for performing a GetKey on startup.

sync/test/engine/mock_connection_manager.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 //
 // Mock ServerConnectionManager class for use in client regression tests.
 
 #include "sync/test/engine/mock_connection_manager.h"
 
 #include <map>
 
 #include "base/location.h"
 #include "base/stringprintf.h"
 #include "sync/engine/syncer_proto_util.h"
+#include "sync/test/engine/test_id_factory.h"
 #include "sync/protocol/bookmark_specifics.pb.h"
 #include "sync/syncable/directory.h"
 #include "sync/syncable/write_transaction.h"
 #include "sync/test/engine/test_id_factory.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using std::map;
 using std::string;
 using sync_pb::ClientToServerMessage;
 using sync_pb::CommitMessage;
@@ skipping 13 matching lines @@
       conflict_all_commits_(false),
       conflict_n_commits_(0),
       next_new_id_(10000),
       store_birthday_("Store BDay!"),
       store_birthday_sent_(false),
       client_stuck_(false),
       commit_time_rename_prepended_string_(""),
       countdown_to_postbuffer_fail_(0),
       directory_(directory),
       mid_commit_observer_(NULL),
+      clear_user_data_response_errortype_(sync_pb::SyncEnums::SUCCESS),
+      get_key_response_errortype_(sync_pb::SyncEnums::SUCCESS),
       throttling_(false),
       fail_with_auth_invalid_(false),
       fail_non_periodic_get_updates_(false),
       client_command_(NULL),
       next_position_in_parent_(2),
       use_legacy_bookmarks_protocol_(false),
       num_get_updates_requests_(0) {
   SetNewTimestamp(0);
   set_auth_token(kValidAuthToken);
 }
@@ skipping 74 matching lines @@
   }
   bool result = true;
   EXPECT_TRUE(!store_birthday_sent_ || post.has_store_birthday() ||
               post.message_contents() == ClientToServerMessage::AUTHENTICATE);
   store_birthday_sent_ = true;
 
   if (post.message_contents() == ClientToServerMessage::COMMIT) {
     ProcessCommit(&post, &response);
   } else if (post.message_contents() == ClientToServerMessage::GET_UPDATES) {
     ProcessGetUpdates(&post, &response);
+  } else if (post.message_contents() ==
+                 ClientToServerMessage::GET_ENCRYPTION_KEY) {
+    ProcessGetKey(&post, &response);
   } else {
     EXPECT_TRUE(false) << "Unknown/unsupported ClientToServerMessage";
     return false;
   }
   if (client_command_.get()) {
     response.mutable_client_command()->CopyFrom(*client_command_.get());
   }
 
   {
     base::AutoLock lock(response_code_override_lock_);
@@ skipping 293 matching lines @@
       sync_pb::DataTypeProgressMarker* new_marker =
           response->mutable_get_updates()->add_new_progress_marker();
       new_marker->set_data_type_id(gu.from_progress_marker(i).data_type_id());
       new_marker->set_token(token);
     }
   }
 
   update_queue_.pop_front();
 }
 
+void MockConnectionManager::SetKeystoreKey(const std::string& key) {
+  // Note: this is not a thread-safe set, ok for now.  NOT ok if tests
+  // run the syncer on the background thread while this method is called.
+  keystore_key_ = key;
+}
+
+void MockConnectionManager::SetGetKeyResponseStatus(
+    sync_pb::SyncEnums::ErrorType errortype ) {
+  // Note: this is not a thread-safe set, ok for now.  NOT ok if tests
+  // run the syncer on the background thread while this method is called.
+  get_key_response_errortype_ = errortype;
+}
+
+void MockConnectionManager::ProcessGetKey(
+    sync_pb::ClientToServerMessage* csm,
+    sync_pb::ClientToServerResponse* response) {
+  CHECK(csm->has_get_encryption_key());

[tim (not reviewing), 2012/07/19 21:42:03]

why CHECK over ASSERT_TRUE?

[Nicolas Zea, 2012/07/24 22:51:24]

Done.

+  ASSERT_EQ(csm->message_contents(), ClientToServerMessage::GET_ENCRYPTION_KEY);
+  response->mutable_get_encryption_key()->set_key(keystore_key_);
+  response->set_error_code(get_key_response_errortype_);
+}
+
 bool MockConnectionManager::ShouldConflictThisCommit() {
   bool conflict = false;
   if (conflict_all_commits_) {
     conflict = true;
   } else if (conflict_n_commits_ > 0) {
     conflict = true;
     --conflict_n_commits_;
   }
   return conflict;
 }
@@ skipping 125 matching lines @@
 
 void MockConnectionManager::UpdateConnectionStatus() {
   if (!server_reachable_) {
     server_status_ = HttpResponse::CONNECTION_UNAVAILABLE;
   } else {
     server_status_ = HttpResponse::SERVER_CONNECTION_OK;
   }
 }
 
 }  // namespace syncer