bit_reader_unittest.cc revision 5821806d5e7f356e8fa4b058a389a808ea183019 
1// Copyright (c) 2012 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 "media/base/bit_reader.h"
6
7#include "testing/gtest/include/gtest/gtest.h"
8
9namespace media {
10
11TEST(BitReaderTest, NormalOperationTest) {
12  uint8 value8;
13  uint64 value64;
14  // 0101 0101 1001 1001 repeats 4 times
15  uint8 buffer[] = {0x55, 0x99, 0x55, 0x99, 0x55, 0x99, 0x55, 0x99};
16  BitReader reader1(buffer, 6);  // Initialize with 6 bytes only
17
18  EXPECT_TRUE(reader1.ReadBits(1, &value8));
19  EXPECT_EQ(value8, 0);
20  EXPECT_TRUE(reader1.ReadBits(8, &value8));
21  EXPECT_EQ(value8, 0xab);  // 1010 1011
22  EXPECT_TRUE(reader1.ReadBits(7, &value64));
23  EXPECT_TRUE(reader1.ReadBits(32, &value64));
24  EXPECT_EQ(value64, 0x55995599u);
25  EXPECT_FALSE(reader1.ReadBits(1, &value8));
26  value8 = 0xff;
27  EXPECT_TRUE(reader1.ReadBits(0, &value8));
28  EXPECT_EQ(value8, 0);
29
30  BitReader reader2(buffer, 8);
31  EXPECT_TRUE(reader2.ReadBits(64, &value64));
32  EXPECT_EQ(value64, 0x5599559955995599ull);
33  EXPECT_FALSE(reader2.ReadBits(1, &value8));
34  EXPECT_TRUE(reader2.ReadBits(0, &value8));
35}
36
37TEST(BitReaderTest, ReadBeyondEndTest) {
38  uint8 value8;
39  uint8 buffer[] = {0x12};
40  BitReader reader1(buffer, sizeof(buffer));
41
42  EXPECT_TRUE(reader1.ReadBits(4, &value8));
43  EXPECT_FALSE(reader1.ReadBits(5, &value8));
44  EXPECT_FALSE(reader1.ReadBits(1, &value8));
45  EXPECT_TRUE(reader1.ReadBits(0, &value8));
46}
47
48}  // namespace media
49