X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/18230eb6b61b4db7907efb00814847dc1e4b9d4d..5e1eac149fc18f51d5a25ac00d957ccaad87b3fa:/tests/textfile/textfiletest.cpp

diff --git a/tests/textfile/textfiletest.cpp b/tests/textfile/textfiletest.cpp
index 244a7ce9fb..01b6da60af 100644
--- a/tests/textfile/textfiletest.cpp
+++ b/tests/textfile/textfiletest.cpp
@@ -42,6 +42,10 @@ private:
         CPPUNIT_TEST( ReadUnix );
         CPPUNIT_TEST( ReadMac );
         CPPUNIT_TEST( ReadMixed );
+#if wxUSE_UNICODE
+        CPPUNIT_TEST( ReadUTF8 );
+        CPPUNIT_TEST( ReadUTF16 );
+#endif // wxUSE_UNICODE
     CPPUNIT_TEST_SUITE_END();
 
     void ReadEmpty();
@@ -49,12 +53,23 @@ private:
     void ReadUnix();
     void ReadMac();
     void ReadMixed();
+#if wxUSE_UNICODE
+    void ReadUTF8();
+    void ReadUTF16();
+#endif // wxUSE_UNICODE
 
     // return the name of the test file we use
     static const char *GetTestFileName() { return "textfiletest.txt"; }
 
     // create the test file with the given contents
-    static void CreateTestFile(const char *contents);
+    static void CreateTestFile(const char *contents)
+    {
+        return CreateTestFile(strlen(contents), contents);
+    }
+
+    // create the test file with the given contents (version must be used if
+    // contents contains NULs)
+    static void CreateTestFile(size_t len, const char *contents);
 
 
     DECLARE_NO_COPY_CLASS(TextFileTestCase)
@@ -66,12 +81,12 @@ CPPUNIT_TEST_SUITE_REGISTRATION( TextFileTestCase );
 // also include in it's own registry so that these tests can be run alone
 CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( TextFileTestCase, "TextFileTestCase" );
 
-void TextFileTestCase::CreateTestFile(const char *contents)
+void TextFileTestCase::CreateTestFile(size_t len, const char *contents)
 {
     FILE *f = fopen(GetTestFileName(), "wb");
     CPPUNIT_ASSERT( f );
 
-    CPPUNIT_ASSERT( fputs(contents, f) >= 0 );
+    CPPUNIT_ASSERT( fwrite(contents, 1, len, f) >= 0 );
     CPPUNIT_ASSERT( fclose(f) == 0 );
 }
 
@@ -82,7 +97,7 @@ void TextFileTestCase::ReadEmpty()
     wxTextFile f;
     CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) );
 
-    CPPUNIT_ASSERT_EQUAL( 0u, f.GetLineCount() );
+    CPPUNIT_ASSERT_EQUAL( (size_t)0, f.GetLineCount() );
 }
 
 void TextFileTestCase::ReadDOS()
@@ -92,7 +107,7 @@ void TextFileTestCase::ReadDOS()
     wxTextFile f;
     CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) );
 
-    CPPUNIT_ASSERT_EQUAL( 3u, f.GetLineCount() );
+    CPPUNIT_ASSERT_EQUAL( (size_t)3, f.GetLineCount() );
     CPPUNIT_ASSERT_EQUAL( wxTextFileType_Dos, f.GetLineType(0) );
     CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(2) );
     CPPUNIT_ASSERT_EQUAL( wxString(_T("bar")), f.GetLine(1) );
@@ -106,7 +121,7 @@ void TextFileTestCase::ReadUnix()
     wxTextFile f;
     CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) );
 
-    CPPUNIT_ASSERT_EQUAL( 3u, f.GetLineCount() );
+    CPPUNIT_ASSERT_EQUAL( (size_t)3, f.GetLineCount() );
     CPPUNIT_ASSERT_EQUAL( wxTextFileType_Unix, f.GetLineType(0) );
     CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(2) );
     CPPUNIT_ASSERT_EQUAL( wxString(_T("bar")), f.GetLine(1) );
@@ -120,7 +135,7 @@ void TextFileTestCase::ReadMac()
     wxTextFile f;
     CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) );
 
-    CPPUNIT_ASSERT_EQUAL( 3u, f.GetLineCount() );
+    CPPUNIT_ASSERT_EQUAL( (size_t)3, f.GetLineCount() );
     CPPUNIT_ASSERT_EQUAL( wxTextFileType_Mac, f.GetLineType(0) );
     CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(2) );
     CPPUNIT_ASSERT_EQUAL( wxString(_T("bar")), f.GetLine(1) );
@@ -134,7 +149,7 @@ void TextFileTestCase::ReadMixed()
     wxTextFile f;
     CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName())) );
 
-    CPPUNIT_ASSERT_EQUAL( 3u, f.GetLineCount() );
+    CPPUNIT_ASSERT_EQUAL( (size_t)3, f.GetLineCount() );
     CPPUNIT_ASSERT_EQUAL( wxTextFileType_Mac, f.GetLineType(0) );
     CPPUNIT_ASSERT_EQUAL( wxTextFileType_Dos, f.GetLineType(1) );
     CPPUNIT_ASSERT_EQUAL( wxTextFileType_Unix, f.GetLineType(2) );
@@ -143,5 +158,48 @@ void TextFileTestCase::ReadMixed()
     CPPUNIT_ASSERT_EQUAL( wxString(_T("baz")), f.GetLastLine() );
 }
 
+#if wxUSE_UNICODE
+
+void TextFileTestCase::ReadUTF8()
+{
+    CreateTestFile("\xd0\x9f\n"
+                   "\xd1\x80\xd0\xb8\xd0\xb2\xd0\xb5\xd1\x82");
+
+    wxTextFile f;
+    CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName()), wxConvUTF8) );
+
+    CPPUNIT_ASSERT_EQUAL( (size_t)2, f.GetLineCount() );
+    CPPUNIT_ASSERT_EQUAL( wxTextFileType_Unix, f.GetLineType(0) );
+    CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(1) );
+#ifdef wxHAVE_U_ESCAPE
+    CPPUNIT_ASSERT_EQUAL( wxString(L"\u041f"), f.GetFirstLine() );
+    CPPUNIT_ASSERT_EQUAL( wxString(L"\u0440\u0438\u0432\u0435\u0442"),
+                          f.GetLastLine() );
+#endif // wxHAVE_U_ESCAPE
+}
+
+void TextFileTestCase::ReadUTF16()
+{
+    CreateTestFile(16,
+                   "\x1f\x04\x0d\x00\x0a\x00"
+                   "\x40\x04\x38\x04\x32\x04\x35\x04\x42\x04");
+
+    wxTextFile f;
+    wxMBConvUTF16LE conv;
+    CPPUNIT_ASSERT( f.Open(wxString::FromAscii(GetTestFileName()), conv) );
+
+    CPPUNIT_ASSERT_EQUAL( (size_t)2, f.GetLineCount() );
+    CPPUNIT_ASSERT_EQUAL( wxTextFileType_Dos, f.GetLineType(0) );
+    CPPUNIT_ASSERT_EQUAL( wxTextFileType_None, f.GetLineType(1) );
+
+#ifdef wxHAVE_U_ESCAPE
+    CPPUNIT_ASSERT_EQUAL( wxString(L"\u041f"), f.GetFirstLine() );
+    CPPUNIT_ASSERT_EQUAL( wxString(L"\u0440\u0438\u0432\u0435\u0442"),
+                          f.GetLastLine() );
+#endif // wxHAVE_U_ESCAPE
+}
+
+#endif // wxUSE_UNICODE
+
 #endif // wxUSE_TEXTFILE