]> git.saurik.com Git - wxWidgets.git/blob - tests/archive/ziptest.cpp
reSWIGged
[wxWidgets.git] / tests / archive / ziptest.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: tests/archive/ziptest.cpp
3 // Purpose: Test the zip classes
4 // Author: Mike Wetherell
5 // RCS-ID: $Id$
6 // Copyright: (c) 2004 Mike Wetherell
7 // Licence: wxWindows licence
8 ///////////////////////////////////////////////////////////////////////////////
9
10 #include "testprec.h"
11
12 #ifdef __BORLANDC__
13 # pragma hdrstop
14 #endif
15
16 #ifndef WX_PRECOMP
17 # include "wx/wx.h"
18 #endif
19
20 #if wxUSE_STREAMS && wxUSE_ZIPSTREAM
21
22 #include "archivetest.h"
23 #include "wx/zipstrm.h"
24
25 using std::string;
26 using std::auto_ptr;
27
28
29 ///////////////////////////////////////////////////////////////////////////////
30 // ArchiveTestCase<wxZipClassFactory> could be used directly, but instead this
31 // derived class is used so that zip specific features can be tested.
32
33 class ZipTestCase : public ArchiveTestCase<wxZipClassFactory>
34 {
35 public:
36 ZipTestCase(string name,
37 int options,
38 const wxString& archiver = wxEmptyString,
39 const wxString& unarchiver = wxEmptyString)
40 :
41 ArchiveTestCase<wxZipClassFactory>(name, new wxZipClassFactory,
42 options, archiver, unarchiver),
43 m_count(0)
44 { }
45
46 protected:
47 void OnCreateArchive(wxZipOutputStream& zip);
48
49 void OnArchiveExtracted(wxZipInputStream& zip, int expectedTotal);
50
51 void OnCreateEntry(wxZipOutputStream& zip,
52 TestEntry& testEntry,
53 wxZipEntry *entry);
54
55 void OnEntryExtracted(wxZipEntry& entry,
56 const TestEntry& testEntry,
57 wxZipInputStream *arc);
58
59 void OnSetNotifier(EntryT& entry);
60
61 int m_count;
62 wxString m_comment;
63 };
64
65 void ZipTestCase::OnCreateArchive(wxZipOutputStream& zip)
66 {
67 m_comment << _T("Comment for test ") << m_id;
68 zip.SetComment(m_comment);
69 }
70
71 void ZipTestCase::OnArchiveExtracted(wxZipInputStream& zip, int expectedTotal)
72 {
73 CPPUNIT_ASSERT(zip.GetComment() == m_comment);
74 CPPUNIT_ASSERT(zip.GetTotalEntries() == expectedTotal);
75 }
76
77 void ZipTestCase::OnCreateEntry(wxZipOutputStream& zip,
78 TestEntry& testEntry,
79 wxZipEntry *entry)
80 {
81 zip.SetLevel((m_id + m_count) % 10);
82
83 if (entry) {
84 switch ((m_id + m_count) % 5) {
85 case 0:
86 {
87 wxString comment = _T("Comment for ") + entry->GetName();
88 entry->SetComment(comment);
89 // lowercase the expected result, and the notifier should do
90 // the same for the zip entries when ModifyArchive() runs
91 testEntry.SetComment(comment.Lower());
92 break;
93 }
94 case 2:
95 entry->SetMethod(wxZIP_METHOD_STORE);
96 break;
97 case 4:
98 entry->SetMethod(wxZIP_METHOD_DEFLATE);
99 break;
100 }
101 entry->SetIsText(testEntry.IsText());
102 }
103
104 m_count++;
105 }
106
107 void ZipTestCase::OnEntryExtracted(wxZipEntry& entry,
108 const TestEntry& testEntry,
109 wxZipInputStream *arc)
110 {
111 // provide some context for the error message so that we know which
112 // iteration of the loop we were on
113 wxString name = _T(" '") + entry.GetName() + _T("'");
114 string error_entry(name.mb_str());
115 string error_context(" failed for entry" + error_entry);
116
117 CPPUNIT_ASSERT_MESSAGE("GetComment" + error_context,
118 entry.GetComment() == testEntry.GetComment());
119
120 // for seekable streams, GetNextEntry() doesn't read the local header so
121 // call OpenEntry() to do it
122 if (arc && (m_options & PipeIn) == 0 && entry.IsDir())
123 arc->OpenEntry(entry);
124
125 CPPUNIT_ASSERT_MESSAGE("IsText" + error_context,
126 entry.IsText() == testEntry.IsText());
127
128 CPPUNIT_ASSERT_MESSAGE("Extra/LocalExtra mismatch for entry" + error_entry,
129 (entry.GetExtraLen() != 0 && entry.GetLocalExtraLen() != 0) ||
130 (entry.GetExtraLen() == 0 && entry.GetLocalExtraLen() == 0));
131 }
132
133 // check the notifier mechanism by using it to fold the entry comments to
134 // lowercase
135 //
136 class ZipNotifier : public wxZipNotifier
137 {
138 public:
139 void OnEntryUpdated(wxZipEntry& entry);
140 };
141
142 void ZipNotifier::OnEntryUpdated(wxZipEntry& entry)
143 {
144 entry.SetComment(entry.GetComment().Lower());
145 }
146
147 void ZipTestCase::OnSetNotifier(EntryT& entry)
148 {
149 static ZipNotifier notifier;
150 entry.SetNotifier(notifier);
151 }
152
153
154 ///////////////////////////////////////////////////////////////////////////////
155 // 'zip - -' produces local headers without the size field set. This is a
156 // case not covered by all the other tests, so this class tests it as a
157 // special case
158
159 class ZipPipeTestCase : public CppUnit::TestCase
160 {
161 public:
162 ZipPipeTestCase(string name, int options) :
163 CppUnit::TestCase(TestId::MakeId() + name),
164 m_options(options),
165 m_id(TestId::GetId())
166 { }
167
168 protected:
169 void runTest();
170 int m_options;
171 int m_id;
172 };
173
174 void ZipPipeTestCase::runTest()
175 {
176 TestOutputStream out(m_options);
177
178 wxString testdata = _T("test data to pipe through zip");
179 wxString cmd = _T("echo ") + testdata + _T(" | zip -q - -");
180
181 {
182 PFileInputStream in(cmd);
183 if (in.Ok())
184 out.Write(in);
185 }
186
187 TestInputStream in(out, m_id % ((m_options & PipeIn) ? 4 : 3));
188 wxZipInputStream zip(in);
189
190 auto_ptr<wxZipEntry> entry(zip.GetNextEntry());
191 CPPUNIT_ASSERT(entry.get() != NULL);
192
193 if ((m_options & PipeIn) == 0)
194 CPPUNIT_ASSERT(entry->GetSize() != wxInvalidOffset);
195
196 char buf[64];
197 size_t len = zip.Read(buf, sizeof(buf) - 1).LastRead();
198
199 while (len > 0 && buf[len - 1] <= 32)
200 --len;
201 buf[len] = 0;
202
203 CPPUNIT_ASSERT(zip.Eof());
204 CPPUNIT_ASSERT(wxString(buf, *wxConvCurrent) == testdata);
205 }
206
207
208 ///////////////////////////////////////////////////////////////////////////////
209 // Zip suite
210
211 class ziptest : public ArchiveTestSuite
212 {
213 public:
214 ziptest();
215 static CppUnit::Test *suite() { return (new ziptest)->makeSuite(); }
216
217 protected:
218 ArchiveTestSuite *makeSuite();
219
220 CppUnit::Test *makeTest(string descr, int options,
221 bool genericInterface, const wxString& archiver,
222 const wxString& unarchiver);
223 };
224
225 ziptest::ziptest()
226 : ArchiveTestSuite("zip")
227 {
228 AddArchiver(_T("zip -qr %s *"));
229 AddUnArchiver(_T("unzip -q %s"));
230 }
231
232 ArchiveTestSuite *ziptest::makeSuite()
233 {
234 ArchiveTestSuite::makeSuite();
235
236 #ifndef WXARC_NO_POPEN
237 // if have popen then can check the piped output of 'zip - -'
238 if (IsInPath(_T("zip")))
239 for (int options = 0; options <= PipeIn; options += PipeIn) {
240 string name = Description(_T("ZipPipeTestCase"), options,
241 false, _T(""), _T("zip -q - -"));
242 addTest(new ZipPipeTestCase(name, options));
243 }
244 #endif
245
246 return this;
247 }
248
249 CppUnit::Test *ziptest::makeTest(
250 string descr,
251 int options,
252 bool genericInterface,
253 const wxString& archiver,
254 const wxString& unarchiver)
255 {
256 // unzip doesn't support piping in the zip
257 if ((options & PipeIn) && !unarchiver.empty())
258 return NULL;
259
260 if (genericInterface)
261 return new ArchiveTestCase<wxArchiveClassFactory>(
262 descr, new wxZipClassFactory,
263 options, archiver, unarchiver);
264 else
265 return new ZipTestCase(descr, options, archiver, unarchiver);
266 }
267
268 CPPUNIT_TEST_SUITE_REGISTRATION(ziptest);
269 CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(ziptest, "archive");
270 CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(ziptest, "archive/zip");
271
272 #endif // wxUSE_STREAMS && wxUSE_ZIPSTREAM