Emulate StartAuthPolicyService from FakeUpstartClient

chromeos/dbus/fake_auth_policy_client.cc

 // Copyright 2016 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 "chromeos/dbus/fake_auth_policy_client.h"
 
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/location.h"
@@ skipping 58 matching lines @@
 FakeAuthPolicyClient::FakeAuthPolicyClient() {}
 
 FakeAuthPolicyClient::~FakeAuthPolicyClient() {}
 
 void FakeAuthPolicyClient::Init(dbus::Bus* bus) {}
 
 void FakeAuthPolicyClient::JoinAdDomain(const std::string& machine_name,
                                         const std::string& user_principal_name,
                                         int password_fd,
                                         const JoinCallback& callback) {
+  if (!started_) {
+    LOG(ERROR) << "authpolicyd not started";
+    callback.Run(authpolicy::ERROR_DBUS_FAILURE);
+    return;
+  }
   if (machine_name.size() > kMaxMachineNameLength) {
     callback.Run(authpolicy::ERROR_MACHINE_NAME_TOO_LONG);
     return;
   }
 
   if (machine_name.empty() ||
       machine_name.find_first_of(kInvalidMachineNameCharacters) !=
           std::string::npos) {
     callback.Run(authpolicy::ERROR_BAD_MACHINE_NAME);
     return;
   }
 
   std::vector<std::string> parts = base::SplitString(
       user_principal_name, "@", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
   if (parts.size() != 2 || parts[0].empty() || parts[1].empty()) {
     callback.Run(authpolicy::ERROR_PARSE_UPN_FAILED);
     return;
   }
 
   callback.Run(authpolicy::ERROR_NONE);
 }
 
 void FakeAuthPolicyClient::AuthenticateUser(
     const std::string& user_principal_name,
     int password_fd,
     const AuthCallback& callback) {
   authpolicy::ActiveDirectoryAccountData account_data;
+  if (!started_) {
+    LOG(ERROR) << "authpolicyd not started";
+    callback.Run(authpolicy::ERROR_DBUS_FAILURE, account_data);
+    return;
+  }
   account_data.set_account_id(base::MD5String(user_principal_name));
   callback.Run(authpolicy::ERROR_NONE, account_data);
 }
 
 void FakeAuthPolicyClient::RefreshDevicePolicy(
     const RefreshPolicyCallback& callback) {
+  if (!started_) {
+    LOG(ERROR) << "authpolicyd not started";
+    callback.Run(false);
+    return;
+  }
   base::FilePath policy_path;
   if (!PathService::Get(chromeos::FILE_OWNER_KEY, &policy_path)) {
     callback.Run(false);
     return;
   }
   policy_path = policy_path.DirName().AppendASCII("stub_device_policy");
 
   em::ChromeDeviceSettingsProto policy;
   std::string payload;
   CHECK(policy.SerializeToString(&payload));
 
   // Drop file for SessionManagerClientStubImpl to read.
   base::PostTaskWithTraitsAndReplyWithResult(
       FROM_HERE, base::TaskTraits()
                      .WithShutdownBehavior(
                          base::TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN)
                      .WithPriority(base::TaskPriority::BACKGROUND)
                      .MayBlock(),
       Bind(&WritePolicyFile, policy_path, payload, "google/chromeos/device"),
       callback);
 }
 
 void FakeAuthPolicyClient::RefreshUserPolicy(
     const AccountId& account_id,
     const RefreshPolicyCallback& callback) {
+  if (!started_) {
+    LOG(ERROR) << "authpolicyd not started";
+    callback.Run(false);
+    return;
+  }
   base::FilePath policy_path;
   if (!PathService::Get(chromeos::DIR_USER_POLICY_KEYS, &policy_path)) {
     callback.Run(false);
     return;
   }
   const cryptohome::Identification cryptohome_identification(account_id);
   const std::string sanitized_username =
       chromeos::CryptohomeClient::GetStubSanitizedUsername(
           cryptohome_identification);
   policy_path = policy_path.AppendASCII(sanitized_username);
   policy_path = policy_path.AppendASCII("stub_policy");
 
   em::CloudPolicySettings policy;
   std::string payload;
   CHECK(policy.SerializeToString(&payload));
 
   // Drop file for SessionManagerClientStubImpl to read.
   base::PostTaskWithTraitsAndReplyWithResult(
       FROM_HERE, base::TaskTraits()
                      .WithShutdownBehavior(
                          base::TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN)
                      .WithPriority(base::TaskPriority::BACKGROUND)
                      .MayBlock(),
       base::Bind(&WritePolicyFile, policy_path, payload,
                  "google/chromeos/user"),
       callback);
 }
 
 }  // namespace chromeos