PlzNavigate: properly update NavigationHandle for 204/205 and download responses

content/browser/loader/resource_dispatcher_host_unittest.cc

 // Copyright (c) 2012 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 <stddef.h>
 
 #include <memory>
 #include <utility>
 #include <vector>
 
@@ skipping 2638 matching lines @@
     std::unique_ptr<NavigationURLLoader> loader = NavigationURLLoader::Create(
         browser_context_.get(), std::move(request_info), nullptr, nullptr,
         &delegate);
 
     // Wait until a response has been received and proceed with the response.
     KickOffRequest();
 
     // Return some data so that the request is identified as a download
     // and the proper resource handlers are created.
     EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage());
+    delegate.WaitForResponseStarted();
     base::RunLoop().RunUntilIdle();
 
     // The UI thread will be informed that the navigation failed with an error
     // code of ERR_ABORTED because the navigation turns out to be a download.
     // The navigation is aborted, but the request goes on as a download.
     EXPECT_EQ(delegate.net_error(), net::ERR_ABORTED);
     EXPECT_EQ(1, host_.pending_requests());
 
     // In PlzNavigate, the renderer cannot cancel the request directly.
     // However, cancelling by context should work.
@@ skipping 1266 matching lines @@
   return nullptr;
 }
 
 INSTANTIATE_TEST_CASE_P(
     ResourceDispatcherHostTests,
     ResourceDispatcherHostTest,
     testing::Values(TestConfig::kDefault,
                     TestConfig::kOptimizeIPCForSmallResourceEnabled));
 
 }  // namespace content