added a unit test for input/output file streams and fixed the problem it exposed...

[wxWidgets.git] / tests / streams / iostreams.cpp

diff --git a/tests/streams/iostreams.cpp b/tests/streams/iostreams.cpp
new file mode 100644 (file)
index 0000000..92a640e
--- /dev/null
+++ b/tests/streams/iostreams.cpp
@@ -0,0 +1,78 @@
+///////////////////////////////////////////////////////////////////////////////
+// Name:        tests/streams/iostreams.cpp
+// Purpose:     unit test for input/output streams
+// Author:      Vadim Zeitlin
+// Created:     2008-06-15
+// RCS-ID:      $Id$
+///////////////////////////////////////////////////////////////////////////////
+
+// ----------------------------------------------------------------------------
+// headers
+// ----------------------------------------------------------------------------
+
+#include "testprec.h"
+
+#ifdef __BORLANDC__
+    #pragma hdrstop
+#endif
+
+#if wxUSE_STREAMS
+
+#include "wx/filename.h"
+#include "wx/wfstream.h"
+
+// --------------------------------------------------------------------------
+// test class
+// --------------------------------------------------------------------------
+
+class IOStreamsTestCase : public CppUnit::TestCase
+{
+public:
+    IOStreamsTestCase() { }
+
+    virtual void tearDown()
+    {
+        if ( !m_fnTemp.empty() )
+        {
+            wxRemoveFile(m_fnTemp);
+            m_fnTemp.clear();
+        }
+    }
+
+private:
+    CPPUNIT_TEST_SUITE( IOStreamsTestCase );
+        CPPUNIT_TEST( FStream );
+        CPPUNIT_TEST( FFStream );
+    CPPUNIT_TEST_SUITE_END();
+
+    void FStream() { wxFileStream s(GetTempFName()); DoTest(s); }
+    void FFStream() { wxFFileStream s(GetTempFName()); DoTest(s); }
+
+    wxString GetTempFName()
+    {
+        m_fnTemp = wxFileName::CreateTempFileName("wxtest");
+        return m_fnTemp;
+    }
+
+    template <class Stream>
+    void DoTest(Stream& s)
+    {
+        s.PutC('x');
+        WX_ASSERT_SIZET_EQUAL( 1, s.LastWrite() );
+
+        s.SeekI(0);
+        CPPUNIT_ASSERT_EQUAL( int('x'), s.GetC() );
+    }
+
+    wxString m_fnTemp;
+
+    DECLARE_NO_COPY_CLASS(IOStreamsTestCase)
+};
+
+// register in the unnamed registry so that these tests are run by default
+CPPUNIT_TEST_SUITE_REGISTRATION( IOStreamsTestCase );
+
+// also include in it's own registry so that these tests can be run alone
+CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( IOStreamsTestCase, "IOStreamsTestCase" );
+
+#endif // wxUSE_STREAMS