Fix policy test server key rotation feature

chrome/browser/policy/test/local_policy_test_server.cc

 // Copyright (c) 2013 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.
 
 #include "chrome/browser/policy/test/local_policy_test_server.h"
 
 #include <ctype.h>
 #include <stdint.h>
 
 #include <algorithm>
 #include <vector>
 
 #include "base/base_paths.h"
 #include "base/files/file_util.h"
 #include "base/json/json_writer.h"
 #include "base/path_service.h"
 #include "base/strings/stringprintf.h"
+#include "base/values.h"
 #include "build/build_config.h"
 #include "components/policy/core/common/cloud/cloud_policy_constants.h"
 #include "crypto/rsa_private_key.h"
 #include "net/test/python_utils.h"
 
 namespace policy {
 
 namespace {
 
 // Filename in the temporary directory storing the policy data.
@@ skipping 79 matching lines @@
   base::FilePath signature_file =
       server_data_dir_.GetPath().Append(kSigningKeySignatureFileName);
   bytes_written = base::WriteFile(
       signature_file,
       signature.c_str(),
       signature.size());
 
   return bytes_written == static_cast<int>(signature.size());
 }
 
+void LocalPolicyTestServer::EnableAutomaticRotationOfSigningKeys() {
+  automatic_rotation_of_signing_keys_enabled_ = true;
+}
+
 void LocalPolicyTestServer::RegisterClient(const std::string& dm_token,
                                            const std::string& device_id) {
   CHECK(server_data_dir_.IsValid());
 
   std::unique_ptr<base::DictionaryValue> client_dict(
       new base::DictionaryValue());
   client_dict->SetString(kClientStateKeyDeviceId, device_id);
   client_dict->SetString(kClientStateKeyDeviceToken, dm_token);
   client_dict->SetString(kClientStateKeyMachineName, std::string());
   client_dict->SetString(kClientStateKeyMachineId, std::string());
@@ skipping 103 matching lines @@
 }
 
 bool LocalPolicyTestServer::GenerateAdditionalArguments(
     base::DictionaryValue* arguments) const {
   if (!net::LocalTestServer::GenerateAdditionalArguments(arguments))
     return false;
 
   arguments->SetString("config-file", config_file_.AsUTF8Unsafe());
   if (!policy_key_.empty())
     arguments->SetString("policy-key", policy_key_.AsUTF8Unsafe());
+  if (automatic_rotation_of_signing_keys_enabled_) {
+    arguments->Set("rotate-policy-keys-automatically",
+                   base::Value::CreateNullValue());
+  }
   if (server_data_dir_.IsValid()) {
     arguments->SetString("data-dir", server_data_dir_.GetPath().AsUTF8Unsafe());
 
     if (!clients_.empty()) {
       std::string json;
       base::JSONWriter::Write(clients_, &json);
       base::FilePath client_state_file =
           server_data_dir_.GetPath().Append(kClientStateFileName);
       if (base::WriteFile(client_state_file, json.c_str(), json.size()) !=
           static_cast<int>(json.size())) {
         return false;
       }
       arguments->SetString("client-state", client_state_file.AsUTF8Unsafe());
     }
   }
 
   return true;
 }
 
 std::string LocalPolicyTestServer::GetSelector(const std::string& type,
                                                const std::string& entity_id) {
   std::string selector = type;
   if (!entity_id.empty())
     selector = base::StringPrintf("%s/%s", type.c_str(), entity_id.c_str());
   std::replace_if(selector.begin(), selector.end(), IsUnsafeCharacter, '_');
   return selector;
 }
 
 }  // namespace policy