Implement CachedMetadata handling on MojoAsyncResourceHandler

content/browser/loader/test_url_loader_client.cc

Index: content/browser/loader/test_url_loader_client.cc
diff --git a/content/browser/loader/test_url_loader_client.cc b/content/browser/loader/test_url_loader_client.cc
index d87ee132ce21179e89e940a4f7ef35d9ac3a8d61..552501936d26bdde6ab7719baf60800762d33154 100644
--- a/content/browser/loader/test_url_loader_client.cc
+++ b/content/browser/loader/test_url_loader_client.cc
@@ -45,6 +45,14 @@ void TestURLLoaderClient::OnDataDownloaded(int64_t data_length,
     quit_closure_for_on_data_downloaded_.Run();
 }
 
+void TestURLLoaderClient::OnReceiveCachedMetadata(
+    const std::vector<uint8_t>& data) {
+  has_received_cached_metadata_ = true;
+  cached_metadata_ = data;
+  if (quit_closure_for_on_receive_cached_metadata_)
+    quit_closure_for_on_receive_cached_metadata_.Run();
+}
+
 void TestURLLoaderClient::OnTransferSizeUpdated(int32_t transfer_size_diff) {
   EXPECT_GT(transfer_size_diff, 0);
   body_transfer_size_ += transfer_size_diff;
@@ -83,6 +91,8 @@ void TestURLLoaderClient::Unbind() {
 }
 
 void TestURLLoaderClient::RunUntilResponseReceived() {
+  if (has_received_response_)
+    return;
   base::RunLoop run_loop;
   quit_closure_for_on_receive_response_ = run_loop.QuitClosure();
   run_loop.Run();
@@ -90,6 +100,8 @@ void TestURLLoaderClient::RunUntilResponseReceived() {
 }
 
 void TestURLLoaderClient::RunUntilRedirectReceived() {
+  if (has_received_redirect_)
+    return;
   base::RunLoop run_loop;
   quit_closure_for_on_receive_redirect_ = run_loop.QuitClosure();
   run_loop.Run();
@@ -97,12 +109,23 @@ void TestURLLoaderClient::RunUntilRedirectReceived() {
 }
 
 void TestURLLoaderClient::RunUntilDataDownloaded() {
+  if (has_data_downloaded_)
+    return;
   base::RunLoop run_loop;
   quit_closure_for_on_data_downloaded_ = run_loop.QuitClosure();
   run_loop.Run();
   quit_closure_for_on_data_downloaded_.Reset();
 }
 
+void TestURLLoaderClient::RunUntilCachedMetadataReceived() {
+  if (has_received_cached_metadata_)
+    return;
+  base::RunLoop run_loop;
+  quit_closure_for_on_receive_cached_metadata_ = run_loop.QuitClosure();
+  run_loop.Run();
+  quit_closure_for_on_receive_cached_metadata_.Reset();
+}
+
 void TestURLLoaderClient::RunUntilResponseBodyArrived() {
   if (response_body_.is_valid())
     return;