1///////////////////////////////////////////////////////////////////////////////
2// Name:        tests/streams/tempfile.cpp
3// Purpose:     Test wxTempFileOutputStream
4// Author:      Mike Wetherell
5// RCS-ID:      $Id: tempfile.cpp 32796 2005-03-13 16:20:51Z MW $
6// Copyright:   (c) 2005 Mike Wetherell
7// Licence:     wxWidgets licence
8///////////////////////////////////////////////////////////////////////////////
9
10#include "testprec.h"
11
12#ifdef __BORLANDC__
13    #pragma hdrstop
14#endif
15
16// for all others, include the necessary headers
17#ifndef WX_PRECOMP
18    #include "wx/wx.h"
19#endif
20
21#include "wx/wfstream.h"
22#include "wx/filename.h"
23#include "bstream.h"
24
25#if wxUSE_STREAMS && wxUSE_FILE
26
27
28///////////////////////////////////////////////////////////////////////////////
29// Self deleting test file
30
31class TestFile
32{
33public:
34    TestFile();
35    ~TestFile() { if (wxFileExists(m_name)) wxRemoveFile(m_name); }
36    wxString GetName() const { return m_name; }
37private:
38    wxString m_name;
39};
40
41// Initialise with a test pattern so we can see if the file is replaced
42//
43TestFile::TestFile()
44{
45    wxFile file;
46    m_name = wxFileName::CreateTempFileName(_T("wxtest"), &file);
47    file.Write("Before", 6);
48}
49
50
51///////////////////////////////////////////////////////////////////////////////
52// The test case
53
54class tempStream : public CppUnit::TestCase
55{
56    CPPUNIT_TEST_SUITE(tempStream);
57        CPPUNIT_TEST(DoNothing);
58        CPPUNIT_TEST(Close);
59        CPPUNIT_TEST(Commit);
60        CPPUNIT_TEST(Discard);
61    CPPUNIT_TEST_SUITE_END();
62
63    void DoNothing() { DoTest(DONOTHING, false); }
64    void Close()     { DoTest(CLOSE, true); }
65    void Commit()    { DoTest(COMMIT, true); }
66    void Discard()   { DoTest(DISCARD, false); }
67
68    enum Action { DONOTHING, CLOSE, COMMIT, DISCARD };
69    void DoTest(Action action, bool shouldHaveCommited);
70};
71
72// the common test code
73//
74void tempStream::DoTest(Action action, bool shouldHaveCommited)
75{
76    TestFile temp;
77
78    {
79        wxTempFileOutputStream out(temp.GetName());
80        out.Write("Affer", 5);
81        CPPUNIT_ASSERT(out.SeekO(2) == 2);
82        out.Write("t", 1);
83        CPPUNIT_ASSERT(out.IsSeekable());
84        CPPUNIT_ASSERT(out.GetLength() == 5);
85        CPPUNIT_ASSERT(out.TellO() == 3);
86
87        switch (action) {
88            case DONOTHING: break;
89            case COMMIT:    out.Commit(); break;
90            case DISCARD:   out.Discard(); break;
91            case CLOSE:     out.Close();
92        }
93    }
94
95    wxFileInputStream in(temp.GetName());
96    char buf[32];
97    in.Read(buf, sizeof(buf));
98    buf[in.LastRead()] = 0;
99    CPPUNIT_ASSERT(strcmp(buf, shouldHaveCommited ? "After" : "Before") == 0);
100}
101
102
103// Register the stream sub suite, by using some stream helper macro.
104// Note: Don't forget to connect it to the base suite (See: bstream.cpp => StreamCase::suite())
105STREAM_TEST_SUBSUITE_NAMED_REGISTRATION(tempStream)
106
107#endif // wxUSE_STREAMS && wxUSE_FILE
108