src/chrome/browser/chromeos/policy/user_cloud_external_data_manager_browsertest.cc

   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 <string>
   6
   7 #include "base/bind.h"
   8 #include "base/callback.h"
   9 #include "base/files/file_path.h"
  10 #include "base/files/file_util.h"
  11 #include "base/memory/scoped_ptr.h"
  12 #include "base/path_service.h"
  13 #include "base/run_loop.h"
  14 #include "base/values.h"
  15 #include "chrome/browser/chromeos/policy/cloud_external_data_manager_base.h"
  16 #include "chrome/browser/chromeos/policy/cloud_external_data_manager_base_test_util.h"
  17 #include "chrome/browser/chromeos/policy/user_cloud_policy_manager_chromeos.h"
  18 #include "chrome/browser/chromeos/policy/user_cloud_policy_manager_factory_chromeos.h"
  19 #include "chrome/browser/policy/profile_policy_connector.h"
  20 #include "chrome/browser/policy/profile_policy_connector_factory.h"
  21 #include "chrome/browser/profiles/profile.h"
  22 #include "chrome/browser/ui/browser.h"
  23 #include "chrome/common/chrome_paths.h"
  24 #include "chrome/test/base/in_process_browser_test.h"
  25 #include "components/policy/core/common/cloud/cloud_policy_core.h"
  26 #include "components/policy/core/common/external_data_fetcher.h"
  27 #include "components/policy/core/common/policy_map.h"
  28 #include "components/policy/core/common/policy_service.h"
  29 #include "content/public/test/test_utils.h"
  30 #include "net/test/embedded_test_server/embedded_test_server.h"
  31 #include "policy/policy_constants.h"
  32 #include "testing/gtest/include/gtest/gtest.h"
  33 #include "url/gurl.h"
  34
  35 namespace policy {
  36
  37 namespace {
  38
  39 const char kExternalDataPath[] = "policy/blank.html";
  40
  41 }  // namespace
  42
  43 typedef InProcessBrowserTest UserCloudExternalDataManagerTest;
  44
  45 IN_PROC_BROWSER_TEST_F(UserCloudExternalDataManagerTest, FetchExternalData) {
  46   CloudExternalDataManagerBase::SetMaxExternalDataSizeForTesting(1000);
  47
  48   ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady());
  49   const GURL url =
  50       embedded_test_server()->GetURL(std::string("/") + kExternalDataPath);
  51
  52   base::FilePath test_dir;
  53   ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &test_dir));
  54   std::string external_data;
  55   ASSERT_TRUE(base::ReadFileToString(test_dir.AppendASCII(kExternalDataPath),
  56                                      &external_data));
  57   ASSERT_FALSE(external_data.empty());
  58
  59   scoped_ptr<base::DictionaryValue> metadata =
  60       test::ConstructExternalDataReference(url.spec(), external_data);
  61 #if defined(OS_CHROMEOS)
  62   UserCloudPolicyManagerChromeOS* policy_manager =
  63       UserCloudPolicyManagerFactoryChromeOS::GetForProfile(
  64           browser()->profile());
  65 #else
  66   UserCloudPolicyManager* policy_manager =
  67       UserCloudPolicyManagerFactory::GetForBrowserContext(browser()->profile());
  68 #endif
  69   ASSERT_TRUE(policy_manager);
  70   // TODO(bartfab): The test injects an ExternalDataFetcher for an arbitrary
  71   // policy. This is only done because there are no policies that reference
  72   // external data yet. Once the first such policy is added, switch the test to
  73   // that policy and stop injecting a manually instantiated ExternalDataFetcher.
  74   test::SetExternalDataReference(policy_manager->core(),
  75                                  key::kHomepageLocation,
  76                                  make_scoped_ptr(metadata->DeepCopy()));
  77   content::RunAllPendingInMessageLoop();
  78
  79   ProfilePolicyConnector* policy_connector =
  80       ProfilePolicyConnectorFactory::GetForProfile(browser()->profile());
  81   ASSERT_TRUE(policy_connector);
  82   const PolicyMap& policies = policy_connector->policy_service()->GetPolicies(
  83       PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()));
  84   const PolicyMap::Entry* policy_entry = policies.Get(key::kHomepageLocation);
  85   ASSERT_TRUE(policy_entry);
  86   EXPECT_TRUE(base::Value::Equals(metadata.get(), policy_entry->value));
  87   ASSERT_TRUE(policy_entry->external_data_fetcher);
  88
  89   base::RunLoop run_loop;
  90   scoped_ptr<std::string> fetched_external_data;
  91   policy_entry->external_data_fetcher->Fetch(base::Bind(
  92       &test::ExternalDataFetchCallback,
  93       &fetched_external_data,
  94       run_loop.QuitClosure()));
  95   run_loop.Run();
  96
  97   ASSERT_TRUE(fetched_external_data);
  98   EXPECT_EQ(external_data, *fetched_external_data);
  99 }
 100
 101 }  // namespace policy