diff --git a/trunk/src/utest/srs_utest.cpp b/trunk/src/utest/srs_utest.cpp index a58e5814c..1a223dedd 100644 --- a/trunk/src/utest/srs_utest.cpp +++ b/trunk/src/utest/srs_utest.cpp @@ -36,67 +36,6 @@ ISrsThreadContext* _srs_context = new ISrsThreadContext(); SrsConfig* _srs_config = NULL; SrsServer* _srs_server = NULL; -MockEmptyIO::MockEmptyIO() -{ -} - -MockEmptyIO::~MockEmptyIO() -{ -} - -bool MockEmptyIO::is_never_timeout(int64_t /*timeout_us*/) -{ - return true; -} - -int MockEmptyIO::read_fully(void* /*buf*/, size_t /*size*/, ssize_t* /*nread*/) -{ - return ERROR_SUCCESS; -} - -int MockEmptyIO::write(void* /*buf*/, size_t /*size*/, ssize_t* /*nwrite*/) -{ - return ERROR_SUCCESS; -} - -void MockEmptyIO::set_recv_timeout(int64_t /*timeout_us*/) -{ -} - -int64_t MockEmptyIO::get_recv_timeout() -{ - return -1; -} - -int64_t MockEmptyIO::get_recv_bytes() -{ - return -1; -} - -void MockEmptyIO::set_send_timeout(int64_t /*timeout_us*/) -{ -} - -int64_t MockEmptyIO::get_send_timeout() -{ - return 0; -} - -int64_t MockEmptyIO::get_send_bytes() -{ - return 0; -} - -int MockEmptyIO::writev(const iovec */*iov*/, int /*iov_size*/, ssize_t* /*nwrite*/) -{ - return ERROR_SUCCESS; -} - -int MockEmptyIO::read(void* /*buf*/, size_t /*size*/, ssize_t* /*nread*/) -{ - return ERROR_SUCCESS; -} - // basic test and samples. VOID TEST(SampleTest, FastSampleInt64Test) { diff --git a/trunk/src/utest/srs_utest.hpp b/trunk/src/utest/srs_utest.hpp index c0fdfaa7d..239cc68c2 100644 --- a/trunk/src/utest/srs_utest.hpp +++ b/trunk/src/utest/srs_utest.hpp @@ -49,34 +49,4 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. // * {ASSERT|EXPECT}_DOUBLE_EQ(expected, actual): Tests that two double values are almost equal. // * {ASSERT|EXPECT}_NEAR(v1, v2, abs_error): Tests that v1 and v2 are within the given distance to each other. -#include - -class MockEmptyIO : public ISrsProtocolReaderWriter -{ -public: - MockEmptyIO(); - virtual ~MockEmptyIO(); -// for protocol -public: - virtual bool is_never_timeout(int64_t timeout_us); -// for handshake. -public: - virtual int read_fully(void* buf, size_t size, ssize_t* nread); - virtual int write(void* buf, size_t size, ssize_t* nwrite); -// for protocol -public: - virtual void set_recv_timeout(int64_t timeout_us); - virtual int64_t get_recv_timeout(); - virtual int64_t get_recv_bytes(); -// for protocol -public: - virtual void set_send_timeout(int64_t timeout_us); - virtual int64_t get_send_timeout(); - virtual int64_t get_send_bytes(); - virtual int writev(const iovec *iov, int iov_size, ssize_t* nwrite); -// for protocol/amf0/msg-codec -public: - virtual int read(void* buf, size_t size, ssize_t* nread); -}; - #endif diff --git a/trunk/src/utest/srs_utest_handshake.cpp b/trunk/src/utest/srs_utest_handshake.cpp index 77a6af4ce..d412328c3 100644 --- a/trunk/src/utest/srs_utest_handshake.cpp +++ b/trunk/src/utest/srs_utest_handshake.cpp @@ -26,6 +26,67 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. #include #include +MockEmptyIO::MockEmptyIO() +{ +} + +MockEmptyIO::~MockEmptyIO() +{ +} + +bool MockEmptyIO::is_never_timeout(int64_t /*timeout_us*/) +{ + return true; +} + +int MockEmptyIO::read_fully(void* /*buf*/, size_t /*size*/, ssize_t* /*nread*/) +{ + return ERROR_SUCCESS; +} + +int MockEmptyIO::write(void* /*buf*/, size_t /*size*/, ssize_t* /*nwrite*/) +{ + return ERROR_SUCCESS; +} + +void MockEmptyIO::set_recv_timeout(int64_t /*timeout_us*/) +{ +} + +int64_t MockEmptyIO::get_recv_timeout() +{ + return -1; +} + +int64_t MockEmptyIO::get_recv_bytes() +{ + return -1; +} + +void MockEmptyIO::set_send_timeout(int64_t /*timeout_us*/) +{ +} + +int64_t MockEmptyIO::get_send_timeout() +{ + return 0; +} + +int64_t MockEmptyIO::get_send_bytes() +{ + return 0; +} + +int MockEmptyIO::writev(const iovec */*iov*/, int /*iov_size*/, ssize_t* /*nwrite*/) +{ + return ERROR_SUCCESS; +} + +int MockEmptyIO::read(void* /*buf*/, size_t /*size*/, ssize_t* /*nread*/) +{ + return ERROR_SUCCESS; +} + #ifdef SRS_AUTO_SSL // verify the sha256 diff --git a/trunk/src/utest/srs_utest_handshake.hpp b/trunk/src/utest/srs_utest_handshake.hpp index 2d678bf7f..44780fb2e 100644 --- a/trunk/src/utest/srs_utest_handshake.hpp +++ b/trunk/src/utest/srs_utest_handshake.hpp @@ -36,4 +36,34 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. using namespace srs; #endif +#include + +class MockEmptyIO : public ISrsProtocolReaderWriter +{ +public: + MockEmptyIO(); + virtual ~MockEmptyIO(); +// for protocol +public: + virtual bool is_never_timeout(int64_t timeout_us); +// for handshake. +public: + virtual int read_fully(void* buf, size_t size, ssize_t* nread); + virtual int write(void* buf, size_t size, ssize_t* nwrite); +// for protocol +public: + virtual void set_recv_timeout(int64_t timeout_us); + virtual int64_t get_recv_timeout(); + virtual int64_t get_recv_bytes(); +// for protocol +public: + virtual void set_send_timeout(int64_t timeout_us); + virtual int64_t get_send_timeout(); + virtual int64_t get_send_bytes(); + virtual int writev(const iovec *iov, int iov_size, ssize_t* nwrite); +// for protocol/amf0/msg-codec +public: + virtual int read(void* buf, size_t size, ssize_t* nread); +}; + #endif