| // Copyright 2017 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. |
| |
| #ifndef EXTENSIONS_RENDERER_SEND_MESSAGE_TESTER_H_ |
| #define EXTENSIONS_RENDERER_SEND_MESSAGE_TESTER_H_ |
| |
| #include <string> |
| |
| #include "base/macros.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(); |
| |
| // 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. |
| void TestSendMessage(const std::string& args, |
| const std::string& expected_message, |
| const MessageTarget& expected_target, |
| bool expected_include_tls_channel_id, |
| PortStatus expected_port_status); |
| |
| // Tests the sendRequest API with the specified expectations. |
| void TestSendRequest(const std::string& args, |
| const std::string& expected_message, |
| const MessageTarget& expected_target, |
| PortStatus expected_port_status); |
| |
| // Tests the connect API with the specified expectaions. |
| void TestConnect(const std::string& args, |
| const std::string& expected_channel, |
| const MessageTarget& expected_target, |
| bool expected_include_tls_channel_id); |
| |
| private: |
| enum Method { |
| SEND_REQUEST, |
| SEND_MESSAGE, |
| }; |
| |
| // Common handler for testing sendMessage and sendRequest. |
| void TestSendMessageOrRequest(const std::string& args, |
| const std::string& expected_message, |
| const MessageTarget& expected_target, |
| bool expected_include_tls_channel_id, |
| PortStatus expected_port_status, |
| Method method); |
| |
| TestIPCMessageSender* ipc_sender_; |
| ScriptContext* script_context_; |
| int next_port_id_; |
| std::string api_namespace_; |
| |
| DISALLOW_COPY_AND_ASSIGN(SendMessageTester); |
| }; |
| |
| } // namespace extensions |
| |
| #endif // EXTENSIONS_RENDERER_SEND_MESSAGE_TESTER_H_ |