1// Copyright 2014 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/extension_api_unittest.h" 6 7#include "base/values.h" 8#include "chrome/browser/extensions/extension_function_test_utils.h" 9#include "chrome/browser/ui/browser.h" 10#include "chrome/browser/ui/tabs/tab_strip_model.h" 11#include "content/public/browser/web_contents.h" 12#include "content/public/common/url_constants.h" 13#include "extensions/browser/extension_function.h" 14#include "extensions/common/extension.h" 15#include "extensions/common/manifest.h" 16#include "extensions/common/manifest_handlers/background_info.h" 17 18namespace utils = extension_function_test_utils; 19 20namespace extensions { 21 22ExtensionApiUnittest::ExtensionApiUnittest() : contents_(NULL) { 23} 24 25ExtensionApiUnittest::~ExtensionApiUnittest() { 26} 27 28void ExtensionApiUnittest::SetUp() { 29 BrowserWithTestWindowTest::SetUp(); 30 extension_ = utils::CreateEmptyExtensionWithLocation(Manifest::UNPACKED); 31} 32 33void ExtensionApiUnittest::CreateBackgroundPage() { 34 if (!contents_) { 35 GURL url = BackgroundInfo::GetBackgroundURL(extension()); 36 if (url.is_empty()) 37 url = GURL(url::kAboutBlankURL); 38 AddTab(browser(), url); 39 contents_ = browser()->tab_strip_model()->GetActiveWebContents(); 40 } 41} 42 43scoped_ptr<base::Value> ExtensionApiUnittest::RunFunctionAndReturnValue( 44 UIThreadExtensionFunction* function, const std::string& args) { 45 function->set_extension(extension()); 46 if (contents_) 47 function->SetRenderViewHost(contents_->GetRenderViewHost()); 48 return scoped_ptr<base::Value>( 49 utils::RunFunctionAndReturnSingleResult(function, args, browser())); 50} 51 52scoped_ptr<base::DictionaryValue> 53ExtensionApiUnittest::RunFunctionAndReturnDictionary( 54 UIThreadExtensionFunction* function, const std::string& args) { 55 base::Value* value = RunFunctionAndReturnValue(function, args).release(); 56 base::DictionaryValue* dict = NULL; 57 58 if (value && !value->GetAsDictionary(&dict)) 59 delete value; 60 61 // We expect to either have successfuly retrieved a dictionary from the value, 62 // or the value to have been NULL. 63 EXPECT_TRUE(dict || !value); 64 return scoped_ptr<base::DictionaryValue>(dict); 65} 66 67scoped_ptr<base::ListValue> ExtensionApiUnittest::RunFunctionAndReturnList( 68 UIThreadExtensionFunction* function, const std::string& args) { 69 base::Value* value = RunFunctionAndReturnValue(function, args).release(); 70 base::ListValue* list = NULL; 71 72 if (value && !value->GetAsList(&list)) 73 delete value; 74 75 // We expect to either have successfuly retrieved a list from the value, 76 // or the value to have been NULL. 77 EXPECT_TRUE(list); 78 return scoped_ptr<base::ListValue>(list); 79} 80 81std::string ExtensionApiUnittest::RunFunctionAndReturnError( 82 UIThreadExtensionFunction* function, const std::string& args) { 83 function->set_extension(extension()); 84 if (contents_) 85 function->SetRenderViewHost(contents_->GetRenderViewHost()); 86 return utils::RunFunctionAndReturnError(function, args, browser()); 87} 88 89void ExtensionApiUnittest::RunFunction( 90 UIThreadExtensionFunction* function, const std::string& args) { 91 RunFunctionAndReturnValue(function, args); 92} 93 94} // namespace extensions 95