147 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			147 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"
 | 
						|
 | 
						|
 | 
						|
// 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)
 |