1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/extensions/extension_function_test_utils.h"
#include <string>
#include "base/json/json_reader.h"
#include "base/values.h"
#include "chrome/browser/ui/browser.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extension_function_test_utils {
base::Value* ParseJSON(const std::string& data) {
const bool kAllowTrailingComma = false;
return base::JSONReader::Read(data, kAllowTrailingComma);
}
base::ListValue* ParseList(const std::string& data) {
scoped_ptr<base::Value> result(ParseJSON(data));
if (result.get() && result->IsType(base::Value::TYPE_LIST))
return static_cast<base::ListValue*>(result.release());
else
return NULL;
}
base::DictionaryValue* ParseDictionary(
const std::string& data) {
scoped_ptr<base::Value> result(ParseJSON(data));
if (result.get() && result->IsType(base::Value::TYPE_DICTIONARY))
return static_cast<base::DictionaryValue*>(result.release());
else
return NULL;
}
bool GetBoolean(base::DictionaryValue* val, const std::string& key) {
bool result = false;
if (!val->GetBoolean(key, &result))
ADD_FAILURE() << key << " does not exist or is not a boolean.";
return result;
}
int GetInteger(base::DictionaryValue* val, const std::string& key) {
int result = 0;
if (!val->GetInteger(key, &result))
ADD_FAILURE() << key << " does not exist or is not an integer.";
return result;
}
std::string GetString(base::DictionaryValue* val, const std::string& key) {
std::string result;
if (!val->GetString(key, &result))
ADD_FAILURE() << key << " does not exist or is not a string.";
return result;
}
base::DictionaryValue* ToDictionary(base::Value* val) {
if (!val || !val->IsType(base::Value::TYPE_DICTIONARY))
ADD_FAILURE() << "value is null or not a dictionary.";
return static_cast<base::DictionaryValue*>(val);
}
std::string RunFunctionAndReturnError(UIThreadExtensionFunction* function,
const std::string& args,
Browser* browser) {
return RunFunctionAndReturnError(function, args, browser, NONE);
}
std::string RunFunctionAndReturnError(UIThreadExtensionFunction* function,
const std::string& args,
Browser* browser,
RunFunctionFlags flags) {
scoped_refptr<ExtensionFunction> function_owner(function);
RunFunction(function, args, browser, flags);
if (function->GetResultValue())
ADD_FAILURE() << function->GetResult();
return function->GetError();
}
base::Value* RunFunctionAndReturnResult(UIThreadExtensionFunction* function,
const std::string& args,
Browser* browser) {
return RunFunctionAndReturnResult(function, args, browser, NONE);
}
base::Value* RunFunctionAndReturnResult(UIThreadExtensionFunction* function,
const std::string& args,
Browser* browser,
RunFunctionFlags flags) {
scoped_refptr<ExtensionFunction> function_owner(function);
RunFunction(function, args, browser, flags);
if (!function->GetError().empty())
ADD_FAILURE() << function->GetError();
return function->GetResultValue()->DeepCopy();
}
void RunFunction(UIThreadExtensionFunction* function,
const std::string& args,
Browser* browser,
RunFunctionFlags flags) {
scoped_ptr<base::ListValue> parsed_args(ParseList(args));
function->SetArgs(parsed_args.get());
function->set_profile(browser->profile());
function->set_include_incognito(flags & INCLUDE_INCOGNITO);
function->Run();
}
} // namespace extension_function_test_utils
|
