Get signals working in the EXTRA_DATA section of MHTML

content/browser/download/mhtml_generation_browsertest.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 <stdint.h>
 #include <memory>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/histogram_tester.h"
 #include "base/threading/thread_restrictions.h"
+#include "content/browser/download/mhtml_generation_manager.h"
 #include "content/browser/renderer_host/render_process_host_impl.h"
 #include "content/common/frame_messages.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/mhtml_generation_params.h"
 #include "content/public/test/browser_test_utils.h"
 #include "content/public/test/content_browser_test.h"
 #include "content/public/test/content_browser_test_utils.h"
 #include "content/public/test/test_utils.h"
 #include "content/shell/browser/shell.h"
@@ skipping 637 matching lines @@
   {
     base::ThreadRestrictions::ScopedAllowIO allow_io_for_content_verification;
     ASSERT_TRUE(base::ReadFileToString(path, &mhtml));
   }
 
   // Make sure the overlay is removed.
   EXPECT_THAT(mhtml, Not(HasSubstr("class=3D\"overlay")));
   EXPECT_THAT(mhtml, Not(HasSubstr("class=3D\"modal")));
 }
 
+IN_PROC_BROWSER_TEST_F(MHTMLGenerationTest, GenerateMHTMLWithExtraData) {
+  const char kFakeSignalData1[] = "FakeSignalData1";
+  const char kFakeSignalData2[] = "OtherMockDataForSignals";
+  const char kFakeContentType[] = "text/plain";
+  const char kFakeContentLocation[] =
+      "cid:signal-data-62691-645341c4-62b3-478e-a8c5-e0dfccc3ca02@mhtml.blink";
+  base::FilePath path(temp_dir_.GetPath());
+  path = path.Append(FILE_PATH_LITERAL("test.mht"));
+  GURL url(embedded_test_server()->GetURL("/page_with_image.html"));
+  MHTMLGenerationParams params(path);
+
+  // Place the extra data we need into the web contents user data.
+  std::string signal_data1(kFakeSignalData1);
+  std::string signal_data2(kFakeSignalData2);
+  std::string content_type(kFakeContentType);
+  std::string content_location(kFakeContentLocation);
+  content::MHTMLGenerationManager::MHTMLExtraPart part1;
+  part1.body = signal_data1;
+  part1.content_type = content_type;
+  part1.content_location = content_location;
+  content::MHTMLGenerationManager::MHTMLExtraPart part2;
+  part2.body = signal_data2;
+  part2.content_type = content_type;
+  part2.content_location = content_location;
+
+  // Add two extra data parts to the MHTML.
+  MHTMLGenerationManager::MHTMLExtraData::AddPartToWebContents(
+      shell()->web_contents(), part1);
+  MHTMLGenerationManager::MHTMLExtraData::AddPartToWebContents(
+      shell()->web_contents(), part2);
+  GenerateMHTML(params, url);
+
+  EXPECT_TRUE(has_mhtml_callback_run());
+
+  std::string mhtml;
+  {
+    base::ThreadRestrictions::ScopedAllowIO allow_io_for_content_verification;
+    ASSERT_TRUE(base::ReadFileToString(path, &mhtml));
+  }
+
+  // Make sure that both extra data parts made it into the mhtml.
+  EXPECT_THAT(mhtml, HasSubstr(kFakeSignalData1));
+  EXPECT_THAT(mhtml, HasSubstr(kFakeSignalData2));
+}
+
 }  // namespace content