| 1 | /////////////////////////////////////////////////////////////////////////////// |
| 2 | // Name: tests/textfile/textfile.cpp |
| 3 | // Purpose: wxTextFile unit test |
| 4 | // Author: Vadim Zeitlin |
| 5 | // Created: 2006-03-31 |
| 6 | // RCS-ID: $Id$ |
| 7 | // Copyright: (c) 2006 Vadim Zeitlin |
| 8 | /////////////////////////////////////////////////////////////////////////////// |
| 9 | |
| 10 | // ---------------------------------------------------------------------------- |
| 11 | // headers |
| 12 | // ---------------------------------------------------------------------------- |
| 13 | |
| 14 | #include "testprec.h" |
| 15 | |
| 16 | #ifdef __BORLANDC__ |
| 17 | #pragma hdrstop |
| 18 | #endif |
| 19 | |
| 20 | #if wxUSE_TEXTFILE |
| 21 | |
| 22 | #ifndef WX_PRECOMP |
| 23 | #endif // WX_PRECOMP |
| 24 | |
| 25 | #include "wx/ffile.h" |
| 26 | #include "wx/textfile.h" |
| 27 | |
| 28 | #ifdef __VISUALC__ |
| 29 | #define unlink _unlink |
| 30 | #endif |
| 31 | |
| 32 | // ---------------------------------------------------------------------------- |
| 33 | // test class |
| 34 | // ---------------------------------------------------------------------------- |
| 35 | |
| 36 | class TextFileTestCase : public CppUnit::TestCase |
| 37 | { |
| 38 | public: |
| 39 | TextFileTestCase() { } |
| 40 | |
| 41 | virtual void tearDown() { unlink(GetTestFileName()); } |
| 42 | |
| 43 | private: |
| 44 | CPPUNIT_TEST_SUITE( TextFileTestCase ); |
| 45 | CPPUNIT_TEST( ReadEmpty ); |
| 46 | CPPUNIT_TEST( ReadDOS ); |
| 47 | CPPUNIT_TEST( ReadUnix ); |
| 48 | CPPUNIT_TEST( ReadMac ); |
| 49 | CPPUNIT_TEST( ReadMixed ); |
| 50 | #if wxUSE_UNICODE |
| 51 | CPPUNIT_TEST( ReadUTF8 ); |
| 52 | CPPUNIT_TEST( ReadUTF16 ); |
| 53 | #endif // wxUSE_UNICODE |
| 54 | CPPUNIT_TEST( ReadBig ); |
| 55 | CPPUNIT_TEST_SUITE_END(); |
| 56 | |
| 57 | void ReadEmpty(); |
| 58 | void ReadDOS(); |
| 59 | void ReadUnix(); |
| 60 | void ReadMac(); |
| 61 | void ReadMixed(); |
| 62 | #if wxUSE_UNICODE |
| 63 | void ReadUTF8(); |
| 64 | void ReadUTF16(); |
| 65 | #endif // wxUSE_UNICODE |
| 66 | void ReadBig(); |
| 67 | |
| 68 | // return the name of the test file we use |
| 69 | static const char *GetTestFileName() { return "textfiletest.txt"; } |
| 70 | |
| 71 | // create the test file with the given contents |
| 72 | static void CreateTestFile(const char *contents) |
| 73 | { |
| 74 | CreateTestFile(strlen(contents), contents); |
| 75 | } |
| 76 | |
| 77 | // create the test file with the given contents (version must be used if |
| 78 | // contents contains NULs) |
| 79 | static void CreateTestFile(size_t len, const char *contents); |
| 80 | |
| 81 | |
| 82 | DECLARE_NO_COPY_CLASS(TextFileTestCase) |
| 83 | }; |
| 84 | |
| 85 | // register in the unnamed registry so that these tests are run by default |
| 86 | CPPUNIT_TEST_SUITE_REGISTRATION( TextFileTestCase ); |
| 87 | |
| 88 | // also include in its own registry so that these tests can be run alone |
| 89 | CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( TextFileTestCase, "TextFileTestCase" ); |
| 90 | |
| 91 | void TextFileTestCase::CreateTestFile(size_t len, const char *contents) |
| 92 | { |
| 93 | FILE *f = fopen(GetTestFileName(), "wb"); |
| 94 | CPPUNIT_ASSERT( f ); |
| 95 | |
| 96 | CPPUNIT_ASSERT_EQUAL( len, fwrite(contents, 1, len, f) ); |
| 97 | CPPUNIT_ASSERT_EQUAL( 0, fclose(f) ); |
| 98 | } |
| 99 | |
| 100 | void TextFileTestCase::ReadEmpty() |
| 101 | { |
| 102 | CreateTestFile(""); |
| 103 | |
| 104 | wxTextFile f; |
| 105 | CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) ); |
| 106 | |
| 107 | CPPUNIT_ASSERT_EQUAL( (size_t)0, f.GetLineCount() ); |
| 108 | } |
| 109 | |
| 110 | void TextFileTestCase::ReadDOS() |
| 111 | { |
| 112 | CreateTestFile("foo\r\nbar\r\nbaz"); |
| 113 | |
| 114 | wxTextFile f; |
| 115 | CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) ); |
| 116 | |
| 117 | CPPUNIT_ASSERT_EQUAL( (size_t)3, f.GetLineCount() ); |
| 118 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_Dos, f.GetLineType(0) ); |
| 119 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(2) ); |
| 120 | CPPUNIT_ASSERT_EQUAL( wxString(wxT("bar")), f.GetLine(1) ); |
| 121 | CPPUNIT_ASSERT_EQUAL( wxString(wxT("baz")), f.GetLastLine() ); |
| 122 | } |
| 123 | |
| 124 | void TextFileTestCase::ReadUnix() |
| 125 | { |
| 126 | CreateTestFile("foo\nbar\nbaz"); |
| 127 | |
| 128 | wxTextFile f; |
| 129 | CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) ); |
| 130 | |
| 131 | CPPUNIT_ASSERT_EQUAL( (size_t)3, f.GetLineCount() ); |
| 132 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_Unix, f.GetLineType(0) ); |
| 133 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(2) ); |
| 134 | CPPUNIT_ASSERT_EQUAL( wxString(wxT("bar")), f.GetLine(1) ); |
| 135 | CPPUNIT_ASSERT_EQUAL( wxString(wxT("baz")), f.GetLastLine() ); |
| 136 | } |
| 137 | |
| 138 | void TextFileTestCase::ReadMac() |
| 139 | { |
| 140 | CreateTestFile("foo\rbar\rbaz"); |
| 141 | |
| 142 | wxTextFile f; |
| 143 | CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) ); |
| 144 | |
| 145 | CPPUNIT_ASSERT_EQUAL( (size_t)3, f.GetLineCount() ); |
| 146 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_Mac, f.GetLineType(0) ); |
| 147 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(2) ); |
| 148 | CPPUNIT_ASSERT_EQUAL( wxString(wxT("bar")), f.GetLine(1) ); |
| 149 | CPPUNIT_ASSERT_EQUAL( wxString(wxT("baz")), f.GetLastLine() ); |
| 150 | } |
| 151 | |
| 152 | void TextFileTestCase::ReadMixed() |
| 153 | { |
| 154 | CreateTestFile("foo\rbar\r\nbaz\n"); |
| 155 | |
| 156 | wxTextFile f; |
| 157 | CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) ); |
| 158 | |
| 159 | CPPUNIT_ASSERT_EQUAL( (size_t)3, f.GetLineCount() ); |
| 160 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_Mac, f.GetLineType(0) ); |
| 161 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_Dos, f.GetLineType(1) ); |
| 162 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_Unix, f.GetLineType(2) ); |
| 163 | CPPUNIT_ASSERT_EQUAL( wxString(wxT("foo")), f.GetFirstLine() ); |
| 164 | CPPUNIT_ASSERT_EQUAL( wxString(wxT("bar")), f.GetLine(1) ); |
| 165 | CPPUNIT_ASSERT_EQUAL( wxString(wxT("baz")), f.GetLastLine() ); |
| 166 | } |
| 167 | |
| 168 | #if wxUSE_UNICODE |
| 169 | |
| 170 | void TextFileTestCase::ReadUTF8() |
| 171 | { |
| 172 | CreateTestFile("\xd0\x9f\n" |
| 173 | "\xd1\x80\xd0\xb8\xd0\xb2\xd0\xb5\xd1\x82"); |
| 174 | |
| 175 | wxTextFile f; |
| 176 | CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName()), wxConvUTF8) ); |
| 177 | |
| 178 | CPPUNIT_ASSERT_EQUAL( (size_t)2, f.GetLineCount() ); |
| 179 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_Unix, f.GetLineType(0) ); |
| 180 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(1) ); |
| 181 | #ifdef wxHAVE_U_ESCAPE |
| 182 | CPPUNIT_ASSERT_EQUAL( wxString(L"\u041f"), f.GetFirstLine() ); |
| 183 | CPPUNIT_ASSERT_EQUAL( wxString(L"\u0440\u0438\u0432\u0435\u0442"), |
| 184 | f.GetLastLine() ); |
| 185 | #endif // wxHAVE_U_ESCAPE |
| 186 | } |
| 187 | |
| 188 | void TextFileTestCase::ReadUTF16() |
| 189 | { |
| 190 | CreateTestFile(16, |
| 191 | "\x1f\x04\x0d\x00\x0a\x00" |
| 192 | "\x40\x04\x38\x04\x32\x04\x35\x04\x42\x04"); |
| 193 | |
| 194 | wxTextFile f; |
| 195 | wxMBConvUTF16LE conv; |
| 196 | CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName()), conv) ); |
| 197 | |
| 198 | CPPUNIT_ASSERT_EQUAL( (size_t)2, f.GetLineCount() ); |
| 199 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_Dos, f.GetLineType(0) ); |
| 200 | CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(1) ); |
| 201 | |
| 202 | #ifdef wxHAVE_U_ESCAPE |
| 203 | CPPUNIT_ASSERT_EQUAL( wxString(L"\u041f"), f.GetFirstLine() ); |
| 204 | CPPUNIT_ASSERT_EQUAL( wxString(L"\u0440\u0438\u0432\u0435\u0442"), |
| 205 | f.GetLastLine() ); |
| 206 | #endif // wxHAVE_U_ESCAPE |
| 207 | } |
| 208 | |
| 209 | #endif // wxUSE_UNICODE |
| 210 | |
| 211 | void TextFileTestCase::ReadBig() |
| 212 | { |
| 213 | static const size_t NUM_LINES = 10000; |
| 214 | |
| 215 | { |
| 216 | wxFFile f(GetTestFileName(), "w"); |
| 217 | for ( size_t n = 0; n < NUM_LINES; n++ ) |
| 218 | { |
| 219 | fprintf(f.fp(), "Line %lu\n", (unsigned long)n + 1); |
| 220 | } |
| 221 | } |
| 222 | |
| 223 | wxTextFile f; |
| 224 | CPPUNIT_ASSERT( f.Open(GetTestFileName()) ); |
| 225 | |
| 226 | CPPUNIT_ASSERT_EQUAL( NUM_LINES, f.GetLineCount() ); |
| 227 | CPPUNIT_ASSERT_EQUAL( wxString("Line 1"), f[0] ); |
| 228 | CPPUNIT_ASSERT_EQUAL( wxString("Line 999"), f[998] ); |
| 229 | CPPUNIT_ASSERT_EQUAL( wxString("Line 1000"), f[999] ); |
| 230 | CPPUNIT_ASSERT_EQUAL( wxString::Format("Line %lu", (unsigned long)NUM_LINES), |
| 231 | f[NUM_LINES - 1] ); |
| 232 | } |
| 233 | |
| 234 | #endif // wxUSE_TEXTFILE |
| 235 | |