1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: tests/streams/socketstream.cpp
3 // Purpose: Test wxSocketInputStream/wxSocketOutputStream
4 // Author: Vadim Zeitlin
6 // Copyright: (c) 2008 Vadim Zeitlin
7 // Licence: wxWidgets licence
8 ///////////////////////////////////////////////////////////////////////////////
10 // For compilers that support precompilation, includes "wx/wx.h".
18 // for all others, include the necessary headers
23 #include "wx/socket.h"
24 #include "wx/sckstrm.h"
25 #include "wx/thread.h"
32 const int TEST_PORT_READ
= 0x7778; // arbitrary, chosen because == "wx"
33 const int TEST_PORT_WRITE
= 0x7779; // well, "wy"
35 // these cond and mutex are used to minimize the risk of the main thread
36 // Connect()-ing before this thread starts Accept()-ing connections but
37 // unfortunately we can't make this truly safe, see comment in
38 // SocketServerThread::Entry()
40 wxCondition
gs_cond(gs_mutex
);
41 } // anonymous namespace
43 // return address for the given port on local host
44 static inline wxIPV4address
LocalAddress(int port
)
53 // A thread which creates a listening socket on the specified port and executes
54 // the given function with each socket which connects to it
55 class SocketServerThread
: public wxThread
58 // port is the port to listen on and function will be called on each
60 SocketServerThread(int port
, void (*accept
)(wxSocketBase
&))
61 : wxThread(wxTHREAD_JOINABLE
),
72 wxSocketServer
srv(LocalAddress(m_port
), wxSOCKET_REUSEADDR
);
74 // FIXME: this is still not atomic, of course and the main thread could
75 // call Connect() before we have time to Accept() but there is
76 // no way to fix it with current API
78 wxMutexLocker
lock(gs_mutex
);
82 wxSocketBase
*socket
= srv
.Accept();
90 void (*m_accept
)(wxSocketBase
&);
92 DECLARE_NO_COPY_CLASS(SocketServerThread
)
95 // The test case for socket streams
97 public BaseStreamTestCase
<wxSocketInputStream
, wxSocketOutputStream
>
101 virtual ~socketStream();
103 virtual void setUp();
104 virtual void tearDown();
106 CPPUNIT_TEST_SUITE(socketStream
);
107 // Base class stream tests the socketStream supports.
108 CPPUNIT_TEST(Input_GetC
);
110 // This one fails because wxSocketInputStream::Eof() is not implemented
112 //CPPUNIT_TEST(Input_Read);
114 // The other ones untested yet
116 CPPUNIT_TEST(Input_Eof
);
117 CPPUNIT_TEST(Input_LastRead
);
118 CPPUNIT_TEST(Input_CanRead
);
119 CPPUNIT_TEST(Input_Peek
);
120 CPPUNIT_TEST(Input_Ungetch
);
122 CPPUNIT_TEST(Output_PutC
);
123 CPPUNIT_TEST(Output_Write
);
124 CPPUNIT_TEST(Output_LastWrite
);
126 CPPUNIT_TEST_SUITE_END();
129 // Implement base class functions.
130 virtual wxSocketInputStream
*DoCreateInStream();
131 virtual wxSocketOutputStream
*DoCreateOutStream();
133 // socket thread functions
134 static void WriteSocket(wxSocketBase
& socket
)
136 socket
.Write("hello, world!", 13);
139 static void ReadSocket(wxSocketBase
& socket
)
142 while ( socket
.Read(&ch
, 1).LastCount() == 1 )
146 wxSocketClient
*m_readSocket
,
148 wxThread
*m_writeThread
,
152 socketStream
::socketStream()
155 m_writeSocket
= NULL
;
163 socketStream
::~socketStream()
168 void socketStream
::setUp()
170 // create the socket threads and wait until they are ready to accept
171 // connections (if we called Connect() before this happens, it would fail)
173 wxMutexLocker
lock(gs_mutex
);
176 new SocketServerThread(TEST_PORT_READ
, &socketStream
::WriteSocket
);
177 CPPUNIT_ASSERT_EQUAL( wxCOND_NO_ERROR
, gs_cond
.Wait() );
180 new SocketServerThread(TEST_PORT_WRITE
, &socketStream
::ReadSocket
);
181 CPPUNIT_ASSERT_EQUAL( wxCOND_NO_ERROR
, gs_cond
.Wait() );
184 m_readSocket
= new wxSocketClient
;
185 m_readSocket
->SetTimeout(3);
186 CPPUNIT_ASSERT( m_readSocket
->Connect(LocalAddress(TEST_PORT_READ
)) );
188 m_writeSocket
= new wxSocketClient
;
189 m_writeSocket
->SetTimeout(3);
190 CPPUNIT_ASSERT( m_writeSocket
->Connect(LocalAddress(TEST_PORT_WRITE
)) );
193 void socketStream
::tearDown()
195 wxDELETE(m_readSocket
);
196 wxDELETE(m_writeSocket
);
198 m_writeThread
->Wait();
199 wxDELETE(m_writeThread
);
201 m_readThread
->Wait();
202 wxDELETE(m_readThread
);
205 wxSocketInputStream
*socketStream
::DoCreateInStream()
207 wxSocketInputStream
*pStrInStream
= new wxSocketInputStream(*m_readSocket
);
208 CPPUNIT_ASSERT(pStrInStream
->IsOk());
212 wxSocketOutputStream
*socketStream
::DoCreateOutStream()
214 wxSocketOutputStream
*pStrOutStream
= new wxSocketOutputStream(*m_writeSocket
);
215 CPPUNIT_ASSERT(pStrOutStream
->IsOk());
216 return pStrOutStream
;
219 // Register the stream sub suite, by using some stream helper macro.
220 STREAM_TEST_SUBSUITE_NAMED_REGISTRATION(socketStream
)