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
26namespace extensions {
27
28using content::BrowserThread;
29
30TestExtensionEnvironment::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
39TestExtensionEnvironment::~TestExtensionEnvironment() {
40#if defined(USE_AURA)
41  aura::Env::DeleteInstance();
42#endif
43}
44
45TestingProfile* TestExtensionEnvironment::profile() const {
46  return profile_.get();
47}
48
49TestExtensionSystem* TestExtensionEnvironment::GetExtensionSystem() {
50  return static_cast<TestExtensionSystem*>(ExtensionSystem::Get(profile()));
51}
52
53ExtensionService* TestExtensionEnvironment::GetExtensionService() {
54  if (extension_service_ == NULL) {
55    extension_service_ = GetExtensionSystem()->CreateExtensionService(
56        CommandLine::ForCurrentProcess(), base::FilePath(), false);
57  }
58  return extension_service_;
59}
60
61ExtensionPrefs* TestExtensionEnvironment::GetExtensionPrefs() {
62  if (extension_prefs_ == NULL) {
63    extension_prefs_ = GetExtensionSystem()->CreateExtensionPrefs(
64        CommandLine::ForCurrentProcess(), base::FilePath());
65  }
66  return extension_prefs_;
67}
68
69const 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
91scoped_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
100