wxWidgets/tests/streams/filestream.cpp

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)