[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 396 matching lines @@
   EXPECT_TRUE(gu.fetch_folders());
   EXPECT_FALSE(gu.has_requested_types());
   if (update_queue_.empty()) {
     GetUpdateResponse();
   }
   sync_pb::GetUpdatesResponse* updates = &update_queue_.front();
   for (int i = 0; i < updates->entries_size(); ++i) {
     if (!updates->entries(i).deleted()) {
       ModelType entry_type = GetModelType(updates->entries(i));
       EXPECT_TRUE(
-        IsModelTypePresentInSpecifics(gu.from_progress_marker(), entry_type))
+          IsModelTypePresentInSpecifics(gu.from_progress_marker(), entry_type))
           << "Syncer did not request updates being provided by the test.";
     }
   }
 
   response->mutable_get_updates()->CopyFrom(*updates);
 
   // Set appropriate progress markers, overriding the value squirreled
   // away by ApplyToken().
   std::string token = response->get_updates().new_progress_marker(0).token();
   response->mutable_get_updates()->clear_new_progress_marker();
   for (int i = 0; i < gu.from_progress_marker_size(); ++i) {
     if (gu.from_progress_marker(i).token() != token) {
       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);
     }
   }
 
+  // Fill the keystore key if requested.
+  if (gu.need_encryption_key())
+    response->mutable_get_updates()->set_encryption_key(keystore_key_);
+
   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;
+}
+
 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