diff --git a/trunk/src/utest/srs_utest_kernel.cpp b/trunk/src/utest/srs_utest_kernel.cpp index 4dd81628e..f3420eac1 100644 --- a/trunk/src/utest/srs_utest_kernel.cpp +++ b/trunk/src/utest/srs_utest_kernel.cpp @@ -108,3 +108,89 @@ VOID TEST(KernelCodecTest, IsAudioSequenceHeader) pp[1] = 0x01; EXPECT_FALSE(SrsFlvCodec::video_is_sequence_header((int8_t*)pp, 2)); } + +MockSrsFileWriter::MockSrsFileWriter() +{ +} + +MockSrsFileWriter::~MockSrsFileWriter() +{ +} + +int MockSrsFileWriter::open(string file) +{ + int ret = ERROR_SUCCESS; + return ret; +} + +void MockSrsFileWriter::close() +{ + int ret = ERROR_SUCCESS; + return; +} + +bool MockSrsFileWriter::is_open() +{ + return true; +} + +int64_t MockSrsFileWriter::tellg() +{ + return 0; +} + +int MockSrsFileWriter::write(void* buf, size_t count, ssize_t* pnwrite) +{ + int ret = ERROR_SUCCESS; + return ret; +} + +MockSrsFileReader::MockSrsFileReader() +{ +} + +MockSrsFileReader::~MockSrsFileReader() +{ +} + +int MockSrsFileReader::open(string file) +{ + int ret = ERROR_SUCCESS; + return ret; +} + +void MockSrsFileReader::close() +{ + int ret = ERROR_SUCCESS; + return; +} + +bool MockSrsFileReader::is_open() +{ + return true; +} + +int64_t MockSrsFileReader::tellg() +{ + return 0; +} + +void MockSrsFileReader::skip(int64_t size) +{ +} + +int64_t MockSrsFileReader::lseek(int64_t offset) +{ + return offset; +} + +int64_t MockSrsFileReader::filesize() +{ + return 0; +} + +int MockSrsFileReader::read(void* buf, size_t count, ssize_t* pnread) +{ + int ret = ERROR_SUCCESS; + return ret; +} diff --git a/trunk/src/utest/srs_utest_kernel.hpp b/trunk/src/utest/srs_utest_kernel.hpp index d9d641b1f..f32543261 100644 --- a/trunk/src/utest/srs_utest_kernel.hpp +++ b/trunk/src/utest/srs_utest_kernel.hpp @@ -32,26 +32,37 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. #include #include -/* -class MockSrsFileStream : public SrsFileStream +class MockSrsFileWriter : public SrsFileWriter { public: - MockSrsFileStream(); - virtual ~MockSrsFileStream(); -public: + MockSrsFileWriter(); + virtual ~MockSrsFileWriter(); public: - virtual int open_write(std::string file); - virtual int open_read(std::string file); + virtual int open(std::string file); virtual void close(); +public: virtual bool is_open(); + virtual int64_t tellg(); public: - virtual int read(void* buf, size_t count, ssize_t* pnread); virtual int write(void* buf, size_t count, ssize_t* pnwrite); +}; + +class MockSrsFileReader : public SrsFileReader +{ +public: + MockSrsFileReader(); + virtual ~MockSrsFileReader(); +public: + virtual int open(std::string file); + virtual void close(); +public: + virtual bool is_open(); virtual int64_t tellg(); + virtual void skip(int64_t size); virtual int64_t lseek(int64_t offset); virtual int64_t filesize(); - virtual void skip(int64_t size); +public: + virtual int read(void* buf, size_t count, ssize_t* pnread); }; -*/ #endif