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 |