]>
Commit | Line | Data |
---|---|---|
e1265174 MW |
1 | /////////////////////////////////////////////////////////////////////////////// |
2 | // Name: tests/streams/tempfile.cpp | |
3 | // Purpose: Test wxTempFileOutputStream | |
4 | // Author: Mike Wetherell | |
e1265174 | 5 | // Copyright: (c) 2005 Mike Wetherell |
526954c5 | 6 | // Licence: wxWindows licence |
e1265174 MW |
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 | #include "wx/wfstream.h" | |
21 | #include "wx/filename.h" | |
22 | #include "bstream.h" | |
23 | ||
24 | #if wxUSE_STREAMS && wxUSE_FILE | |
25 | ||
caa96da7 | 26 | #include "testfile.h" |
e1265174 MW |
27 | |
28 | /////////////////////////////////////////////////////////////////////////////// | |
29 | // The test case | |
30 | ||
31 | class tempStream : public CppUnit::TestCase | |
32 | { | |
33 | CPPUNIT_TEST_SUITE(tempStream); | |
34 | CPPUNIT_TEST(DoNothing); | |
35 | CPPUNIT_TEST(Close); | |
36 | CPPUNIT_TEST(Commit); | |
37 | CPPUNIT_TEST(Discard); | |
38 | CPPUNIT_TEST_SUITE_END(); | |
39 | ||
40 | void DoNothing() { DoTest(DONOTHING, false); } | |
41 | void Close() { DoTest(CLOSE, true); } | |
42 | void Commit() { DoTest(COMMIT, true); } | |
43 | void Discard() { DoTest(DISCARD, false); } | |
44 | ||
45 | enum Action { DONOTHING, CLOSE, COMMIT, DISCARD }; | |
46 | void DoTest(Action action, bool shouldHaveCommited); | |
47 | }; | |
48 | ||
49 | // the common test code | |
50 | // | |
51 | void tempStream::DoTest(Action action, bool shouldHaveCommited) | |
52 | { | |
53 | TestFile temp; | |
54 | ||
55 | { | |
56 | wxTempFileOutputStream out(temp.GetName()); | |
57 | out.Write("Affer", 5); | |
58 | CPPUNIT_ASSERT(out.SeekO(2) == 2); | |
59 | out.Write("t", 1); | |
60 | CPPUNIT_ASSERT(out.IsSeekable()); | |
61 | CPPUNIT_ASSERT(out.GetLength() == 5); | |
62 | CPPUNIT_ASSERT(out.TellO() == 3); | |
63 | ||
64 | switch (action) { | |
65 | case DONOTHING: break; | |
66 | case COMMIT: out.Commit(); break; | |
67 | case DISCARD: out.Discard(); break; | |
68 | case CLOSE: out.Close(); | |
69 | } | |
70 | } | |
71 | ||
72 | wxFileInputStream in(temp.GetName()); | |
73 | char buf[32]; | |
74 | in.Read(buf, sizeof(buf)); | |
75 | buf[in.LastRead()] = 0; | |
76 | CPPUNIT_ASSERT(strcmp(buf, shouldHaveCommited ? "After" : "Before") == 0); | |
77 | } | |
78 | ||
79 | ||
80 | // Register the stream sub suite, by using some stream helper macro. | |
81 | // Note: Don't forget to connect it to the base suite (See: bstream.cpp => StreamCase::suite()) | |
82 | STREAM_TEST_SUBSUITE_NAMED_REGISTRATION(tempStream) | |
83 | ||
84 | #endif // wxUSE_STREAMS && wxUSE_FILE |