| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_TEST_HELPER_H_ | 5 #ifndef CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_TEST_HELPER_H_ |
| 6 #define CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_TEST_HELPER_H_ | 6 #define CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_TEST_HELPER_H_ |
| 7 | 7 |
| 8 #include <vector> | 8 #include <vector> |
| 9 | 9 |
| 10 #include "base/callback.h" | 10 #include "base/callback.h" |
| (...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 60 | 60 |
| 61 // IPC sink for EmbeddedWorker messages. | 61 // IPC sink for EmbeddedWorker messages. |
| 62 IPC::TestSink* ipc_sink() { return &sink_; } | 62 IPC::TestSink* ipc_sink() { return &sink_; } |
| 63 // Inner IPC sink for script context messages sent via EmbeddedWorker. | 63 // Inner IPC sink for script context messages sent via EmbeddedWorker. |
| 64 IPC::TestSink* inner_ipc_sink() { return &inner_sink_; } | 64 IPC::TestSink* inner_ipc_sink() { return &inner_sink_; } |
| 65 | 65 |
| 66 ServiceWorkerContextCore* context(); | 66 ServiceWorkerContextCore* context(); |
| 67 ServiceWorkerContextWrapper* context_wrapper() { return wrapper_.get(); } | 67 ServiceWorkerContextWrapper* context_wrapper() { return wrapper_.get(); } |
| 68 void ShutdownContext(); | 68 void ShutdownContext(); |
| 69 | 69 |
| 70 int mock_render_process_id() const { return mock_render_process_id_;} |
| 71 |
| 70 protected: | 72 protected: |
| 71 // Called when StartWorker, StopWorker and SendMessageToWorker message | 73 // Called when StartWorker, StopWorker and SendMessageToWorker message |
| 72 // is sent to the embedded worker. Override if necessary. By default | 74 // is sent to the embedded worker. Override if necessary. By default |
| 73 // they verify given parameters and: | 75 // they verify given parameters and: |
| 74 // - OnStartWorker calls SimulateWorkerStarted | 76 // - OnStartWorker calls SimulateWorkerStarted |
| 75 // - OnStopWorker calls SimulateWorkerStoped | 77 // - OnStopWorker calls SimulateWorkerStoped |
| 76 // - OnSendMessageToWorker calls the message's respective On*Event handler | 78 // - OnSendMessageToWorker calls the message's respective On*Event handler |
| 77 virtual void OnStartWorker(int embedded_worker_id, | 79 virtual void OnStartWorker(int embedded_worker_id, |
| 78 int64 service_worker_version_id, | 80 int64 service_worker_version_id, |
| 79 const GURL& scope, | 81 const GURL& scope, |
| 80 const GURL& script_url); | 82 const GURL& script_url, |
| 83 bool pause_after_download); |
| 84 virtual void OnResumeAfterDownload(int embedded_worker_id); |
| 81 virtual void OnStopWorker(int embedded_worker_id); | 85 virtual void OnStopWorker(int embedded_worker_id); |
| 82 virtual bool OnMessageToWorker(int thread_id, | 86 virtual bool OnMessageToWorker(int thread_id, |
| 83 int embedded_worker_id, | 87 int embedded_worker_id, |
| 84 const IPC::Message& message); | 88 const IPC::Message& message); |
| 85 | 89 |
| 86 // On*Event handlers. Called by the default implementation of | 90 // On*Event handlers. Called by the default implementation of |
| 87 // OnMessageToWorker when events are sent to the embedded | 91 // OnMessageToWorker when events are sent to the embedded |
| 88 // worker. By default they just return success via | 92 // worker. By default they just return success via |
| 89 // SimulateSendReplyToBrowser. | 93 // SimulateSendReplyToBrowser. |
| 90 virtual void OnActivateEvent(int embedded_worker_id, int request_id); | 94 virtual void OnActivateEvent(int embedded_worker_id, int request_id); |
| 91 virtual void OnInstallEvent(int embedded_worker_id, | 95 virtual void OnInstallEvent(int embedded_worker_id, |
| 92 int request_id, | 96 int request_id, |
| 93 int active_version_id); | 97 int active_version_id); |
| 94 virtual void OnFetchEvent(int embedded_worker_id, | 98 virtual void OnFetchEvent(int embedded_worker_id, |
| 95 int request_id, | 99 int request_id, |
| 96 const ServiceWorkerFetchRequest& request); | 100 const ServiceWorkerFetchRequest& request); |
| 97 | 101 |
| 98 // These functions simulate sending an EmbeddedHostMsg message to the | 102 // These functions simulate sending an EmbeddedHostMsg message to the |
| 99 // browser. | 103 // browser. |
| 104 void SimulatePausedAfterDownload(int embedded_worker_id); |
| 105 void SimulateWorkerScriptLoaded(int embedded_worker_id); |
| 100 void SimulateWorkerStarted(int thread_id, int embedded_worker_id); | 106 void SimulateWorkerStarted(int thread_id, int embedded_worker_id); |
| 101 void SimulateWorkerStopped(int embedded_worker_id); | 107 void SimulateWorkerStopped(int embedded_worker_id); |
| 102 void SimulateSend(IPC::Message* message); | 108 void SimulateSend(IPC::Message* message); |
| 103 | 109 |
| 104 protected: | |
| 105 EmbeddedWorkerRegistry* registry(); | 110 EmbeddedWorkerRegistry* registry(); |
| 106 | 111 |
| 107 private: | 112 private: |
| 108 void OnStartWorkerStub(const EmbeddedWorkerMsg_StartWorker_Params& params); | 113 void OnStartWorkerStub(const EmbeddedWorkerMsg_StartWorker_Params& params); |
| 114 void OnResumeAfterDownloadStub(int embedded_worker_id); |
| 109 void OnStopWorkerStub(int embedded_worker_id); | 115 void OnStopWorkerStub(int embedded_worker_id); |
| 110 void OnMessageToWorkerStub(int thread_id, | 116 void OnMessageToWorkerStub(int thread_id, |
| 111 int embedded_worker_id, | 117 int embedded_worker_id, |
| 112 const IPC::Message& message); | 118 const IPC::Message& message); |
| 113 void OnActivateEventStub(int request_id); | 119 void OnActivateEventStub(int request_id); |
| 114 void OnInstallEventStub(int request_id, int active_version_id); | 120 void OnInstallEventStub(int request_id, int active_version_id); |
| 115 void OnFetchEventStub(int request_id, | 121 void OnFetchEventStub(int request_id, |
| 116 const ServiceWorkerFetchRequest& request); | 122 const ServiceWorkerFetchRequest& request); |
| 117 | 123 |
| 118 scoped_refptr<ServiceWorkerContextWrapper> wrapper_; | 124 scoped_refptr<ServiceWorkerContextWrapper> wrapper_; |
| 119 | 125 |
| 120 IPC::TestSink sink_; | 126 IPC::TestSink sink_; |
| 121 IPC::TestSink inner_sink_; | 127 IPC::TestSink inner_sink_; |
| 122 | 128 |
| 123 int next_thread_id_; | 129 int next_thread_id_; |
| 130 int mock_render_process_id_; |
| 124 | 131 |
| 125 // Updated each time MessageToWorker message is received. | 132 // Updated each time MessageToWorker message is received. |
| 126 int current_embedded_worker_id_; | 133 int current_embedded_worker_id_; |
| 127 | 134 |
| 128 base::WeakPtrFactory<EmbeddedWorkerTestHelper> weak_factory_; | 135 base::WeakPtrFactory<EmbeddedWorkerTestHelper> weak_factory_; |
| 129 | 136 |
| 130 DISALLOW_COPY_AND_ASSIGN(EmbeddedWorkerTestHelper); | 137 DISALLOW_COPY_AND_ASSIGN(EmbeddedWorkerTestHelper); |
| 131 }; | 138 }; |
| 132 | 139 |
| 133 } // namespace content | 140 } // namespace content |
| 134 | 141 |
| 135 #endif // CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_TEST_HELPER_H_ | 142 #endif // CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_TEST_HELPER_H_ |
| OLD | NEW |