cfx_memorystream_unittest.cpp

Go to the documentation of this file.

// Copyright 2019 The PDFium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
 
#include "core/fxcrt/cfx_memorystream.h"
 
#include "core/fxcrt/retain_ptr.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
 
TEST(CFXMemoryStreamTest, ReadBlockAtOffset) {
  auto stream = pdfium::MakeRetain<CFX_MemoryStream>();
  const uint8_t kData1[] = {'a', 'b', 'c', 'd', 'e', 'f'};
  ASSERT_TRUE(stream->WriteBlock(kData1));
  ASSERT_THAT(stream->GetSpan(),
              testing::ElementsAre('a', 'b', 'c', 'd', 'e', 'f'));
 
  uint8_t buffer[3];
  ASSERT_TRUE(stream->ReadBlockAtOffset(buffer, 2));
  ASSERT_THAT(buffer, testing::ElementsAre('c', 'd', 'e'));
 
  ASSERT_TRUE(stream->ReadBlockAtOffset(buffer, 0));
  ASSERT_THAT(buffer, testing::ElementsAre('a', 'b', 'c'));
 
  ASSERT_TRUE(stream->ReadBlockAtOffset(buffer, 3));
  ASSERT_THAT(buffer, testing::ElementsAre('d', 'e', 'f'));
 
  ASSERT_FALSE(stream->ReadBlockAtOffset(buffer, 4));
  ASSERT_THAT(buffer, testing::ElementsAre('d', 'e', 'f'));
}
 
TEST(CFXMemoryStreamTest, WriteBlock) {
  auto stream = pdfium::MakeRetain<CFX_MemoryStream>();
  ASSERT_TRUE(stream->WriteBlock({}));
  ASSERT_TRUE(stream->GetSpan().empty());
 
  const uint8_t kData1[] = {'a', 'b', 'c'};
  ASSERT_TRUE(stream->WriteBlock(kData1));
  ASSERT_THAT(stream->GetSpan(), testing::ElementsAre('a', 'b', 'c'));
 
  ASSERT_TRUE(stream->WriteBlock(kData1));
  ASSERT_THAT(stream->GetSpan(),
              testing::ElementsAre('a', 'b', 'c', 'a', 'b', 'c'));
 
  ASSERT_TRUE(stream->WriteBlock({}));
  ASSERT_THAT(stream->GetSpan(),
              testing::ElementsAre('a', 'b', 'c', 'a', 'b', 'c'));
}
 
TEST(CFXMemoryStreamTest, WriteZeroBytes) {
  auto stream = pdfium::MakeRetain<CFX_MemoryStream>();
  const uint8_t kData1[] = {'a', 'b', 'c'};
  ASSERT_TRUE(stream->WriteBlock(kData1));
  ASSERT_THAT(stream->GetSpan(), testing::ElementsAre('a', 'b', 'c'));
 
  ASSERT_TRUE(stream->WriteBlock({}));
  ASSERT_THAT(stream->GetSpan(), testing::ElementsAre('a', 'b', 'c'));
}