OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <stdint.h> | 5 #include <stdint.h> |
6 #include <memory> | 6 #include <memory> |
7 | 7 |
8 #include "base/bind.h" | 8 #include "base/bind.h" |
9 #include "base/callback.h" | 9 #include "base/callback.h" |
10 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
(...skipping 669 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
680 const char kFakeContentLocation[] = | 680 const char kFakeContentLocation[] = |
681 "cid:signal-data-62691-645341c4-62b3-478e-a8c5-e0dfccc3ca02@mhtml.blink"; | 681 "cid:signal-data-62691-645341c4-62b3-478e-a8c5-e0dfccc3ca02@mhtml.blink"; |
682 base::FilePath path(temp_dir_.GetPath()); | 682 base::FilePath path(temp_dir_.GetPath()); |
683 path = path.Append(FILE_PATH_LITERAL("test.mht")); | 683 path = path.Append(FILE_PATH_LITERAL("test.mht")); |
684 GURL url(embedded_test_server()->GetURL("/page_with_image.html")); | 684 GURL url(embedded_test_server()->GetURL("/page_with_image.html")); |
685 MHTMLGenerationParams params(path); | 685 MHTMLGenerationParams params(path); |
686 | 686 |
687 // Place the extra data we need into the web contents user data. | 687 // Place the extra data we need into the web contents user data. |
688 std::string content_type(kFakeContentType); | 688 std::string content_type(kFakeContentType); |
689 std::string content_location(kFakeContentLocation); | 689 std::string content_location(kFakeContentLocation); |
| 690 std::string extra_headers; |
690 | 691 |
691 // Get the MHTMLExtraParts | 692 // Get the MHTMLExtraParts |
692 MHTMLExtraParts* extra_parts = | 693 MHTMLExtraParts* extra_parts = |
693 MHTMLExtraParts::FromWebContents(shell()->web_contents()); | 694 MHTMLExtraParts::FromWebContents(shell()->web_contents()); |
694 | 695 |
695 // Add two extra data parts to the MHTML. | 696 // Add two extra data parts to the MHTML. |
696 extra_parts->AddExtraMHTMLPart(content_type, content_location, | 697 extra_parts->AddExtraMHTMLPart(content_type, content_location, extra_headers, |
697 kFakeSignalData1); | 698 kFakeSignalData1); |
698 extra_parts->AddExtraMHTMLPart(content_type, content_location, | 699 extra_parts->AddExtraMHTMLPart(content_type, content_location, extra_headers, |
699 kFakeSignalData2); | 700 kFakeSignalData2); |
700 EXPECT_EQ(extra_parts->size(), 2); | 701 EXPECT_EQ(extra_parts->size(), 2); |
701 GenerateMHTML(params, url); | 702 GenerateMHTML(params, url); |
702 | 703 |
703 EXPECT_TRUE(has_mhtml_callback_run()); | 704 EXPECT_TRUE(has_mhtml_callback_run()); |
704 | 705 |
705 std::string mhtml; | 706 std::string mhtml; |
706 { | 707 { |
707 base::ThreadRestrictions::ScopedAllowIO allow_io_for_content_verification; | 708 base::ThreadRestrictions::ScopedAllowIO allow_io_for_content_verification; |
708 ASSERT_TRUE(base::ReadFileToString(path, &mhtml)); | 709 ASSERT_TRUE(base::ReadFileToString(path, &mhtml)); |
709 } | 710 } |
710 | 711 |
711 // Make sure that both extra data parts made it into the mhtml. | 712 // Make sure that both extra data parts made it into the mhtml. |
712 EXPECT_THAT(mhtml, HasSubstr(kFakeSignalData1)); | 713 EXPECT_THAT(mhtml, HasSubstr(kFakeSignalData1)); |
713 EXPECT_THAT(mhtml, HasSubstr(kFakeSignalData2)); | 714 EXPECT_THAT(mhtml, HasSubstr(kFakeSignalData2)); |
714 } | 715 } |
715 | 716 |
716 } // namespace content | 717 } // namespace content |
OLD | NEW |