big_endian_unittest.cc revision c2e0dbddbe15c98d52c4786dac06cb8952a8ae6d 
1// Copyright (c) 2011 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "base/strings/string_piece.h"
6#include "net/base/big_endian.h"
7#include "testing/gtest/include/gtest/gtest.h"
8
9namespace net {
10
11TEST(BigEndianReaderTest, ReadsValues) {
12  char data[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0xA, 0xB, 0xC };
13  char buf[2];
14  uint8 u8;
15  uint16 u16;
16  uint32 u32;
17  base::StringPiece piece;
18  BigEndianReader reader(data, sizeof(data));
19
20  EXPECT_TRUE(reader.Skip(2));
21  EXPECT_EQ(data + 2, reader.ptr());
22  EXPECT_EQ(reader.remaining(), static_cast<int>(sizeof(data)) - 2);
23  EXPECT_TRUE(reader.ReadBytes(buf, sizeof(buf)));
24  EXPECT_EQ(0x2, buf[0]);
25  EXPECT_EQ(0x3, buf[1]);
26  EXPECT_TRUE(reader.ReadU8(&u8));
27  EXPECT_EQ(0x4, u8);
28  EXPECT_TRUE(reader.ReadU16(&u16));
29  EXPECT_EQ(0x0506, u16);
30  EXPECT_TRUE(reader.ReadU32(&u32));
31  EXPECT_EQ(0x0708090Au, u32);
32  base::StringPiece expected(reader.ptr(), 2);
33  EXPECT_TRUE(reader.ReadPiece(&piece, 2));
34  EXPECT_EQ(2u, piece.size());
35  EXPECT_EQ(expected.data(), piece.data());
36}
37
38TEST(BigEndianReaderTest, RespectsLength) {
39  char data[4];
40  char buf[2];
41  uint8 u8;
42  uint16 u16;
43  uint32 u32;
44  base::StringPiece piece;
45  BigEndianReader reader(data, sizeof(data));
46  // 4 left
47  EXPECT_FALSE(reader.Skip(6));
48  EXPECT_TRUE(reader.Skip(1));
49  // 3 left
50  EXPECT_FALSE(reader.ReadU32(&u32));
51  EXPECT_FALSE(reader.ReadPiece(&piece, 4));
52  EXPECT_TRUE(reader.Skip(2));
53  // 1 left
54  EXPECT_FALSE(reader.ReadU16(&u16));
55  EXPECT_FALSE(reader.ReadBytes(buf, 2));
56  EXPECT_TRUE(reader.Skip(1));
57  // 0 left
58  EXPECT_FALSE(reader.ReadU8(&u8));
59  EXPECT_EQ(0, reader.remaining());
60}
61
62TEST(BigEndianWriterTest, WritesValues) {
63  char expected[] = { 0, 0, 2, 3, 4, 5, 6, 7, 8, 9, 0xA };
64  char data[sizeof(expected)];
65  char buf[] = { 0x2, 0x3 };
66  memset(data, 0, sizeof(data));
67  BigEndianWriter writer(data, sizeof(data));
68
69  EXPECT_TRUE(writer.Skip(2));
70  EXPECT_TRUE(writer.WriteBytes(buf, sizeof(buf)));
71  EXPECT_TRUE(writer.WriteU8(0x4));
72  EXPECT_TRUE(writer.WriteU16(0x0506));
73  EXPECT_TRUE(writer.WriteU32(0x0708090A));
74  EXPECT_EQ(0, memcmp(expected, data, sizeof(expected)));
75}
76
77TEST(BigEndianWriterTest, RespectsLength) {
78  char data[4];
79  char buf[2];
80  uint8 u8 = 0;
81  uint16 u16 = 0;
82  uint32 u32 = 0;
83  BigEndianWriter writer(data, sizeof(data));
84  // 4 left
85  EXPECT_FALSE(writer.Skip(6));
86  EXPECT_TRUE(writer.Skip(1));
87  // 3 left
88  EXPECT_FALSE(writer.WriteU32(u32));
89  EXPECT_TRUE(writer.Skip(2));
90  // 1 left
91  EXPECT_FALSE(writer.WriteU16(u16));
92  EXPECT_FALSE(writer.WriteBytes(buf, 2));
93  EXPECT_TRUE(writer.Skip(1));
94  // 0 left
95  EXPECT_FALSE(writer.WriteU8(u8));
96  EXPECT_EQ(0, writer.remaining());
97}
98
99}  // namespace net
100
101