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
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef EXTENSIONS_RENDERER_API_MESSAGING_SEND_MESSAGE_TESTER_H_
#define EXTENSIONS_RENDERER_API_MESSAGING_SEND_MESSAGE_TESTER_H_
#include <string>
#include "base/memory/raw_ptr.h"
#include "v8/include/v8-forward.h"
namespace extensions {
class ScriptContext;
class TestIPCMessageSender;
struct MessageTarget;
// A helper class for testing the sendMessage, sendRequest, and connect API
// calls, since these are used across three different API namespaces
// (chrome.runtime, chrome.tabs, and chrome.extension).
class SendMessageTester {
public:
SendMessageTester(TestIPCMessageSender* ipc_sender,
ScriptContext* script_context,
int next_port_id,
const std::string& api_namespace);
SendMessageTester(const SendMessageTester&) = delete;
SendMessageTester& operator=(const SendMessageTester&) = delete;
~SendMessageTester();
// Whether we expect the port to be open or closed at the end of the call.
enum PortStatus {
CLOSED,
OPEN,
};
// Tests the sendMessage API with the specified expectations.
v8::Local<v8::Value> TestSendMessage(const std::string& args,
const std::string& expected_message,
const MessageTarget& expected_target,
PortStatus expected_port_status);
// Tests the sendRequest API with the specified expectations.
v8::Local<v8::Value> TestSendRequest(const std::string& args,
const std::string& expected_message,
const MessageTarget& expected_target,
PortStatus expected_port_status);
// Tests the sendNativeMessage API with the specified expectations.
v8::Local<v8::Value> TestSendNativeMessage(
const std::string& args,
const std::string& expected_message,
const std::string& expected_application_name);
// Tests the connect API with the specified expectations.
void TestConnect(const std::string& args,
const std::string& expected_channel,
const MessageTarget& expected_target);
private:
enum Method {
SEND_REQUEST,
SEND_MESSAGE,
SEND_NATIVE_MESSAGE,
};
// Common handler for testing sendMessage and sendRequest.
void TestSendMessageOrRequest(const std::string& args,
const std::string& expected_message,
const MessageTarget& expected_target,
PortStatus expected_port_status,
Method method,
v8::Local<v8::Value>& out_value);
raw_ptr<TestIPCMessageSender> ipc_sender_;
raw_ptr<ScriptContext> script_context_;
int next_port_id_;
std::string api_namespace_;
};
} // namespace extensions
#endif // EXTENSIONS_RENDERER_API_MESSAGING_SEND_MESSAGE_TESTER_H_
|
