cloud_external_data_manager_base_test_util.cc revision 5d1f7b1de12d16ceb2c938c56701a3e8bfa558f7 
1// Copyright 2013 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 "chrome/browser/chromeos/policy/cloud_external_data_manager_base_test_util.h"
6
7#include "base/callback.h"
8#include "base/memory/weak_ptr.h"
9#include "base/strings/string_number_conversions.h"
10#include "base/values.h"
11#include "components/policy/core/common/cloud/cloud_external_data_manager.h"
12#include "components/policy/core/common/cloud/cloud_policy_core.h"
13#include "components/policy/core/common/cloud/cloud_policy_store.h"
14#include "components/policy/core/common/external_data_fetcher.h"
15#include "components/policy/core/common/policy_map.h"
16#include "components/policy/core/common/policy_types.h"
17#include "crypto/sha2.h"
18#include "testing/gtest/include/gtest/gtest.h"
19
20namespace policy {
21namespace test {
22
23void ExternalDataFetchCallback(scoped_ptr<std::string>* destination,
24                               const base::Closure& done_callback,
25                               scoped_ptr<std::string> data) {
26  *destination = data.Pass();
27  done_callback.Run();
28}
29
30scoped_ptr<base::DictionaryValue> ConstructExternalDataReference(
31    const std::string& url,
32    const std::string& data) {
33  const std::string hash = crypto::SHA256HashString(data);
34  scoped_ptr<base::DictionaryValue> metadata(new base::DictionaryValue);
35  metadata->SetStringWithoutPathExpansion("url", url);
36  metadata->SetStringWithoutPathExpansion("hash", base::HexEncode(hash.c_str(),
37                                                                  hash.size()));
38  return metadata.Pass();
39}
40
41void SetExternalDataReference(CloudPolicyCore* core,
42                              const std::string& policy,
43                              scoped_ptr<base::DictionaryValue> metadata) {
44  CloudPolicyStore* store = core->store();
45  ASSERT_TRUE(store);
46  PolicyMap policy_map;
47  policy_map.Set(policy,
48                 POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER,
49                 metadata.release(),
50                 new ExternalDataFetcher(store->external_data_manager(),
51                                         policy));
52  store->SetPolicyMapForTesting(policy_map);
53}
54
55}  // namespace test
56}  // namespace policy
57