Expose WebServiceWorkerNetworkProvider on DataSource

third_party/WebKit/Source/web/WebEmbeddedWorkerImplTest.cpp

 // Copyright 2015 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 "public/web/WebEmbeddedWorker.h"
 
 #include <memory>
 #include "platform/testing/URLTestHelpers.h"
 #include "platform/testing/UnitTestHelpers.h"
 #include "public/platform/Platform.h"
@@ skipping 10 matching lines @@
 namespace blink {
 namespace {
 
 class MockServiceWorkerContextClient : public WebServiceWorkerContextClient {
  public:
   MockServiceWorkerContextClient() : m_hasAssociatedRegistration(true) {}
   ~MockServiceWorkerContextClient() override {}
   MOCK_METHOD0(workerReadyForInspection, void());
   MOCK_METHOD0(workerContextFailedToStart, void());
   MOCK_METHOD0(workerScriptLoaded, void());
-  MOCK_METHOD1(createServiceWorkerNetworkProvider,
-               WebServiceWorkerNetworkProvider*(WebDataSource*));
+  MOCK_METHOD0(createServiceWorkerNetworkProvider,
+               WebServiceWorkerNetworkProvider*());
   MOCK_METHOD0(createServiceWorkerProvider, WebServiceWorkerProvider*());
   bool hasAssociatedRegistration() override {
     return m_hasAssociatedRegistration;
   }
   void setHasAssociatedRegistration(bool hasAssociatedRegistration) {
     m_hasAssociatedRegistration = hasAssociatedRegistration;
   }
   void getClient(const WebString&,
                  std::unique_ptr<WebServiceWorkerClientCallbacks>) override {
     NOTREACHED();
@@ skipping 93 matching lines @@
   m_worker->terminateWorkerContext();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 }
 
 TEST_F(WebEmbeddedWorkerImplTest, TerminateWhileLoadingScript) {
   EXPECT_CALL(*m_mockClient, workerReadyForInspection()).Times(1);
   m_worker->startWorkerContext(m_startData);
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the shadow page.
-  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider(::testing::_))
+  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider())
       .WillOnce(::testing::Return(nullptr));
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Terminate before loading the script.
   EXPECT_CALL(*m_mockClient, workerContextFailedToStart()).Times(1);
   m_worker->terminateWorkerContext();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 }
 
 TEST_F(WebEmbeddedWorkerImplTest, TerminateWhilePausedAfterDownload) {
   EXPECT_CALL(*m_mockClient, workerReadyForInspection()).Times(1);
   m_startData.pauseAfterDownloadMode =
       WebEmbeddedWorkerStartData::PauseAfterDownload;
   m_worker->startWorkerContext(m_startData);
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the shadow page.
-  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider(::testing::_))
+  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider())
       .WillOnce(::testing::Return(nullptr));
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the script.
   EXPECT_CALL(*m_mockClient, workerScriptLoaded()).Times(1);
   EXPECT_CALL(*m_mockClient, createServiceWorkerProvider()).Times(0);
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
@@ skipping 15 matching lines @@
   error.domain = "WebEmbeddedWorkerImplTest";
   Platform::current()->getURLLoaderMockFactory()->registerErrorURL(
       scriptURL, response, error);
   m_startData.scriptURL = scriptURL;
 
   EXPECT_CALL(*m_mockClient, workerReadyForInspection()).Times(1);
   m_worker->startWorkerContext(m_startData);
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the shadow page.
-  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider(::testing::_))
+  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider())
       .WillOnce(::testing::Return(nullptr));
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the script.
   EXPECT_CALL(*m_mockClient, workerScriptLoaded()).Times(0);
   EXPECT_CALL(*m_mockClient, createServiceWorkerProvider()).Times(0);
   EXPECT_CALL(*m_mockClient, workerContextFailedToStart()).Times(1);
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 }
 
 TEST_F(WebEmbeddedWorkerImplTest, NoRegistration) {
   EXPECT_CALL(*m_mockClient, workerReadyForInspection()).Times(1);
   m_startData.pauseAfterDownloadMode =
       WebEmbeddedWorkerStartData::PauseAfterDownload;
   m_worker->startWorkerContext(m_startData);
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the shadow page.
-  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider(::testing::_))
+  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider())
       .WillOnce(::testing::Return(nullptr));
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the script.
   m_mockClient->setHasAssociatedRegistration(false);
   EXPECT_CALL(*m_mockClient, workerScriptLoaded()).Times(0);
   EXPECT_CALL(*m_mockClient, createServiceWorkerProvider()).Times(0);
   EXPECT_CALL(*m_mockClient, workerContextFailedToStart()).Times(1);
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 }
 
 // The running worker is detected as a memory leak. crbug.com/586897
 #if defined(ADDRESS_SANITIZER)
 #define MAYBE_DontPauseAfterDownload DISABLED_DontPauseAfterDownload
 #else
 #define MAYBE_DontPauseAfterDownload DontPauseAfterDownload
 #endif
 
 TEST_F(WebEmbeddedWorkerImplTest, MAYBE_DontPauseAfterDownload) {
   EXPECT_CALL(*m_mockClient, workerReadyForInspection()).Times(1);
   m_worker->startWorkerContext(m_startData);
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the shadow page.
-  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider(::testing::_))
+  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider())
       .WillOnce(::testing::Return(nullptr));
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the script.
   EXPECT_CALL(*m_mockClient, workerScriptLoaded()).Times(1);
   EXPECT_CALL(*m_mockClient, createServiceWorkerProvider())
       .WillOnce(::testing::Return(nullptr));
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 }
 
 // The running worker is detected as a memory leak. crbug.com/586897
 #if defined(ADDRESS_SANITIZER)
 #define MAYBE_PauseAfterDownload DISABLED_PauseAfterDownload
 #else
 #define MAYBE_PauseAfterDownload PauseAfterDownload
 #endif
 
 TEST_F(WebEmbeddedWorkerImplTest, MAYBE_PauseAfterDownload) {
   EXPECT_CALL(*m_mockClient, workerReadyForInspection()).Times(1);
   m_startData.pauseAfterDownloadMode =
       WebEmbeddedWorkerStartData::PauseAfterDownload;
   m_worker->startWorkerContext(m_startData);
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the shadow page.
-  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider(::testing::_))
+  EXPECT_CALL(*m_mockClient, createServiceWorkerNetworkProvider())
       .WillOnce(::testing::Return(nullptr));
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Load the script.
   EXPECT_CALL(*m_mockClient, workerScriptLoaded()).Times(1);
   EXPECT_CALL(*m_mockClient, createServiceWorkerProvider()).Times(0);
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 
   // Resume after download.
   EXPECT_CALL(*m_mockClient, createServiceWorkerProvider())
       .WillOnce(::testing::Return(nullptr));
   m_worker->resumeAfterDownload();
   ::testing::Mock::VerifyAndClearExpectations(m_mockClient);
 }
 
 }  // namespace blink