Adds a modified copy of talk/base to webrtc/base. It is the first step in migrating talk/base to webrtc/base. BUG=N/A R=niklas.enbom@webrtc.org Review URL: https://webrtc-codereview.appspot.com/17479005 git-svn-id: http://webrtc.googlecode.com/svn/trunk@6129 4adac7df-926f-26a2-2b94-8c16560cd09d
diff --git a/webrtc/base/bytebuffer_unittest.cc b/webrtc/base/bytebuffer_unittest.cc new file mode 100644 index 0000000..f4b0504 --- /dev/null +++ b/webrtc/base/bytebuffer_unittest.cc
@@ -0,0 +1,211 @@ +/* + * 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/bytebuffer.h" +#include "webrtc/base/byteorder.h" +#include "webrtc/base/common.h" +#include "webrtc/base/gunit.h" + +namespace rtc { + +TEST(ByteBufferTest, TestByteOrder) { + uint16 n16 = 1; + uint32 n32 = 1; + uint64 n64 = 1; + + EXPECT_EQ(n16, NetworkToHost16(HostToNetwork16(n16))); + EXPECT_EQ(n32, NetworkToHost32(HostToNetwork32(n32))); + EXPECT_EQ(n64, NetworkToHost64(HostToNetwork64(n64))); + + if (IsHostBigEndian()) { + // The host is the network (big) endian. + EXPECT_EQ(n16, HostToNetwork16(n16)); + EXPECT_EQ(n32, HostToNetwork32(n32)); + EXPECT_EQ(n64, HostToNetwork64(n64)); + + // GetBE converts big endian to little endian here. + EXPECT_EQ(n16 >> 8, GetBE16(&n16)); + EXPECT_EQ(n32 >> 24, GetBE32(&n32)); + EXPECT_EQ(n64 >> 56, GetBE64(&n64)); + } else { + // The host is little endian. + EXPECT_NE(n16, HostToNetwork16(n16)); + EXPECT_NE(n32, HostToNetwork32(n32)); + EXPECT_NE(n64, HostToNetwork64(n64)); + + // GetBE converts little endian to big endian here. + EXPECT_EQ(GetBE16(&n16), HostToNetwork16(n16)); + EXPECT_EQ(GetBE32(&n32), HostToNetwork32(n32)); + EXPECT_EQ(GetBE64(&n64), HostToNetwork64(n64)); + + // GetBE converts little endian to big endian here. + EXPECT_EQ(n16 << 8, GetBE16(&n16)); + EXPECT_EQ(n32 << 24, GetBE32(&n32)); + EXPECT_EQ(n64 << 56, GetBE64(&n64)); + } +} + +TEST(ByteBufferTest, TestBufferLength) { + ByteBuffer buffer; + size_t size = 0; + EXPECT_EQ(size, buffer.Length()); + + buffer.WriteUInt8(1); + ++size; + EXPECT_EQ(size, buffer.Length()); + + buffer.WriteUInt16(1); + size += 2; + EXPECT_EQ(size, buffer.Length()); + + buffer.WriteUInt24(1); + size += 3; + EXPECT_EQ(size, buffer.Length()); + + buffer.WriteUInt32(1); + size += 4; + EXPECT_EQ(size, buffer.Length()); + + buffer.WriteUInt64(1); + size += 8; + EXPECT_EQ(size, buffer.Length()); + + EXPECT_TRUE(buffer.Consume(0)); + EXPECT_EQ(size, buffer.Length()); + + EXPECT_TRUE(buffer.Consume(4)); + size -= 4; + EXPECT_EQ(size, buffer.Length()); +} + +TEST(ByteBufferTest, TestGetSetReadPosition) { + ByteBuffer buffer("ABCDEF", 6); + EXPECT_EQ(6U, buffer.Length()); + ByteBuffer::ReadPosition pos(buffer.GetReadPosition()); + EXPECT_TRUE(buffer.SetReadPosition(pos)); + EXPECT_EQ(6U, buffer.Length()); + std::string read; + EXPECT_TRUE(buffer.ReadString(&read, 3)); + EXPECT_EQ("ABC", read); + EXPECT_EQ(3U, buffer.Length()); + EXPECT_TRUE(buffer.SetReadPosition(pos)); + EXPECT_EQ(6U, buffer.Length()); + read.clear(); + EXPECT_TRUE(buffer.ReadString(&read, 3)); + EXPECT_EQ("ABC", read); + EXPECT_EQ(3U, buffer.Length()); + // For a resize by writing Capacity() number of bytes. + size_t capacity = buffer.Capacity(); + buffer.ReserveWriteBuffer(buffer.Capacity()); + EXPECT_EQ(capacity + 3U, buffer.Length()); + EXPECT_FALSE(buffer.SetReadPosition(pos)); + read.clear(); + EXPECT_TRUE(buffer.ReadString(&read, 3)); + EXPECT_EQ("DEF", read); +} + +TEST(ByteBufferTest, TestReadWriteBuffer) { + ByteBuffer::ByteOrder orders[2] = { ByteBuffer::ORDER_HOST, + ByteBuffer::ORDER_NETWORK }; + for (size_t i = 0; i < ARRAY_SIZE(orders); i++) { + ByteBuffer buffer(orders[i]); + EXPECT_EQ(orders[i], buffer.Order()); + uint8 ru8; + EXPECT_FALSE(buffer.ReadUInt8(&ru8)); + + // Write and read uint8. + uint8 wu8 = 1; + buffer.WriteUInt8(wu8); + EXPECT_TRUE(buffer.ReadUInt8(&ru8)); + EXPECT_EQ(wu8, ru8); + EXPECT_EQ(0U, buffer.Length()); + + // Write and read uint16. + uint16 wu16 = (1 << 8) + 1; + buffer.WriteUInt16(wu16); + uint16 ru16; + EXPECT_TRUE(buffer.ReadUInt16(&ru16)); + EXPECT_EQ(wu16, ru16); + EXPECT_EQ(0U, buffer.Length()); + + // Write and read uint24. + uint32 wu24 = (3 << 16) + (2 << 8) + 1; + buffer.WriteUInt24(wu24); + uint32 ru24; + EXPECT_TRUE(buffer.ReadUInt24(&ru24)); + EXPECT_EQ(wu24, ru24); + EXPECT_EQ(0U, buffer.Length()); + + // Write and read uint32. + uint32 wu32 = (4 << 24) + (3 << 16) + (2 << 8) + 1; + buffer.WriteUInt32(wu32); + uint32 ru32; + EXPECT_TRUE(buffer.ReadUInt32(&ru32)); + EXPECT_EQ(wu32, ru32); + EXPECT_EQ(0U, buffer.Length()); + + // Write and read uint64. + uint32 another32 = (8 << 24) + (7 << 16) + (6 << 8) + 5; + uint64 wu64 = (static_cast<uint64>(another32) << 32) + wu32; + buffer.WriteUInt64(wu64); + uint64 ru64; + EXPECT_TRUE(buffer.ReadUInt64(&ru64)); + EXPECT_EQ(wu64, ru64); + EXPECT_EQ(0U, buffer.Length()); + + // Write and read string. + std::string write_string("hello"); + buffer.WriteString(write_string); + std::string read_string; + EXPECT_TRUE(buffer.ReadString(&read_string, write_string.size())); + EXPECT_EQ(write_string, read_string); + EXPECT_EQ(0U, buffer.Length()); + + // Write and read bytes + char write_bytes[] = "foo"; + buffer.WriteBytes(write_bytes, 3); + char read_bytes[3]; + EXPECT_TRUE(buffer.ReadBytes(read_bytes, 3)); + for (int i = 0; i < 3; ++i) { + EXPECT_EQ(write_bytes[i], read_bytes[i]); + } + EXPECT_EQ(0U, buffer.Length()); + + // Write and read reserved buffer space + char* write_dst = buffer.ReserveWriteBuffer(3); + memcpy(write_dst, write_bytes, 3); + memset(read_bytes, 0, 3); + EXPECT_TRUE(buffer.ReadBytes(read_bytes, 3)); + for (int i = 0; i < 3; ++i) { + EXPECT_EQ(write_bytes[i], read_bytes[i]); + } + EXPECT_EQ(0U, buffer.Length()); + + // Write and read in order. + buffer.WriteUInt8(wu8); + buffer.WriteUInt16(wu16); + buffer.WriteUInt24(wu24); + buffer.WriteUInt32(wu32); + buffer.WriteUInt64(wu64); + EXPECT_TRUE(buffer.ReadUInt8(&ru8)); + EXPECT_EQ(wu8, ru8); + EXPECT_TRUE(buffer.ReadUInt16(&ru16)); + EXPECT_EQ(wu16, ru16); + EXPECT_TRUE(buffer.ReadUInt24(&ru24)); + EXPECT_EQ(wu24, ru24); + EXPECT_TRUE(buffer.ReadUInt32(&ru32)); + EXPECT_EQ(wu32, ru32); + EXPECT_TRUE(buffer.ReadUInt64(&ru64)); + EXPECT_EQ(wu64, ru64); + EXPECT_EQ(0U, buffer.Length()); + } +} + +} // namespace rtc