Index: third_party/zlib/google/zip_reader_unittest.cc |
diff --git a/third_party/zlib/google/zip_reader_unittest.cc b/third_party/zlib/google/zip_reader_unittest.cc |
index 3056aba8ab855d03c91d35c71e7da24db721bdaa..76122f9af11100f8bc0b0e52bfe056d67fdf67a7 100644 |
--- a/third_party/zlib/google/zip_reader_unittest.cc |
+++ b/third_party/zlib/google/zip_reader_unittest.cc |
@@ -588,6 +588,47 @@ TEST_F(ZipReaderTest, ExtractCurrentEntryToString) { |
reader.Close(); |
} |
+TEST_F(ZipReaderTest, ExtractPartOfCurrentEntryToString) { |
+ // test_mismatch_size.zip contains files with names from 0.txt to 7.txt with |
+ // sizes from 0 to 7 bytes respectively, being the contents of each file a |
+ // substring of "0123456" starting at '0'. |
+ base::FilePath test_zip_file = |
+ test_data_dir_.AppendASCII("test_mismatch_size.zip"); |
+ |
+ ZipReader reader; |
+ std::string contents; |
+ ASSERT_TRUE(reader.Open(test_zip_file)); |
+ |
+ base::FilePath file_name0 = base::FilePath::FromUTF8Unsafe("0.txt"); |
+ ASSERT_TRUE(reader.LocateAndOpenEntry(file_name0)); |
+ EXPECT_TRUE(reader.ExtractPartOfCurrentEntryToString(0, &contents)); |
+ EXPECT_EQ(0, memcmp(contents.c_str(), "", 0)); |
+ EXPECT_FALSE(reader.ExtractPartOfCurrentEntryToString(1, &contents)); |
+ EXPECT_EQ(0, memcmp(contents.c_str(), "", 0)); |
+ |
+ base::FilePath file_name1 = base::FilePath::FromUTF8Unsafe("1.txt"); |
+ ASSERT_TRUE(reader.LocateAndOpenEntry(file_name1)); |
+ EXPECT_TRUE(reader.ExtractPartOfCurrentEntryToString(0, &contents)); |
+ EXPECT_EQ(0, memcmp(contents.c_str(), "", 0)); |
+ EXPECT_TRUE(reader.ExtractPartOfCurrentEntryToString(1, &contents)); |
+ EXPECT_EQ(0, memcmp(contents.c_str(), "0", 1)); |
+ EXPECT_FALSE(reader.ExtractPartOfCurrentEntryToString(2, &contents)); |
+ EXPECT_EQ(0, memcmp(contents.c_str(), "", 0)); |
+ |
+ base::FilePath file_name4 = base::FilePath::FromUTF8Unsafe("4.txt"); |
+ ASSERT_TRUE(reader.LocateAndOpenEntry(file_name4)); |
+ EXPECT_TRUE(reader.ExtractPartOfCurrentEntryToString(0, &contents)); |
+ EXPECT_EQ(0, memcmp(contents.c_str(), "", 0)); |
+ EXPECT_TRUE(reader.ExtractPartOfCurrentEntryToString(2, &contents)); |
+ EXPECT_EQ(0, memcmp(contents.c_str(), "01", 2)); |
+ EXPECT_TRUE(reader.ExtractPartOfCurrentEntryToString(4, &contents)); |
+ EXPECT_EQ(0, memcmp(contents.c_str(), "0123", 4)); |
+ EXPECT_FALSE(reader.ExtractPartOfCurrentEntryToString(5, &contents)); |
+ EXPECT_EQ(0, memcmp(contents.c_str(), "", 0)); |
+ |
+ reader.Close(); |
+} |
+ |
// This test exposes http://crbug.com/430959, at least on OS X |
TEST_F(ZipReaderTest, DISABLED_LeakDetectionTest) { |
for (int i = 0; i < 100000; ++i) { |
@@ -648,6 +689,53 @@ TEST_F(ZipReaderTest, ExtractCurrentEntrySuccess) { |
ASSERT_TRUE(reader.ExtractCurrentEntry(&mock_writer)); |
} |
+// Test that when WriterDelegate::PrepareMock returns false, no other methods on |
+// the delegate are called and the partial extraction fails. |
+TEST_F(ZipReaderTest, ExtractPartOfCurrentEntryPrepareFailure) { |
+ testing::StrictMock<MockWriterDelegate> mock_writer; |
+ |
+ EXPECT_CALL(mock_writer, PrepareOutput()).WillOnce(Return(false)); |
+ |
+ base::FilePath target_path(FILE_PATH_LITERAL("foo/bar/quux.txt")); |
+ ZipReader reader; |
+ |
+ ASSERT_TRUE(reader.Open(test_zip_file_)); |
+ ASSERT_TRUE(reader.LocateAndOpenEntry(target_path)); |
+ ASSERT_FALSE(reader.ExtractPartOfCurrentEntry(&mock_writer, 1)); |
+} |
+ |
+// Test that when WriterDelegate::WriteBytes returns false, no other methods on |
+// the delegate are called and the partial extraction fails. |
+TEST_F(ZipReaderTest, ExtractPartOfCurrentEntryWriteBytesFailure) { |
+ testing::StrictMock<MockWriterDelegate> mock_writer; |
+ |
+ EXPECT_CALL(mock_writer, PrepareOutput()).WillOnce(Return(true)); |
+ EXPECT_CALL(mock_writer, WriteBytes(_, _)).WillOnce(Return(false)); |
+ |
+ base::FilePath target_path(FILE_PATH_LITERAL("foo/bar/quux.txt")); |
+ ZipReader reader; |
+ |
+ ASSERT_TRUE(reader.Open(test_zip_file_)); |
+ ASSERT_TRUE(reader.LocateAndOpenEntry(target_path)); |
+ ASSERT_FALSE(reader.ExtractPartOfCurrentEntry(&mock_writer, 1)); |
+} |
+ |
+// Test that partial extraction succeeds when the writer delegate reports all is |
+// well. |
+TEST_F(ZipReaderTest, ExtractPartOfCurrentEntrySuccess) { |
+ testing::StrictMock<MockWriterDelegate> mock_writer; |
+ |
+ EXPECT_CALL(mock_writer, PrepareOutput()).WillOnce(Return(true)); |
+ EXPECT_CALL(mock_writer, WriteBytes(_, _)).WillRepeatedly(Return(true)); |
+ |
+ base::FilePath target_path(FILE_PATH_LITERAL("foo/bar/quux.txt")); |
+ ZipReader reader; |
+ |
+ ASSERT_TRUE(reader.Open(test_zip_file_)); |
+ ASSERT_TRUE(reader.LocateAndOpenEntry(target_path)); |
+ ASSERT_TRUE(reader.ExtractPartOfCurrentEntry(&mock_writer, 1)); |
+} |
+ |
class FileWriterDelegateTest : public ::testing::Test { |
protected: |
void SetUp() override { |