Use wxString::To8BitData() instead of mb_str() to handle NULs correctly.
[wxWidgets.git] / tests / streams / sstream.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: tests/streams/sstream.cpp
3 // Purpose: Test wxStringInputStream/wxStringOutputStream
4 // Author: Vadim Zeitlin
5 // RCS-ID: $Id$
6 // Copyright: (c) 2004 Vadim Zeitlin
7 // Licence: wxWindows licence
8 ///////////////////////////////////////////////////////////////////////////////
9
10 // For compilers that support precompilation, includes "wx/wx.h".
11 // and "wx/cppunit.h"
12 #include "testprec.h"
13
14 #ifdef __BORLANDC__
15 #pragma hdrstop
16 #endif
17
18 // for all others, include the necessary headers
19 #ifndef WX_PRECOMP
20 #endif
21
22 #include "wx/sstream.h"
23
24 #include "bstream.h"
25
26 ///////////////////////////////////////////////////////////////////////////////
27 // The test case
28 //
29 // Try to fully test wxStringInputStream and wxStringOutputStream
30
31 class strStream :
32 public BaseStreamTestCase<wxStringInputStream, wxStringOutputStream>
33 {
34 public:
35 strStream();
36 virtual ~strStream();
37
38 CPPUNIT_TEST_SUITE(strStream);
39 // Base class stream tests the strStream supports.
40 CPPUNIT_TEST(Input_GetSize);
41 CPPUNIT_TEST(Input_GetC);
42 CPPUNIT_TEST(Input_Read);
43 CPPUNIT_TEST(Input_Eof);
44 CPPUNIT_TEST(Input_LastRead);
45 CPPUNIT_TEST(Input_CanRead);
46 CPPUNIT_TEST(Input_SeekI);
47 CPPUNIT_TEST(Input_TellI);
48 CPPUNIT_TEST(Input_Peek);
49 CPPUNIT_TEST(Input_Ungetch);
50
51 CPPUNIT_TEST(Output_PutC);
52 CPPUNIT_TEST(Output_Write);
53 CPPUNIT_TEST(Output_LastWrite);
54 // seeking currently not supported by output string stream
55 //CPPUNIT_TEST(Output_SeekO);
56 //CPPUNIT_TEST(Output_TellO);
57
58 // Other test specific for String stream test case.
59 CPPUNIT_TEST(Output_Check);
60 CPPUNIT_TEST_SUITE_END();
61
62 protected:
63 void Output_Check();
64
65 private:
66 // Implement base class functions.
67 virtual wxStringInputStream *DoCreateInStream();
68 virtual wxStringOutputStream *DoCreateOutStream();
69
70 // output the given string to wxStringOutputStream and check that its
71 // contents is exactly the same string
72 void CheckString(const wxString& text);
73
74 wxString m_str;
75 };
76
77 strStream::strStream()
78 {
79 static const size_t LEN = 256;
80 m_str.reserve(LEN);
81 for ( size_t n = 0; n < LEN; n++ )
82 {
83 m_str += wxChar(wxT('A') + n % (wxT('Z') - wxT('A') + 1));
84 }
85 }
86
87 strStream::~strStream()
88 {
89 }
90
91 wxStringInputStream *strStream::DoCreateInStream()
92 {
93 wxStringInputStream *pStrInStream = new wxStringInputStream(m_str);
94 CPPUNIT_ASSERT(pStrInStream->IsOk());
95 return pStrInStream;
96 }
97
98 wxStringOutputStream *strStream::DoCreateOutStream()
99 {
100 wxStringOutputStream *pStrOutStream = new wxStringOutputStream();
101 CPPUNIT_ASSERT(pStrOutStream->IsOk());
102 return pStrOutStream;
103 }
104
105 void strStream::CheckString(const wxString& text)
106 {
107 wxStringOutputStream sos;
108
109 const wxCharBuffer buf(text.To8BitData());
110 sos.Write(buf, buf.length());
111
112 CPPUNIT_ASSERT_EQUAL( text, sos.GetString() );
113 }
114
115 void strStream::Output_Check()
116 {
117 CheckString("Hello world!");
118 CheckString(wxString("hi\0dden", 8));
119 }
120
121 // Register the stream sub suite, by using some stream helper macro.
122 STREAM_TEST_SUBSUITE_NAMED_REGISTRATION(strStream)