1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071 |
- // Copyright 2017 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include "components/zucchini/buffer_sink.h"
- #include <stddef.h>
- #include <stdint.h>
- #include <vector>
- #include "testing/gtest/include/gtest/gtest.h"
- namespace zucchini {
- constexpr uint8_t kUninit = 0xFF;
- class BufferSinkTest : public testing::Test {
- protected:
- BufferSinkTest()
- : buffer_(10, kUninit), sink_(buffer_.data(), buffer_.size()) {}
- std::vector<uint8_t> buffer_;
- BufferSink sink_;
- };
- TEST_F(BufferSinkTest, PutValue) {
- EXPECT_EQ(size_t(10), sink_.Remaining());
- EXPECT_TRUE(sink_.PutValue(uint32_t(0x76543210)));
- EXPECT_EQ(size_t(6), sink_.Remaining());
- EXPECT_TRUE(sink_.PutValue(uint32_t(0xFEDCBA98)));
- EXPECT_EQ(size_t(2), sink_.Remaining());
- EXPECT_FALSE(sink_.PutValue(uint32_t(0x00)));
- EXPECT_EQ(size_t(2), sink_.Remaining());
- EXPECT_TRUE(sink_.PutValue(uint16_t(0x0010)));
- EXPECT_EQ(size_t(0), sink_.Remaining());
- // Assuming little-endian architecture.
- EXPECT_EQ(std::vector<uint8_t>(
- {0x10, 0x32, 0x54, 0x76, 0x98, 0xBA, 0xDC, 0xFE, 0x10, 0x00}),
- buffer_);
- }
- TEST_F(BufferSinkTest, PutRange) {
- std::vector<uint8_t> range = {0x10, 0x32, 0x54, 0x76, 0x98, 0xBA,
- 0xDC, 0xFE, 0x10, 0x00, 0x42};
- EXPECT_EQ(size_t(10), sink_.Remaining());
- EXPECT_FALSE(sink_.PutRange(range.begin(), range.end()));
- EXPECT_EQ(size_t(10), sink_.Remaining());
- EXPECT_TRUE(sink_.PutRange(range.begin(), range.begin() + 8));
- EXPECT_EQ(size_t(2), sink_.Remaining());
- EXPECT_EQ(std::vector<uint8_t>({0x10, 0x32, 0x54, 0x76, 0x98, 0xBA, 0xDC,
- 0xFE, kUninit, kUninit}),
- buffer_);
- EXPECT_FALSE(sink_.PutRange(range.begin(), range.begin() + 4));
- EXPECT_EQ(size_t(2), sink_.Remaining());
- // range is not written
- EXPECT_EQ(std::vector<uint8_t>({0x10, 0x32, 0x54, 0x76, 0x98, 0xBA, 0xDC,
- 0xFE, kUninit, kUninit}),
- buffer_);
- }
- } // namespace zucchini
|