chrome/browser/extensions/test_extension_environment.cc

   1 // Copyright (c) 2012 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/extensions/test_extension_environment.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/json/json_writer.h"
   9 #include "base/run_loop.h"
  10 #include "base/values.h"
  11 #include "chrome/browser/extensions/extension_service.h"
  12 #include "chrome/browser/extensions/test_extension_system.h"
  13 #include "chrome/browser/sessions/session_tab_helper.h"
  14 #include "chrome/test/base/testing_profile.h"
  15 #include "content/public/test/test_utils.h"
  16 #include "content/public/test/web_contents_tester.h"
  17 #include "extensions/common/extension.h"
  18 #include "extensions/common/extension_builder.h"
  19 #include "extensions/common/value_builder.h"
  20 #include "testing/gtest/include/gtest/gtest.h"
  21
  22 #if defined(USE_AURA)
  23 #include "ui/aura/env.h"
  24 #endif
  25
  26 namespace extensions {
  27
  28 using content::BrowserThread;
  29
  30 TestExtensionEnvironment::TestExtensionEnvironment()
  31     : profile_(new TestingProfile),
  32       extension_service_(NULL),
  33       extension_prefs_(NULL) {
  34 #if defined(USE_AURA)
  35   aura::Env::CreateInstance(true);
  36 #endif
  37 }
  38
  39 TestExtensionEnvironment::~TestExtensionEnvironment() {
  40 #if defined(USE_AURA)
  41   aura::Env::DeleteInstance();
  42 #endif
  43 }
  44
  45 TestingProfile* TestExtensionEnvironment::profile() const {
  46   return profile_.get();
  47 }
  48
  49 TestExtensionSystem* TestExtensionEnvironment::GetExtensionSystem() {
  50   return static_cast<TestExtensionSystem*>(ExtensionSystem::Get(profile()));
  51 }
  52
  53 ExtensionService* TestExtensionEnvironment::GetExtensionService() {
  54   if (extension_service_ == NULL) {
  55     extension_service_ = GetExtensionSystem()->CreateExtensionService(
  56         base::CommandLine::ForCurrentProcess(), base::FilePath(), false);
  57   }
  58   return extension_service_;
  59 }
  60
  61 ExtensionPrefs* TestExtensionEnvironment::GetExtensionPrefs() {
  62   if (extension_prefs_ == NULL) {
  63     extension_prefs_ = GetExtensionSystem()->CreateExtensionPrefs(
  64         base::CommandLine::ForCurrentProcess(), base::FilePath());
  65   }
  66   return extension_prefs_;
  67 }
  68
  69 const Extension* TestExtensionEnvironment::MakeExtension(
  70     const base::Value& manifest_extra) {
  71   scoped_ptr<base::DictionaryValue> manifest = DictionaryBuilder()
  72       .Set("name", "Extension")
  73       .Set("version", "1.0")
  74       .Set("manifest_version", 2)
  75       .Build();
  76   const base::DictionaryValue* manifest_extra_dict;
  77   if (manifest_extra.GetAsDictionary(&manifest_extra_dict)) {
  78     manifest->MergeDictionary(manifest_extra_dict);
  79   } else {
  80     std::string manifest_json;
  81     base::JSONWriter::Write(&manifest_extra, &manifest_json);
  82     ADD_FAILURE() << "Expected dictionary; got \"" << manifest_json << "\"";
  83   }
  84
  85   scoped_refptr<Extension> result =
  86       ExtensionBuilder().SetManifest(manifest.Pass()).Build();
  87   GetExtensionService()->AddExtension(result.get());
  88   return result.get();
  89 }
  90
  91 scoped_ptr<content::WebContents> TestExtensionEnvironment::MakeTab() const {
  92   scoped_ptr<content::WebContents> contents(
  93       content::WebContentsTester::CreateTestWebContents(profile(), NULL));
  94   // Create a tab id.
  95   SessionTabHelper::CreateForWebContents(contents.get());
  96   return contents.Pass();
  97 }
  98
  99 }  // namespace extensions