1 files changed, 6 insertions, 6 deletions
diff --git a/chromium/components/pdf/browser/plugin_response_writer_unittest.cc b/chromium/components/pdf/browser/plugin_response_writer_unittest.cc
index 19f59446d2b..6cc52a98533 100644
--- a/chromium/components/pdf/browser/plugin_response_writer_unittest.cc
+++ b/chromium/components/pdf/browser/plugin_response_writer_unittest.cc
@@ -79,8 +79,9 @@ class BodyDrainer {
 class PluginResponseWriterTest : public testing::Test {
  protected:
   PluginResponseWriterTest() {
-    ON_CALL(mock_client_, OnStartLoadingResponseBody)
-        .WillByDefault([this](mojo::ScopedDataPipeConsumerHandle body) {
+    ON_CALL(mock_client_, OnReceiveResponse)
+        .WillByDefault([this](network::mojom::URLResponseHeadPtr head,
+                              mojo::ScopedDataPipeConsumerHandle body) {
           body_drainer_ = std::make_unique<BodyDrainer>(std::move(body));
         });
   }
@@ -139,14 +140,13 @@ TEST_F(PluginResponseWriterTest, Start) {
     testing::InSequence in_sequence;
 
     EXPECT_CALL(mock_client_, OnReceiveResponse)
-        .WillOnce([](network::mojom::URLResponseHeadPtr head,
-                     mojo::ScopedDataPipeConsumerHandle body) {
+        .WillOnce([this](network::mojom::URLResponseHeadPtr head,
+                         mojo::ScopedDataPipeConsumerHandle body) {
           EXPECT_EQ(200, head->headers->response_code());
           EXPECT_EQ("text/html", head->mime_type);
+          body_drainer_ = std::make_unique<BodyDrainer>(std::move(body));
         });
 
-    EXPECT_CALL(mock_client_, OnStartLoadingResponseBody);
-
     EXPECT_CALL(mock_client_, OnComplete)
         .WillOnce(
             [&run_loop](const network::URLLoaderCompletionStatus& status) {