| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 <windows.h> | 5 #include <windows.h> |
| 6 | 6 |
| 7 #include "base/files/file_util.h" | 7 #include "base/files/file_util.h" |
| 8 #include "base/files/scoped_temp_dir.h" | 8 #include "base/files/scoped_temp_dir.h" |
| 9 #include "base/path_service.h" | 9 #include "base/path_service.h" |
| 10 #include "chrome/common/chrome_paths.h" | 10 #include "chrome/common/chrome_paths.h" |
| (...skipping 17 matching lines...) Expand all Loading... |
| 28 // The path to input data used in tests. | 28 // The path to input data used in tests. |
| 29 base::FilePath data_dir_; | 29 base::FilePath data_dir_; |
| 30 }; | 30 }; |
| 31 | 31 |
| 32 } // namespace | 32 } // namespace |
| 33 | 33 |
| 34 // Test that we can open archives successfully. | 34 // Test that we can open archives successfully. |
| 35 TEST_F(LzmaUtilTest, OpenArchiveTest) { | 35 TEST_F(LzmaUtilTest, OpenArchiveTest) { |
| 36 base::FilePath archive = data_dir_.AppendASCII("archive1.7z"); | 36 base::FilePath archive = data_dir_.AppendASCII("archive1.7z"); |
| 37 LzmaUtil lzma_util; | 37 LzmaUtil lzma_util; |
| 38 EXPECT_EQ(lzma_util.OpenArchive(archive.value()), NO_ERROR); | 38 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 39 lzma_util.OpenArchive(archive.value())); |
| 39 | 40 |
| 40 // We allow opening another archive (which will automatically close the first | 41 // We allow opening another archive (which will automatically close the first |
| 41 // archive). | 42 // archive). |
| 42 archive = data_dir_.AppendASCII("archive2.7z"); | 43 archive = data_dir_.AppendASCII("archive2.7z"); |
| 43 EXPECT_EQ(lzma_util.OpenArchive(archive.value()), NO_ERROR); | 44 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 45 lzma_util.OpenArchive(archive.value())); |
| 44 | 46 |
| 45 // Explicitly close and open the first archive again. | 47 // Explicitly close and open the first archive again. |
| 46 lzma_util.CloseArchive(); | 48 lzma_util.CloseArchive(); |
| 47 archive = data_dir_.AppendASCII("archive1.7z"); | 49 archive = data_dir_.AppendASCII("archive1.7z"); |
| 48 EXPECT_EQ(lzma_util.OpenArchive(archive.value()), NO_ERROR); | 50 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 51 lzma_util.OpenArchive(archive.value())); |
| 49 | 52 |
| 50 // Make sure non-existent archive returns error. | 53 // Make sure non-existent archive returns error. |
| 51 archive = data_dir_.AppendASCII("archive.non_existent.7z"); | 54 archive = data_dir_.AppendASCII("archive.non_existent.7z"); |
| 52 EXPECT_EQ(lzma_util.OpenArchive(archive.value()), ERROR_FILE_NOT_FOUND); | 55 EXPECT_EQ(static_cast<DWORD>(ERROR_FILE_NOT_FOUND), |
| 56 lzma_util.OpenArchive(archive.value())); |
| 53 } | 57 } |
| 54 | 58 |
| 55 // Test that we can extract archives successfully. | 59 // Test that we can extract archives successfully. |
| 56 TEST_F(LzmaUtilTest, UnPackTest) { | 60 TEST_F(LzmaUtilTest, UnPackTest) { |
| 57 base::FilePath extract_dir(temp_dir_.path()); | 61 base::FilePath extract_dir(temp_dir_.path()); |
| 58 extract_dir = extract_dir.AppendASCII("UnPackTest"); | 62 extract_dir = extract_dir.AppendASCII("UnPackTest"); |
| 59 ASSERT_FALSE(base::PathExists(extract_dir)); | 63 ASSERT_FALSE(base::PathExists(extract_dir)); |
| 60 EXPECT_TRUE(base::CreateDirectory(extract_dir)); | 64 EXPECT_TRUE(base::CreateDirectory(extract_dir)); |
| 61 ASSERT_TRUE(base::PathExists(extract_dir)); | 65 ASSERT_TRUE(base::PathExists(extract_dir)); |
| 62 | 66 |
| 63 base::FilePath archive = data_dir_.AppendASCII("archive1.7z"); | 67 base::FilePath archive = data_dir_.AppendASCII("archive1.7z"); |
| 64 LzmaUtil lzma_util; | 68 LzmaUtil lzma_util; |
| 65 EXPECT_EQ(lzma_util.OpenArchive(archive.value()), NO_ERROR); | 69 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 70 lzma_util.OpenArchive(archive.value())); |
| 66 std::wstring unpacked_file; | 71 std::wstring unpacked_file; |
| 67 EXPECT_EQ(lzma_util.UnPack(extract_dir.value(), &unpacked_file), | 72 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 68 NO_ERROR); | 73 lzma_util.UnPack(extract_dir.value(), &unpacked_file)); |
| 69 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("a.exe"))); | 74 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("a.exe"))); |
| 70 EXPECT_TRUE(unpacked_file == extract_dir.AppendASCII("a.exe").value()); | 75 EXPECT_TRUE(unpacked_file == extract_dir.AppendASCII("a.exe").value()); |
| 71 | 76 |
| 72 archive = data_dir_.AppendASCII("archive2.7z"); | 77 archive = data_dir_.AppendASCII("archive2.7z"); |
| 73 EXPECT_EQ(lzma_util.OpenArchive(archive.value()), NO_ERROR); | 78 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 74 EXPECT_EQ(lzma_util.UnPack(extract_dir.value(), &unpacked_file), | 79 lzma_util.OpenArchive(archive.value())); |
| 75 NO_ERROR); | 80 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 81 lzma_util.UnPack(extract_dir.value(), &unpacked_file)); |
| 76 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("b.exe"))); | 82 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("b.exe"))); |
| 77 EXPECT_TRUE(unpacked_file == extract_dir.AppendASCII("b.exe").value()); | 83 EXPECT_TRUE(unpacked_file == extract_dir.AppendASCII("b.exe").value()); |
| 78 | 84 |
| 79 lzma_util.CloseArchive(); | 85 lzma_util.CloseArchive(); |
| 80 archive = data_dir_.AppendASCII("invalid_archive.7z"); | 86 archive = data_dir_.AppendASCII("invalid_archive.7z"); |
| 81 EXPECT_EQ(lzma_util.UnPack(extract_dir.value(), &unpacked_file), | 87 EXPECT_EQ(static_cast<DWORD>(ERROR_INVALID_HANDLE), |
| 82 ERROR_INVALID_HANDLE); | 88 lzma_util.UnPack(extract_dir.value(), &unpacked_file)); |
| 83 EXPECT_EQ(lzma_util.OpenArchive(archive.value()), NO_ERROR); | 89 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 84 EXPECT_EQ(lzma_util.UnPack(extract_dir.value(), &unpacked_file), | 90 lzma_util.OpenArchive(archive.value())); |
| 85 ERROR_INVALID_HANDLE); | 91 EXPECT_EQ(static_cast<DWORD>(ERROR_INVALID_HANDLE), |
| 92 lzma_util.UnPack(extract_dir.value(), &unpacked_file)); |
| 86 | 93 |
| 87 archive = data_dir_.AppendASCII("archive3.7z"); | 94 archive = data_dir_.AppendASCII("archive3.7z"); |
| 88 EXPECT_EQ(lzma_util.OpenArchive(archive.value()), NO_ERROR); | 95 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 89 EXPECT_EQ(lzma_util.UnPack(extract_dir.value(), &unpacked_file), | 96 lzma_util.OpenArchive(archive.value())); |
| 90 NO_ERROR); | 97 EXPECT_EQ(static_cast<DWORD>(NO_ERROR), |
| 98 lzma_util.UnPack(extract_dir.value(), &unpacked_file)); |
| 91 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("archive\\a.exe"))); | 99 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("archive\\a.exe"))); |
| 92 EXPECT_TRUE(base::PathExists( | 100 EXPECT_TRUE(base::PathExists( |
| 93 extract_dir.AppendASCII("archive\\sub_dir\\text.txt"))); | 101 extract_dir.AppendASCII("archive\\sub_dir\\text.txt"))); |
| 94 } | 102 } |
| 95 | 103 |
| 96 // Test the static method that can be used to unpack archives. | 104 // Test the static method that can be used to unpack archives. |
| 97 TEST_F(LzmaUtilTest, UnPackArchiveTest) { | 105 TEST_F(LzmaUtilTest, UnPackArchiveTest) { |
| 98 base::FilePath extract_dir(temp_dir_.path()); | 106 base::FilePath extract_dir(temp_dir_.path()); |
| 99 extract_dir = extract_dir.AppendASCII("UnPackArchiveTest"); | 107 extract_dir = extract_dir.AppendASCII("UnPackArchiveTest"); |
| 100 ASSERT_FALSE(base::PathExists(extract_dir)); | 108 ASSERT_FALSE(base::PathExists(extract_dir)); |
| 101 EXPECT_TRUE(base::CreateDirectory(extract_dir)); | 109 EXPECT_TRUE(base::CreateDirectory(extract_dir)); |
| 102 ASSERT_TRUE(base::PathExists(extract_dir)); | 110 ASSERT_TRUE(base::PathExists(extract_dir)); |
| 103 | 111 |
| 104 base::FilePath archive = data_dir_.AppendASCII("archive1.7z"); | 112 base::FilePath archive = data_dir_.AppendASCII("archive1.7z"); |
| 105 std::wstring unpacked_file; | 113 std::wstring unpacked_file; |
| 106 EXPECT_EQ(LzmaUtil::UnPackArchive(archive.value(), extract_dir.value(), | 114 EXPECT_EQ(NO_ERROR, |
| 107 &unpacked_file), NO_ERROR); | 115 LzmaUtil::UnPackArchive(archive.value(), extract_dir.value(), |
| 116 &unpacked_file)); |
| 108 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("a.exe"))); | 117 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("a.exe"))); |
| 109 EXPECT_TRUE(unpacked_file == extract_dir.AppendASCII("a.exe").value()); | 118 EXPECT_TRUE(unpacked_file == extract_dir.AppendASCII("a.exe").value()); |
| 110 | 119 |
| 111 archive = data_dir_.AppendASCII("archive2.7z"); | 120 archive = data_dir_.AppendASCII("archive2.7z"); |
| 112 EXPECT_EQ(LzmaUtil::UnPackArchive(archive.value(), extract_dir.value(), | 121 EXPECT_EQ(NO_ERROR, |
| 113 &unpacked_file), NO_ERROR); | 122 LzmaUtil::UnPackArchive(archive.value(), extract_dir.value(), |
| 123 &unpacked_file)); |
| 114 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("b.exe"))); | 124 EXPECT_TRUE(base::PathExists(extract_dir.AppendASCII("b.exe"))); |
| 115 EXPECT_TRUE(unpacked_file == extract_dir.AppendASCII("b.exe").value()); | 125 EXPECT_TRUE(unpacked_file == extract_dir.AppendASCII("b.exe").value()); |
| 116 | 126 |
| 117 archive = data_dir_.AppendASCII("invalid_archive.7z"); | 127 archive = data_dir_.AppendASCII("invalid_archive.7z"); |
| 118 EXPECT_NE(LzmaUtil::UnPackArchive(archive.value(), extract_dir.value(), | 128 EXPECT_NE(NO_ERROR, |
| 119 &unpacked_file), NO_ERROR); | 129 LzmaUtil::UnPackArchive(archive.value(), extract_dir.value(), |
| 130 &unpacked_file)); |
| 120 } | 131 } |
| OLD | NEW |