Index: webrtc/base/stream_unittest.cc |
diff --git a/webrtc/base/stream_unittest.cc b/webrtc/base/stream_unittest.cc |
deleted file mode 100644 |
index 0fbb6ab835f0bd5a1be1689796acd448f942b459..0000000000000000000000000000000000000000 |
--- a/webrtc/base/stream_unittest.cc |
+++ /dev/null |
@@ -1,374 +0,0 @@ |
-/* |
- * Copyright 2004 The WebRTC Project Authors. All rights reserved. |
- * |
- * Use of this source code is governed by a BSD-style license |
- * that can be found in the LICENSE file in the root of the source |
- * tree. An additional intellectual property rights grant can be found |
- * in the file PATENTS. All contributing project authors may |
- * be found in the AUTHORS file in the root of the source tree. |
- */ |
- |
-#include "webrtc/base/fileutils.h" |
-#include "webrtc/base/gunit.h" |
-#include "webrtc/base/pathutils.h" |
-#include "webrtc/base/stream.h" |
- |
-namespace rtc { |
- |
-/////////////////////////////////////////////////////////////////////////////// |
-// TestStream |
-/////////////////////////////////////////////////////////////////////////////// |
- |
-class TestStream : public StreamInterface { |
- public: |
- TestStream() : pos_(0) { } |
- |
- virtual StreamState GetState() const { return SS_OPEN; } |
- virtual StreamResult Read(void* buffer, size_t buffer_len, |
- size_t* read, int* error) { |
- unsigned char* uc_buffer = static_cast<unsigned char*>(buffer); |
- for (size_t i = 0; i < buffer_len; ++i) { |
- uc_buffer[i] = static_cast<unsigned char>(pos_++); |
- } |
- if (read) |
- *read = buffer_len; |
- return SR_SUCCESS; |
- } |
- virtual StreamResult Write(const void* data, size_t data_len, |
- size_t* written, int* error) { |
- if (error) |
- *error = -1; |
- return SR_ERROR; |
- } |
- virtual void Close() { } |
- virtual bool SetPosition(size_t position) { |
- pos_ = position; |
- return true; |
- } |
- virtual bool GetPosition(size_t* position) const { |
- if (position) *position = pos_; |
- return true; |
- } |
- virtual bool GetSize(size_t* size) const { |
- return false; |
- } |
- virtual bool GetAvailable(size_t* size) const { |
- return false; |
- } |
- |
- private: |
- size_t pos_; |
-}; |
- |
-bool VerifyTestBuffer(unsigned char* buffer, size_t len, |
- unsigned char value) { |
- bool passed = true; |
- for (size_t i = 0; i < len; ++i) { |
- if (buffer[i] != value++) { |
- passed = false; |
- break; |
- } |
- } |
- // Ensure that we don't pass again without re-writing |
- memset(buffer, 0, len); |
- return passed; |
-} |
- |
-void SeekTest(StreamInterface* stream, const unsigned char value) { |
- size_t bytes; |
- unsigned char buffer[13] = { 0 }; |
- const size_t kBufSize = sizeof(buffer); |
- |
- EXPECT_EQ(stream->Read(buffer, kBufSize, &bytes, nullptr), SR_SUCCESS); |
- EXPECT_EQ(bytes, kBufSize); |
- EXPECT_TRUE(VerifyTestBuffer(buffer, kBufSize, value)); |
- EXPECT_TRUE(stream->GetPosition(&bytes)); |
- EXPECT_EQ(13U, bytes); |
- |
- EXPECT_TRUE(stream->SetPosition(7)); |
- |
- EXPECT_EQ(stream->Read(buffer, kBufSize, &bytes, nullptr), SR_SUCCESS); |
- EXPECT_EQ(bytes, kBufSize); |
- EXPECT_TRUE(VerifyTestBuffer(buffer, kBufSize, value + 7)); |
- EXPECT_TRUE(stream->GetPosition(&bytes)); |
- EXPECT_EQ(20U, bytes); |
-} |
- |
-TEST(FifoBufferTest, TestAll) { |
- const size_t kSize = 16; |
- const char in[kSize * 2 + 1] = "0123456789ABCDEFGHIJKLMNOPQRSTUV"; |
- char out[kSize * 2]; |
- void* p; |
- const void* q; |
- size_t bytes; |
- FifoBuffer buf(kSize); |
- StreamInterface* stream = &buf; |
- |
- // Test assumptions about base state |
- EXPECT_EQ(SS_OPEN, stream->GetState()); |
- EXPECT_EQ(SR_BLOCK, stream->Read(out, kSize, &bytes, nullptr)); |
- EXPECT_TRUE(nullptr != stream->GetReadData(&bytes)); |
- EXPECT_EQ((size_t)0, bytes); |
- stream->ConsumeReadData(0); |
- EXPECT_TRUE(nullptr != stream->GetWriteBuffer(&bytes)); |
- EXPECT_EQ(kSize, bytes); |
- stream->ConsumeWriteBuffer(0); |
- |
- // Try a full write |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize, &bytes, nullptr)); |
- EXPECT_EQ(kSize, bytes); |
- |
- // Try a write that should block |
- EXPECT_EQ(SR_BLOCK, stream->Write(in, kSize, &bytes, nullptr)); |
- |
- // Try a full read |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize, &bytes, nullptr)); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize)); |
- |
- // Try a read that should block |
- EXPECT_EQ(SR_BLOCK, stream->Read(out, kSize, &bytes, nullptr)); |
- |
- // Try a too-big write |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize * 2, &bytes, nullptr)); |
- EXPECT_EQ(bytes, kSize); |
- |
- // Try a too-big read |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize * 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize)); |
- |
- // Try some small writes and reads |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize / 2)); |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize / 2)); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize / 2)); |
- |
- // Try wraparound reads and writes in the following pattern |
- // WWWWWWWWWWWW.... 0123456789AB.... |
- // RRRRRRRRXXXX.... ........89AB.... |
- // WWWW....XXXXWWWW 4567....89AB0123 |
- // XXXX....RRRRXXXX 4567........0123 |
- // XXXXWWWWWWWWXXXX 4567012345670123 |
- // RRRRXXXXXXXXRRRR ....01234567.... |
- // ....RRRRRRRR.... ................ |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize * 3 / 4, &bytes, nullptr)); |
- EXPECT_EQ(kSize * 3 / 4, bytes); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize / 2)); |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize / 4, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 4 , bytes); |
- EXPECT_EQ(0, memcmp(in + kSize / 2, out, kSize / 4)); |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2 , bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize / 2)); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize / 2 , bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize / 2)); |
- |
- // Use GetWriteBuffer to reset the read_position for the next tests |
- stream->GetWriteBuffer(&bytes); |
- stream->ConsumeWriteBuffer(0); |
- |
- // Try using GetReadData to do a full read |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize, &bytes, nullptr)); |
- q = stream->GetReadData(&bytes); |
- EXPECT_TRUE(nullptr != q); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_EQ(0, memcmp(q, in, kSize)); |
- stream->ConsumeReadData(kSize); |
- EXPECT_EQ(SR_BLOCK, stream->Read(out, kSize, &bytes, nullptr)); |
- |
- // Try using GetReadData to do some small reads |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize, &bytes, nullptr)); |
- q = stream->GetReadData(&bytes); |
- EXPECT_TRUE(nullptr != q); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_EQ(0, memcmp(q, in, kSize / 2)); |
- stream->ConsumeReadData(kSize / 2); |
- q = stream->GetReadData(&bytes); |
- EXPECT_TRUE(nullptr != q); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(0, memcmp(q, in + kSize / 2, kSize / 2)); |
- stream->ConsumeReadData(kSize / 2); |
- EXPECT_EQ(SR_BLOCK, stream->Read(out, kSize, &bytes, nullptr)); |
- |
- // Try using GetReadData in a wraparound case |
- // WWWWWWWWWWWWWWWW 0123456789ABCDEF |
- // RRRRRRRRRRRRXXXX ............CDEF |
- // WWWWWWWW....XXXX 01234567....CDEF |
- // ............RRRR 01234567........ |
- // RRRRRRRR........ ................ |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize, &bytes, nullptr)); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize * 3 / 4, &bytes, nullptr)); |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize / 2, &bytes, nullptr)); |
- q = stream->GetReadData(&bytes); |
- EXPECT_TRUE(nullptr != q); |
- EXPECT_EQ(kSize / 4, bytes); |
- EXPECT_EQ(0, memcmp(q, in + kSize * 3 / 4, kSize / 4)); |
- stream->ConsumeReadData(kSize / 4); |
- q = stream->GetReadData(&bytes); |
- EXPECT_TRUE(nullptr != q); |
- EXPECT_EQ(kSize / 2, bytes); |
- EXPECT_EQ(0, memcmp(q, in, kSize / 2)); |
- stream->ConsumeReadData(kSize / 2); |
- |
- // Use GetWriteBuffer to reset the read_position for the next tests |
- stream->GetWriteBuffer(&bytes); |
- stream->ConsumeWriteBuffer(0); |
- |
- // Try using GetWriteBuffer to do a full write |
- p = stream->GetWriteBuffer(&bytes); |
- EXPECT_TRUE(nullptr != p); |
- EXPECT_EQ(kSize, bytes); |
- memcpy(p, in, kSize); |
- stream->ConsumeWriteBuffer(kSize); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize, &bytes, nullptr)); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize)); |
- |
- // Try using GetWriteBuffer to do some small writes |
- p = stream->GetWriteBuffer(&bytes); |
- EXPECT_TRUE(nullptr != p); |
- EXPECT_EQ(kSize, bytes); |
- memcpy(p, in, kSize / 2); |
- stream->ConsumeWriteBuffer(kSize / 2); |
- p = stream->GetWriteBuffer(&bytes); |
- EXPECT_TRUE(nullptr != p); |
- EXPECT_EQ(kSize / 2, bytes); |
- memcpy(p, in + kSize / 2, kSize / 2); |
- stream->ConsumeWriteBuffer(kSize / 2); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize, &bytes, nullptr)); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize)); |
- |
- // Try using GetWriteBuffer in a wraparound case |
- // WWWWWWWWWWWW.... 0123456789AB.... |
- // RRRRRRRRXXXX.... ........89AB.... |
- // ........XXXXWWWW ........89AB0123 |
- // WWWW....XXXXXXXX 4567....89AB0123 |
- // RRRR....RRRRRRRR ................ |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize * 3 / 4, &bytes, nullptr)); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize / 2, &bytes, nullptr)); |
- p = stream->GetWriteBuffer(&bytes); |
- EXPECT_TRUE(nullptr != p); |
- EXPECT_EQ(kSize / 4, bytes); |
- memcpy(p, in, kSize / 4); |
- stream->ConsumeWriteBuffer(kSize / 4); |
- p = stream->GetWriteBuffer(&bytes); |
- EXPECT_TRUE(nullptr != p); |
- EXPECT_EQ(kSize / 2, bytes); |
- memcpy(p, in + kSize / 4, kSize / 4); |
- stream->ConsumeWriteBuffer(kSize / 4); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize * 3 / 4, &bytes, nullptr)); |
- EXPECT_EQ(kSize * 3 / 4, bytes); |
- EXPECT_EQ(0, memcmp(in + kSize / 2, out, kSize / 4)); |
- EXPECT_EQ(0, memcmp(in, out + kSize / 4, kSize / 4)); |
- |
- // Check that the stream is now empty |
- EXPECT_EQ(SR_BLOCK, stream->Read(out, kSize, &bytes, nullptr)); |
- |
- // Try growing the buffer |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize, &bytes, nullptr)); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_TRUE(buf.SetCapacity(kSize * 2)); |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in + kSize, kSize, &bytes, nullptr)); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize * 2, &bytes, nullptr)); |
- EXPECT_EQ(kSize * 2, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize * 2)); |
- |
- // Try shrinking the buffer |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize, &bytes, nullptr)); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_TRUE(buf.SetCapacity(kSize)); |
- EXPECT_EQ(SR_BLOCK, stream->Write(in, kSize, &bytes, nullptr)); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize, &bytes, nullptr)); |
- EXPECT_EQ(kSize, bytes); |
- EXPECT_EQ(0, memcmp(in, out, kSize)); |
- |
- // Write to the stream, close it, read the remaining bytes |
- EXPECT_EQ(SR_SUCCESS, stream->Write(in, kSize / 2, &bytes, nullptr)); |
- stream->Close(); |
- EXPECT_EQ(SS_CLOSED, stream->GetState()); |
- EXPECT_EQ(SR_EOS, stream->Write(in, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(SR_SUCCESS, stream->Read(out, kSize / 2, &bytes, nullptr)); |
- EXPECT_EQ(0, memcmp(in, out, kSize / 2)); |
- EXPECT_EQ(SR_EOS, stream->Read(out, kSize / 2, &bytes, nullptr)); |
-} |
- |
-TEST(FifoBufferTest, FullBufferCheck) { |
- FifoBuffer buff(10); |
- buff.ConsumeWriteBuffer(10); |
- |
- size_t free; |
- EXPECT_TRUE(buff.GetWriteBuffer(&free) != nullptr); |
- EXPECT_EQ(0U, free); |
-} |
- |
-TEST(FifoBufferTest, WriteOffsetAndReadOffset) { |
- const size_t kSize = 16; |
- const char in[kSize * 2 + 1] = "0123456789ABCDEFGHIJKLMNOPQRSTUV"; |
- char out[kSize * 2]; |
- FifoBuffer buf(kSize); |
- |
- // Write 14 bytes. |
- EXPECT_EQ(SR_SUCCESS, buf.Write(in, 14, nullptr, nullptr)); |
- |
- // Make sure data is in |buf|. |
- size_t buffered; |
- EXPECT_TRUE(buf.GetBuffered(&buffered)); |
- EXPECT_EQ(14u, buffered); |
- |
- // Read 10 bytes. |
- buf.ConsumeReadData(10); |
- |
- // There should be now 12 bytes of available space. |
- size_t remaining; |
- EXPECT_TRUE(buf.GetWriteRemaining(&remaining)); |
- EXPECT_EQ(12u, remaining); |
- |
- // Write at offset 12, this should fail. |
- EXPECT_EQ(SR_BLOCK, buf.WriteOffset(in, 10, 12, nullptr)); |
- |
- // Write 8 bytes at offset 4, this wraps around the buffer. |
- EXPECT_EQ(SR_SUCCESS, buf.WriteOffset(in, 8, 4, nullptr)); |
- |
- // Number of available space remains the same until we call |
- // ConsumeWriteBuffer(). |
- EXPECT_TRUE(buf.GetWriteRemaining(&remaining)); |
- EXPECT_EQ(12u, remaining); |
- buf.ConsumeWriteBuffer(12); |
- |
- // There's 4 bytes bypassed and 4 bytes no read so skip them and verify the |
- // 8 bytes written. |
- size_t read; |
- EXPECT_EQ(SR_SUCCESS, buf.ReadOffset(out, 8, 8, &read)); |
- EXPECT_EQ(8u, read); |
- EXPECT_EQ(0, memcmp(out, in, 8)); |
- |
- // There should still be 16 bytes available for reading. |
- EXPECT_TRUE(buf.GetBuffered(&buffered)); |
- EXPECT_EQ(16u, buffered); |
- |
- // Read at offset 16, this should fail since we don't have that much data. |
- EXPECT_EQ(SR_BLOCK, buf.ReadOffset(out, 10, 16, nullptr)); |
-} |
- |
-} // namespace rtc |