src/chrome/browser/extensions/api/test/test_api.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/api/test/test_api.h"
   6
   7 #include <string>
   8
   9 #include "base/command_line.h"
  10 #include "base/memory/singleton.h"
  11 #include "chrome/browser/chrome_notification_types.h"
  12 #include "chrome/browser/extensions/extension_function_dispatcher.h"
  13 #include "chrome/browser/extensions/extension_service.h"
  14 #include "chrome/browser/profiles/profile.h"
  15 #include "chrome/browser/ui/browser_commands.h"
  16 #include "chrome/common/chrome_switches.h"
  17 #include "chrome/common/extensions/api/test.h"
  18 #include "content/public/browser/notification_service.h"
  19 #include "extensions/browser/quota_service.h"
  20
  21 namespace {
  22
  23 // If you see this error in your test, you need to set the config state
  24 // to be returned by chrome.test.getConfig().  Do this by calling
  25 // TestGetConfigFunction::set_test_config_state(Value* state)
  26 // in test set up.
  27 const char kNoTestConfigDataError[] = "Test configuration was not set.";
  28
  29 const char kNotTestProcessError[] =
  30     "The chrome.test namespace is only available in tests.";
  31
  32 }  // namespace
  33
  34 namespace extensions {
  35
  36 namespace CreateIncognitoTab = api::test::CreateIncognitoTab;
  37 namespace Log = api::test::Log;
  38 namespace NotifyFail = api::test::NotifyFail;
  39 namespace PassMessage = api::test::PassMessage;
  40
  41 TestExtensionFunction::~TestExtensionFunction() {}
  42
  43 void TestExtensionFunction::Run() {
  44   if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kTestType)) {
  45     error_ = kNotTestProcessError;
  46     SendResponse(false);
  47     return;
  48   }
  49   SendResponse(RunImpl());
  50 }
  51
  52 TestNotifyPassFunction::~TestNotifyPassFunction() {}
  53
  54 bool TestNotifyPassFunction::RunImpl() {
  55   content::NotificationService::current()->Notify(
  56       chrome::NOTIFICATION_EXTENSION_TEST_PASSED,
  57       content::Source<content::BrowserContext>(dispatcher()->browser_context()),
  58       content::NotificationService::NoDetails());
  59   return true;
  60 }
  61
  62 TestNotifyFailFunction::~TestNotifyFailFunction() {}
  63
  64 bool TestNotifyFailFunction::RunImpl() {
  65   scoped_ptr<NotifyFail::Params> params(NotifyFail::Params::Create(*args_));
  66   EXTENSION_FUNCTION_VALIDATE(params.get());
  67   content::NotificationService::current()->Notify(
  68       chrome::NOTIFICATION_EXTENSION_TEST_FAILED,
  69       content::Source<content::BrowserContext>(dispatcher()->browser_context()),
  70       content::Details<std::string>(&params->message));
  71   return true;
  72 }
  73
  74 TestLogFunction::~TestLogFunction() {}
  75
  76 bool TestLogFunction::RunImpl() {
  77   scoped_ptr<Log::Params> params(Log::Params::Create(*args_));
  78   EXTENSION_FUNCTION_VALIDATE(params.get());
  79   VLOG(1) << params->message;
  80   return true;
  81 }
  82
  83 TestResetQuotaFunction::~TestResetQuotaFunction() {}
  84
  85 bool TestResetQuotaFunction::RunImpl() {
  86   ExtensionService* service = GetProfile()->GetExtensionService();
  87   QuotaService* quota = service->quota_service();
  88   quota->Purge();
  89   quota->violation_errors_.clear();
  90   return true;
  91 }
  92
  93 TestCreateIncognitoTabFunction::
  94    ~TestCreateIncognitoTabFunction() {}
  95
  96 bool TestCreateIncognitoTabFunction::RunImpl() {
  97   scoped_ptr<CreateIncognitoTab::Params> params(
  98       CreateIncognitoTab::Params::Create(*args_));
  99   EXTENSION_FUNCTION_VALIDATE(params.get());
 100   chrome::OpenURLOffTheRecord(
 101       GetProfile(), GURL(params->url), chrome::GetActiveDesktop());
 102   return true;
 103 }
 104
 105 bool TestSendMessageFunction::RunImpl() {
 106   scoped_ptr<PassMessage::Params> params(
 107       PassMessage::Params::Create(*args_));
 108   EXTENSION_FUNCTION_VALIDATE(params.get());
 109   content::NotificationService::current()->Notify(
 110       chrome::NOTIFICATION_EXTENSION_TEST_MESSAGE,
 111       content::Source<TestSendMessageFunction>(this),
 112       content::Details<std::string>(&params->message));
 113   return true;
 114 }
 115
 116 TestSendMessageFunction::~TestSendMessageFunction() {}
 117
 118 void TestSendMessageFunction::Reply(const std::string& message) {
 119   SetResult(new base::StringValue(message));
 120   SendResponse(true);
 121 }
 122
 123 // static
 124 void TestGetConfigFunction::set_test_config_state(
 125     base::DictionaryValue* value) {
 126   TestConfigState* test_config_state = TestConfigState::GetInstance();
 127   test_config_state->set_config_state(value);
 128 }
 129
 130 TestGetConfigFunction::TestConfigState::TestConfigState()
 131   : config_state_(NULL) {}
 132
 133 // static
 134 TestGetConfigFunction::TestConfigState*
 135 TestGetConfigFunction::TestConfigState::GetInstance() {
 136   return Singleton<TestConfigState>::get();
 137 }
 138
 139 TestGetConfigFunction::~TestGetConfigFunction() {}
 140
 141 bool TestGetConfigFunction::RunImpl() {
 142   TestConfigState* test_config_state = TestConfigState::GetInstance();
 143
 144   if (!test_config_state->config_state()) {
 145     error_ = kNoTestConfigDataError;
 146     return false;
 147   }
 148
 149   SetResult(test_config_state->config_state()->DeepCopy());
 150   return true;
 151 }
 152
 153 }  // namespace extensions