1 files changed, 105 insertions, 0 deletions
diff --git a/test/nsockstream_test.cpp b/test/nsockstream_test.cpp
new file mode 100644
index 0000000..ce8d77b
--- /dev/null
+++ b/test/nsockstream_test.cpp
@@ -0,0 +1,105 @@
+/*
+*  psocksxx - A C++ wrapper for POSIX sockets
+*  Copyright (C) 2013 Uditha Atukorala
+*
+*  This software library is free software; you can redistribute it and/or modify
+*  it under the terms of the GNU Lesser General Public License as published by
+*  the Free Software Foundation; either version 3 of the License, or
+*  (at your option) any later version.
+*
+*  This software library is distributed in the hope that it will be useful,
+*  but WITHOUT ANY WARRANTY; without even the implied warranty of
+*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+*  GNU Lesser General Public License for more details.
+*
+*  You should have received a copy of the GNU Lesser General Public License
+*  along with this software library. If not, see <http://www.gnu.org/licenses/>.
+*
+*/
+
+#include "nsockstream_test.h"
+
+#include <psocksxx/nsockstream.h>
+
+
+// register the fixture into the 'registry'
+CPPUNIT_TEST_SUITE_REGISTRATION( nsockstream_test );
+
+// use namespace psocksxx
+using namespace psocksxx;
+
+
+void nsockstream_test::test_constructor_tcp() {
+
+	// TCP/IP
+	CPPUNIT_ASSERT_NO_THROW( nsockstream n( sockstreambuf::sock_stream, sockstreambuf::ipproto_tcp ) );
+
+}
+
+
+void nsockstream_test::test_constructor_udp() {
+
+	// UDP
+	CPPUNIT_ASSERT_NO_THROW( nsockstream n( sockstreambuf::sock_dgram, sockstreambuf::ipproto_udp ) );
+
+}
+
+
+void nsockstream_test::test_constructor_streambuf() {
+
+	// socket stream buffer
+	sockstreambuf * ssb = new sockstreambuf();
+
+	// network socket stream
+	nsockstream ss( ssb );
+
+	// assert
+	CPPUNIT_ASSERT( ssb == ss.rdbuf() );
+
+}
+
+
+void nsockstream_test::test_set_timeout() {
+
+	// socket stream buffer
+	sockstreambuf * ssb = new sockstreambuf();
+
+	// network socket stream
+	nsockstream ss( ssb );
+
+	time_t sec = 1;
+	suseconds_t usec = 500;
+
+	// set timeout
+	const timeval * t = ss.timeout( sec, usec );
+
+	// validate
+	CPPUNIT_ASSERT( sec == t->tv_sec );
+	CPPUNIT_ASSERT( usec == t->tv_usec );
+
+}
+
+
+void nsockstream_test::test_clear_timeout() {
+
+	// socket stream buffer
+	sockstreambuf * ssb = new sockstreambuf();
+
+	// network socket stream
+	nsockstream ss( ssb );
+
+	// clear the timeout before a timeout is set
+	CPPUNIT_ASSERT( 0 == ss.clear_timeout() );
+
+
+	time_t sec = 1;
+	suseconds_t usec = 500;
+
+	// set timeout
+	const timeval * t = ss.timeout( sec, usec );
+
+	// clear the timeout after a timeout is set
+	CPPUNIT_ASSERT( 0 == ss.clear_timeout() );
+
+}
+