Index: content/browser/service_worker/embedded_worker_test_helper.h |
diff --git a/content/browser/service_worker/embedded_worker_test_helper.h b/content/browser/service_worker/embedded_worker_test_helper.h |
index 88bf11bee43571ba4fb8a2c205c564bd3de553f9..c797939ea3591fa105ac131e435932d6d79002e6 100644 |
--- a/content/browser/service_worker/embedded_worker_test_helper.h |
+++ b/content/browser/service_worker/embedded_worker_test_helper.h |
@@ -15,9 +15,13 @@ |
#include "base/containers/hash_tables.h" |
#include "base/macros.h" |
#include "base/memory/weak_ptr.h" |
+#include "base/optional.h" |
+#include "content/common/service_worker/embedded_worker.mojom.h" |
#include "ipc/ipc_listener.h" |
#include "ipc/ipc_test_sink.h" |
+#include "mojo/public/cpp/bindings/binding.h" |
#include "services/shell/public/interfaces/interface_provider.mojom.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
#include "url/gurl.h" |
@@ -60,6 +64,39 @@ struct ServiceWorkerFetchRequest; |
class EmbeddedWorkerTestHelper : public IPC::Sender, |
public IPC::Listener { |
public: |
+ class MockEmbeddedWorkerInstanceClient |
+ : public mojom::EmbeddedWorkerInstanceClient { |
+ public: |
+ explicit MockEmbeddedWorkerInstanceClient( |
+ base::WeakPtr<EmbeddedWorkerTestHelper> helper); |
+ ~MockEmbeddedWorkerInstanceClient() override; |
+ |
+ static void Bind(const base::WeakPtr<EmbeddedWorkerTestHelper>& helper, |
+ mojom::EmbeddedWorkerInstanceClientRequest request); |
+ |
+ // MOCK_METHOD cannot be used for StartWorker because the mock method cannot |
+ // take move-only objects directly. |
falken
2016/09/07 05:01:42
I'd rather we avoid gmock if possible.
shimazu
2016/09/12 06:28:19
Done.
|
+ MOCK_METHOD1(MockStartWorker, |
+ void(mojom::EmbeddedWorkerStartWorkerParams* params)); |
+ |
+ private: |
+ // Implementation of mojo interface; they just calls a mock function. |
+ void StartWorker( |
+ mojom::EmbeddedWorkerStartWorkerParamsPtr params) override { |
+ MockStartWorker(params.get()); |
+ } |
+ |
+ // Default behavior which can be overridden by using mock interfaces. |
+ void DefaultStartWorker(mojom::EmbeddedWorkerStartWorkerParams* params); |
+ |
+ base::WeakPtr<EmbeddedWorkerTestHelper> helper_; |
+ mojo::Binding<mojom::EmbeddedWorkerInstanceClient> binding_; |
+ |
+ base::Optional<int> embedded_worker_id_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(MockEmbeddedWorkerInstanceClient); |
+ }; |
+ |
// If |user_data_directory| is empty, the context makes storage stuff in |
// memory. |
explicit EmbeddedWorkerTestHelper(const base::FilePath& user_data_directory); |
@@ -80,6 +117,15 @@ class EmbeddedWorkerTestHelper : public IPC::Sender, |
// Inner IPC sink for script context messages sent via EmbeddedWorker. |
IPC::TestSink* inner_ipc_sink() { return &inner_sink_; } |
+ // Creates mock instances before IPCs are fired to check IPCs are called as |
+ // your expectation. |
+ void PrepareMockInstanceClients(int number_of_clients); |
+ |
+ std::vector<std::unique_ptr<MockEmbeddedWorkerInstanceClient>>* |
+ mock_instance_clients() { |
+ return &mock_instance_clients_; |
+ } |
+ |
ServiceWorkerContextCore* context(); |
ServiceWorkerContextWrapper* context_wrapper() { return wrapper_.get(); } |
void ShutdownContext(); |
@@ -95,9 +141,8 @@ class EmbeddedWorkerTestHelper : public IPC::Sender, |
return embedded_worker_id_service_worker_version_id_map_; |
} |
- // Only used for tests that force creating a new render process. There is no |
- // corresponding MockRenderProcessHost. |
- int new_render_process_id() const { return mock_render_process_id_ + 1; } |
+ // Only used for tests that force creating a new render process. |
+ int new_render_process_id() const { return new_mock_render_process_id_; } |
TestBrowserContext* browser_context() { return browser_context_.get(); } |
@@ -181,16 +226,24 @@ class EmbeddedWorkerTestHelper : public IPC::Sender, |
std::unique_ptr<TestBrowserContext> browser_context_; |
std::unique_ptr<MockRenderProcessHost> render_process_host_; |
+ std::unique_ptr<MockRenderProcessHost> new_render_process_host_; |
scoped_refptr<ServiceWorkerContextWrapper> wrapper_; |
IPC::TestSink sink_; |
IPC::TestSink inner_sink_; |
+ std::vector<std::unique_ptr<MockEmbeddedWorkerInstanceClient>> |
+ mock_instance_clients_; |
+ size_t mock_instance_clients_next_; |
+ |
int next_thread_id_; |
int mock_render_process_id_; |
+ int new_mock_render_process_id_; |
std::unique_ptr<shell::InterfaceRegistry> render_process_interface_registry_; |
+ std::unique_ptr<shell::InterfaceRegistry> |
+ new_render_process_interface_registry_; |
std::map<int, int64_t> embedded_worker_id_service_worker_version_id_map_; |