63 lines
1.7 KiB
C++
63 lines
1.7 KiB
C++
// Copyright 2017 Google LLC. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include "packager/media/base/bit_writer.h"
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
using testing::ElementsAreArray;
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
TEST(BitWriterTest, Simple) {
|
|
std::vector<uint8_t> storage;
|
|
BitWriter writer(&storage);
|
|
writer.WriteBits(1, 1);
|
|
EXPECT_EQ(1u, writer.BitPos());
|
|
EXPECT_EQ(0u, writer.BytePos());
|
|
writer.Flush();
|
|
// Bits are byte-aligned after flush.
|
|
EXPECT_EQ(8u, writer.BitPos());
|
|
EXPECT_EQ(1u, writer.BytePos());
|
|
|
|
EXPECT_THAT(storage, ElementsAreArray({0x80}));
|
|
}
|
|
|
|
TEST(BitWriterTest, Test) {
|
|
std::vector<uint8_t> storage;
|
|
BitWriter writer(&storage);
|
|
writer.WriteBits(0, 1);
|
|
EXPECT_EQ(1u, writer.BitPos());
|
|
EXPECT_EQ(0u, writer.BytePos());
|
|
writer.WriteBits(0xab, 8);
|
|
EXPECT_EQ(9u, writer.BitPos());
|
|
EXPECT_EQ(1u, writer.BytePos());
|
|
writer.WriteBits(0x34, 6);
|
|
EXPECT_EQ(15u, writer.BitPos());
|
|
EXPECT_EQ(1u, writer.BytePos());
|
|
writer.WriteBits(0x55995599, 32);
|
|
EXPECT_EQ(47u, writer.BitPos());
|
|
EXPECT_EQ(5u, writer.BytePos());
|
|
writer.WriteBits(1, 1);
|
|
EXPECT_EQ(48u, writer.BitPos());
|
|
EXPECT_EQ(6u, writer.BytePos());
|
|
writer.WriteBits(0x13, 21);
|
|
EXPECT_EQ(69u, writer.BitPos());
|
|
EXPECT_EQ(8u, writer.BytePos());
|
|
writer.Flush();
|
|
// Bits are byte-aligned after flush.
|
|
EXPECT_EQ(72u, writer.BitPos());
|
|
EXPECT_EQ(9u, writer.BytePos());
|
|
|
|
EXPECT_THAT(storage, ElementsAreArray({0x55, 0xe8, 0xab, 0x32, 0xab, 0x33,
|
|
0x00, 0x00, 0x98}));
|
|
}
|
|
|
|
} // namespace media
|
|
} // namespace shaka
|