ServiceWorker: Mojofication of EWInstance::StopWorker

content/browser/service_worker/embedded_worker_test_helper.cc

 // Copyright 2014 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 "content/browser/service_worker/embedded_worker_test_helper.h"
 
 #include <map>
 #include <string>
 #include <utility>
 
@@ skipping 113 matching lines @@
 };
 
 EmbeddedWorkerTestHelper::MockEmbeddedWorkerInstanceClient::
     MockEmbeddedWorkerInstanceClient(
         base::WeakPtr<EmbeddedWorkerTestHelper> helper)
     : helper_(helper), binding_(this) {
   ON_CALL(*this, MockStartWorker(_))
       .WillByDefault(Invoke(
           this, &EmbeddedWorkerTestHelper::MockEmbeddedWorkerInstanceClient::
                     DefaultStartWorker));
+  ON_CALL(*this, MockStopWorker(_))
+      .WillByDefault(Invoke(
+          this, &EmbeddedWorkerTestHelper::MockEmbeddedWorkerInstanceClient::
+                    DefaultStopWorker));
 }
 
 EmbeddedWorkerTestHelper::MockEmbeddedWorkerInstanceClient::
     ~MockEmbeddedWorkerInstanceClient() {}
 
 void EmbeddedWorkerTestHelper::MockEmbeddedWorkerInstanceClient::
     DefaultStartWorker(mojom::EmbeddedWorkerStartWorkerParams* params) {
   if (!helper_)
     return;
 
   embedded_worker_id_ = params->embedded_worker_id;
 
   EmbeddedWorkerInstance* worker =
       helper_->registry()->GetWorker(params->embedded_worker_id);
   ASSERT_TRUE(worker != NULL);
   EXPECT_EQ(EmbeddedWorkerStatus::STARTING, worker->status());
 
   base::ThreadTaskRunnerHandle::Get()->PostTask(
       FROM_HERE, base::Bind(&EmbeddedWorkerTestHelper::OnStartWorker,
                             helper_->weak_factory_.GetWeakPtr(),
                             params->embedded_worker_id,
                             params->service_worker_version_id, params->scope,
                             params->script_url, params->pause_after_download));
 }
 
+void EmbeddedWorkerTestHelper::MockEmbeddedWorkerInstanceClient::
+    DefaultStopWorker(const StopWorkerCallback& callback) {
+  if (!helper_)
+    return;
+
+  ASSERT_TRUE(embedded_worker_id_);
+  EmbeddedWorkerInstance* worker =
+      helper_->registry()->GetWorker(embedded_worker_id_.value());
+  ASSERT_TRUE(worker != NULL);
+  EXPECT_EQ(EmbeddedWorkerStatus::STOPPING, worker->status());
+
+  callback.Run();
+}
+
 // static
 void EmbeddedWorkerTestHelper::MockEmbeddedWorkerInstanceClient::Bind(
     const base::WeakPtr<EmbeddedWorkerTestHelper>& helper,
     mojom::EmbeddedWorkerInstanceClientRequest request) {
   ASSERT_GE(helper->mock_instance_clients()->size(),
             helper->mock_instance_clients_next_);
   if (helper->mock_instance_clients()->size() ==
       helper->mock_instance_clients_next_) {
     helper->mock_instance_clients()->push_back(
         base::MakeUnique<NiceMock<MockEmbeddedWorkerInstanceClient>>(helper));
@@ skipping 376 matching lines @@
 
 MessagePortMessageFilter*
 EmbeddedWorkerTestHelper::NewMessagePortMessageFilter() {
   scoped_refptr<MessagePortMessageFilter> filter(
       new MockMessagePortMessageFilter);
   message_port_message_filters_.push_back(filter);
   return filter.get();
 }
 
 }  // namespace content