150 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			150 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| ///////////////////////////////////////////////////////////////////////////////
 | |
| // Name:        tests/streams/filestream.cpp
 | |
| // Purpose:     Test wxFileInputStream/wxFileOutputStream
 | |
| // Author:      Hans Van Leemputten
 | |
| // Copyright:   (c) 2004 Hans Van Leemputten
 | |
| // Licence:     wxWindows licence
 | |
| ///////////////////////////////////////////////////////////////////////////////
 | |
| 
 | |
| // For compilers that support precompilation, includes "wx/wx.h".
 | |
| // and "wx/cppunit.h"
 | |
| #include "testprec.h"
 | |
| 
 | |
| #ifdef __BORLANDC__
 | |
|     #pragma hdrstop
 | |
| #endif
 | |
| 
 | |
| // for all others, include the necessary headers
 | |
| #ifndef WX_PRECOMP
 | |
|     #include "wx/wx.h"
 | |
| #endif
 | |
| 
 | |
| #include "wx/wfstream.h"
 | |
| 
 | |
| #include "bstream.h"
 | |
| 
 | |
| #define DATABUFFER_SIZE     1024
 | |
| 
 | |
| static const wxString FILENAME_FILEINSTREAM = wxT("fileinstream.test");
 | |
| static const wxString FILENAME_FILEOUTSTREAM = wxT("fileoutstream.test");
 | |
| 
 | |
| ///////////////////////////////////////////////////////////////////////////////
 | |
| // The test case
 | |
| //
 | |
| // Try to fully test wxFileInputStream and wxFileOutputStream
 | |
| 
 | |
| class fileStream : public BaseStreamTestCase<wxFileInputStream, wxFileOutputStream>
 | |
| {
 | |
| public:
 | |
|     fileStream();
 | |
| 
 | |
|     CPPUNIT_TEST_SUITE(fileStream);
 | |
|         // Base class stream tests the fileStream supports.
 | |
|         CPPUNIT_TEST(Input_GetSize);
 | |
|         CPPUNIT_TEST(Input_GetC);
 | |
|         CPPUNIT_TEST(Input_Read);
 | |
|         CPPUNIT_TEST(Input_Eof);
 | |
|         CPPUNIT_TEST(Input_LastRead);
 | |
|         CPPUNIT_TEST(Input_CanRead);
 | |
|         CPPUNIT_TEST(Input_SeekI);
 | |
|         CPPUNIT_TEST(Input_TellI);
 | |
|         CPPUNIT_TEST(Input_Peek);
 | |
|         CPPUNIT_TEST(Input_Ungetch);
 | |
| 
 | |
|         CPPUNIT_TEST(Output_PutC);
 | |
|         CPPUNIT_TEST(Output_Write);
 | |
|         CPPUNIT_TEST(Output_LastWrite);
 | |
|         CPPUNIT_TEST(Output_SeekO);
 | |
|         CPPUNIT_TEST(Output_TellO);
 | |
| 
 | |
|         // Other test specific for File stream test case.
 | |
|     CPPUNIT_TEST_SUITE_END();
 | |
| 
 | |
| protected:
 | |
|     // Add own test here.
 | |
| 
 | |
| private:
 | |
|     // Implement base class functions.
 | |
|     virtual wxFileInputStream  *DoCreateInStream() wxOVERRIDE;
 | |
|     virtual wxFileOutputStream *DoCreateOutStream() wxOVERRIDE;
 | |
|     virtual void DoDeleteOutStream() wxOVERRIDE;
 | |
| 
 | |
| private:
 | |
|     wxString GetInFileName() const;
 | |
| };
 | |
| 
 | |
| fileStream::fileStream()
 | |
| {
 | |
|     m_bSeekInvalidBeyondEnd = false;
 | |
| }
 | |
| 
 | |
| wxFileInputStream *fileStream::DoCreateInStream()
 | |
| {
 | |
|     wxFileInputStream *pFileInStream = new wxFileInputStream(GetInFileName());
 | |
|     CPPUNIT_ASSERT(pFileInStream->IsOk());
 | |
|     return pFileInStream;
 | |
| }
 | |
| wxFileOutputStream *fileStream::DoCreateOutStream()
 | |
| {
 | |
|     wxFileOutputStream *pFileOutStream = new wxFileOutputStream(FILENAME_FILEOUTSTREAM);
 | |
|     CPPUNIT_ASSERT(pFileOutStream->IsOk());
 | |
|     return pFileOutStream;
 | |
| }
 | |
| 
 | |
| void fileStream::DoDeleteOutStream()
 | |
| {
 | |
|     ::wxRemoveFile(FILENAME_FILEOUTSTREAM);
 | |
| }
 | |
| 
 | |
| wxString fileStream::GetInFileName() const
 | |
| {
 | |
|     class AutoRemoveFile
 | |
|     {
 | |
|     public:
 | |
|         AutoRemoveFile()
 | |
|         {
 | |
|             m_created = false;
 | |
|         }
 | |
| 
 | |
|         ~AutoRemoveFile()
 | |
|         {
 | |
|             if ( m_created )
 | |
|                 wxRemoveFile(FILENAME_FILEINSTREAM);
 | |
|         }
 | |
| 
 | |
|         bool ShouldCreate()
 | |
|         {
 | |
|             if ( m_created )
 | |
|                 return false;
 | |
| 
 | |
|             m_created = true;
 | |
| 
 | |
|             return true;
 | |
|         }
 | |
| 
 | |
|     private:
 | |
|         bool m_created;
 | |
|     };
 | |
| 
 | |
|     static AutoRemoveFile autoFile;
 | |
|     if ( autoFile.ShouldCreate() )
 | |
|     {
 | |
|         // Make sure we have a input file...
 | |
|         char buf[DATABUFFER_SIZE];
 | |
|         wxFileOutputStream out(FILENAME_FILEINSTREAM);
 | |
| 
 | |
|         // Init the data buffer.
 | |
|         for (size_t i = 0; i < DATABUFFER_SIZE; i++)
 | |
|             buf[i] = (i % 0xFF);
 | |
| 
 | |
|         // Save the data
 | |
|         out.Write(buf, DATABUFFER_SIZE);
 | |
|     }
 | |
| 
 | |
|     return FILENAME_FILEINSTREAM;
 | |
| }
 | |
| 
 | |
| // Register the stream sub suite, by using some stream helper macro.
 | |
| // Note: Don't forget to connect it to the base suite (See: bstream.cpp => StreamCase::suite())
 | |
| STREAM_TEST_SUBSUITE_NAMED_REGISTRATION(fileStream)
 |