]>
Commit | Line | Data |
---|---|---|
1 | /////////////////////////////////////////////////////////////////////////////// | |
2 | // Name: tests/filetype/filetype.cpp | |
3 | // Purpose: Test wxGetFileKind and wxStreamBase::IsSeekable | |
4 | // Author: Mike Wetherell | |
5 | // Copyright: (c) 2005 Mike Wetherell | |
6 | // Licence: wxWindows licence | |
7 | /////////////////////////////////////////////////////////////////////////////// | |
8 | ||
9 | #include "testprec.h" | |
10 | ||
11 | #ifdef __BORLANDC__ | |
12 | #pragma hdrstop | |
13 | #endif | |
14 | ||
15 | // for all others, include the necessary headers | |
16 | #ifndef WX_PRECOMP | |
17 | #include "wx/wx.h" | |
18 | #endif | |
19 | ||
20 | #if wxUSE_STREAMS | |
21 | ||
22 | #ifdef __UNIX__ | |
23 | #include <sys/socket.h> | |
24 | #endif | |
25 | ||
26 | #include "wx/file.h" | |
27 | #include "wx/ffile.h" | |
28 | #include "wx/wfstream.h" | |
29 | #include "wx/filename.h" | |
30 | #include "wx/socket.h" | |
31 | #include "wx/sckstrm.h" | |
32 | #include "wx/mstream.h" | |
33 | ||
34 | #ifdef __VISUALC__ | |
35 | #define isatty _isatty | |
36 | #define fdopen _fdopen | |
37 | #define fileno _fileno | |
38 | #endif | |
39 | ||
40 | /////////////////////////////////////////////////////////////////////////////// | |
41 | // The test case | |
42 | ||
43 | class FileKindTestCase : public CppUnit::TestCase | |
44 | { | |
45 | CPPUNIT_TEST_SUITE(FileKindTestCase); | |
46 | CPPUNIT_TEST(File); | |
47 | #if defined __UNIX__ || defined _MSC_VER || defined __MINGW32__ | |
48 | CPPUNIT_TEST(Pipe); | |
49 | #endif | |
50 | #if defined __UNIX__ | |
51 | CPPUNIT_TEST(Socket); | |
52 | #endif | |
53 | CPPUNIT_TEST(Stdin); | |
54 | CPPUNIT_TEST(MemoryStream); | |
55 | #if wxUSE_SOCKETS | |
56 | CPPUNIT_TEST(SocketStream); | |
57 | #endif | |
58 | CPPUNIT_TEST_SUITE_END(); | |
59 | ||
60 | void File(); | |
61 | void Pipe(); | |
62 | void Socket(); | |
63 | void Stdin(); | |
64 | void MemoryStream(); | |
65 | #if wxUSE_SOCKETS | |
66 | void SocketStream(); | |
67 | #endif | |
68 | ||
69 | void TestFILE(wxFFile& file, bool expected); | |
70 | void TestFd(wxFile& file, bool expected); | |
71 | }; | |
72 | ||
73 | // test a wxFFile and wxFFileInput/OutputStreams of a known type | |
74 | // | |
75 | void FileKindTestCase::TestFILE(wxFFile& file, bool expected) | |
76 | { | |
77 | CPPUNIT_ASSERT(file.IsOpened()); | |
78 | CPPUNIT_ASSERT((wxGetFileKind(file.fp()) == wxFILE_KIND_DISK) == expected); | |
79 | CPPUNIT_ASSERT((file.GetKind() == wxFILE_KIND_DISK) == expected); | |
80 | ||
81 | wxFFileInputStream inStream(file); | |
82 | CPPUNIT_ASSERT(inStream.IsSeekable() == expected); | |
83 | ||
84 | wxFFileOutputStream outStream(file); | |
85 | CPPUNIT_ASSERT(outStream.IsSeekable() == expected); | |
86 | } | |
87 | ||
88 | // test a wxFile and wxFileInput/OutputStreams of a known type | |
89 | // | |
90 | void FileKindTestCase::TestFd(wxFile& file, bool expected) | |
91 | { | |
92 | CPPUNIT_ASSERT(file.IsOpened()); | |
93 | CPPUNIT_ASSERT((wxGetFileKind(file.fd()) == wxFILE_KIND_DISK) == expected); | |
94 | CPPUNIT_ASSERT((file.GetKind() == wxFILE_KIND_DISK) == expected); | |
95 | ||
96 | wxFileInputStream inStream(file); | |
97 | CPPUNIT_ASSERT(inStream.IsSeekable() == expected); | |
98 | ||
99 | wxFileOutputStream outStream(file); | |
100 | CPPUNIT_ASSERT(outStream.IsSeekable() == expected); | |
101 | } | |
102 | ||
103 | struct TempFile | |
104 | { | |
105 | ~TempFile() { if (!m_name.IsEmpty()) wxRemoveFile(m_name); } | |
106 | wxString m_name; | |
107 | }; | |
108 | ||
109 | // test with an ordinary file | |
110 | // | |
111 | void FileKindTestCase::File() | |
112 | { | |
113 | TempFile tmp; // put first | |
114 | wxFile file; | |
115 | tmp.m_name = wxFileName::CreateTempFileName(wxT("wxft"), &file); | |
116 | TestFd(file, true); | |
117 | file.Close(); | |
118 | ||
119 | wxFFile ffile(tmp.m_name); | |
120 | TestFILE(ffile, true); | |
121 | } | |
122 | ||
123 | // test with a pipe | |
124 | // | |
125 | #if defined __UNIX__ || defined _MSC_VER || defined __MINGW32__ | |
126 | void FileKindTestCase::Pipe() | |
127 | { | |
128 | int afd[2]; | |
129 | int rc; | |
130 | #ifdef __UNIX__ | |
131 | rc = pipe(afd); | |
132 | #else | |
133 | rc = _pipe(afd, 256, O_BINARY); | |
134 | #endif | |
135 | CPPUNIT_ASSERT_EQUAL_MESSAGE("Failed to create pipe", 0, rc); | |
136 | ||
137 | wxFile file0(afd[0]); | |
138 | wxFile file1(afd[1]); | |
139 | TestFd(file0, false); | |
140 | file0.Detach(); | |
141 | ||
142 | wxFFile ffile(fdopen(afd[0], "r")); | |
143 | TestFILE(ffile, false); | |
144 | } | |
145 | #endif | |
146 | ||
147 | // test with a socket | |
148 | // | |
149 | #if defined __UNIX__ | |
150 | void FileKindTestCase::Socket() | |
151 | { | |
152 | int s = socket(PF_INET, SOCK_STREAM, 0); | |
153 | ||
154 | wxFile file(s); | |
155 | TestFd(file, false); | |
156 | file.Detach(); | |
157 | ||
158 | wxFFile ffile(fdopen(s, "r")); | |
159 | TestFILE(ffile, false); | |
160 | } | |
161 | #endif | |
162 | ||
163 | // Socket streams should be non-seekable | |
164 | // | |
165 | #if wxUSE_SOCKETS | |
166 | void FileKindTestCase::SocketStream() | |
167 | { | |
168 | wxSocketClient client; | |
169 | wxSocketInputStream inStream(client); | |
170 | CPPUNIT_ASSERT(!inStream.IsSeekable()); | |
171 | wxSocketOutputStream outStream(client); | |
172 | CPPUNIT_ASSERT(!outStream.IsSeekable()); | |
173 | ||
174 | wxBufferedInputStream nonSeekableBufferedInput(inStream); | |
175 | CPPUNIT_ASSERT(!nonSeekableBufferedInput.IsSeekable()); | |
176 | wxBufferedOutputStream nonSeekableBufferedOutput(outStream); | |
177 | CPPUNIT_ASSERT(!nonSeekableBufferedOutput.IsSeekable()); | |
178 | } | |
179 | #endif | |
180 | ||
181 | // Memory streams should be seekable | |
182 | // | |
183 | void FileKindTestCase::MemoryStream() | |
184 | { | |
185 | char buf[20]; | |
186 | wxMemoryInputStream inStream(buf, sizeof(buf)); | |
187 | CPPUNIT_ASSERT(inStream.IsSeekable()); | |
188 | wxMemoryOutputStream outStream(buf, sizeof(buf)); | |
189 | CPPUNIT_ASSERT(outStream.IsSeekable()); | |
190 | ||
191 | wxBufferedInputStream seekableBufferedInput(inStream); | |
192 | CPPUNIT_ASSERT(seekableBufferedInput.IsSeekable()); | |
193 | wxBufferedOutputStream seekableBufferedOutput(outStream); | |
194 | CPPUNIT_ASSERT(seekableBufferedOutput.IsSeekable()); | |
195 | } | |
196 | ||
197 | // Stdin will usually be a terminal, if so then test it | |
198 | // | |
199 | void FileKindTestCase::Stdin() | |
200 | { | |
201 | if (isatty(0)) | |
202 | CPPUNIT_ASSERT(wxGetFileKind(0) == wxFILE_KIND_TERMINAL); | |
203 | if (isatty(fileno(stdin))) | |
204 | CPPUNIT_ASSERT(wxGetFileKind(stdin) == wxFILE_KIND_TERMINAL); | |
205 | } | |
206 | ||
207 | // register in the unnamed registry so that these tests are run by default | |
208 | CPPUNIT_TEST_SUITE_REGISTRATION(FileKindTestCase); | |
209 | ||
210 | // also include in its own registry so that these tests can be run alone | |
211 | CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(FileKindTestCase, "FileKindTestCase"); | |
212 | ||
213 | #endif // wxUSE_STREAMS |