crashpad/util/stream/file_encoder_test.cc
Peter Boström 1aa478d161 Remove DISALLOW_* macros in crashpad
This change was partially scripted and partially done manually with vim
regex + manually placing the deleted constructors.

The script change looked for destructors in the public: section of a
class, if that existed the deleted constructors would go before the
destructor.

For manual placement I looked for any constructor in the public: section
of the corresponding class. If there wasn't one, then it would ideally
have gone as the first entry except below enums, classes and typedefs.
This may not have been perfect, but is hopefully good enough. Fingers
crossed.

#include "base/macros.h" is removed from files that don't use
ignore_result, which is the only other thing defined in base/macros.h.

Bug: chromium:1010217
Change-Id: I099526255a40b1ac1264904b4ece2f3f503c9418
Reviewed-on: https://chromium-review.googlesource.com/c/crashpad/crashpad/+/3171034
Reviewed-by: Mark Mentovai <mark@chromium.org>
Commit-Queue: Peter Boström <pbos@chromium.org>
2021-09-21 15:09:44 +00:00

112 lines
3.4 KiB
C++

// Copyright 2019 The Crashpad Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "util/stream/file_encoder.h"
#include <algorithm>
#include <memory>
#include <string>
#include "base/files/file_path.h"
#include "gtest/gtest.h"
#include "test/scoped_temp_dir.h"
#include "util/file/file_io.h"
#include "util/stream/file_output_stream.h"
namespace crashpad {
namespace test {
namespace {
constexpr size_t kBufferSize = 4096;
class FileEncoderTest : public testing::Test {
public:
FileEncoderTest() {}
void Verify(size_t size) {
std::string contents;
ASSERT_TRUE(LoggingReadEntireFile(decoded_, &contents));
ASSERT_EQ(contents.size(), size);
EXPECT_EQ(memcmp(deterministic_input_.get(), contents.data(), size), 0);
}
const uint8_t* BuildDeterministicInput(size_t size) {
deterministic_input_ = std::make_unique<uint8_t[]>(size);
uint8_t* deterministic_input_base = deterministic_input_.get();
while (size-- > 0)
deterministic_input_base[size] = static_cast<uint8_t>(size);
return deterministic_input_base;
}
void GenerateOrigFile(size_t size) {
ScopedFileHandle write_handle(OpenFileForWrite(
orig_, FileWriteMode::kCreateOrFail, FilePermissions::kOwnerOnly));
ASSERT_TRUE(write_handle.is_valid());
FileOutputStream out(write_handle.get());
const uint8_t* buf = BuildDeterministicInput(size);
while (size > 0) {
size_t m = std::min(kBufferSize, size);
ASSERT_TRUE(out.Write(buf, m));
size -= m;
buf += m;
}
ASSERT_TRUE(out.Flush());
}
FileEncoder* encoder() const { return encoder_.get(); }
FileEncoder* decoder() const { return decoder_.get(); }
protected:
void SetUp() override {
temp_dir_ = std::make_unique<ScopedTempDir>();
orig_ = base::FilePath(temp_dir_->path().Append(FILE_PATH_LITERAL("orig")));
encoded_ =
base::FilePath(temp_dir_->path().Append(FILE_PATH_LITERAL("encoded")));
decoded_ =
base::FilePath(temp_dir_->path().Append(FILE_PATH_LITERAL("decoded")));
encoder_ = std::make_unique<FileEncoder>(
FileEncoder::Mode::kEncode, orig_, encoded_);
decoder_ = std::make_unique<FileEncoder>(
FileEncoder::Mode::kDecode, encoded_, decoded_);
}
private:
std::unique_ptr<ScopedTempDir> temp_dir_;
base::FilePath orig_;
base::FilePath encoded_;
base::FilePath decoded_;
std::unique_ptr<FileEncoder> encoder_;
std::unique_ptr<FileEncoder> decoder_;
std::unique_ptr<uint8_t[]> deterministic_input_;
};
TEST_F(FileEncoderTest, ProcessShortFile) {
GenerateOrigFile(kBufferSize - 512);
EXPECT_TRUE(encoder()->Process());
EXPECT_TRUE(decoder()->Process());
Verify(kBufferSize - 512);
}
TEST_F(FileEncoderTest, ProcessLongFile) {
GenerateOrigFile(kBufferSize + 512);
EXPECT_TRUE(encoder()->Process());
EXPECT_TRUE(decoder()->Process());
Verify(kBufferSize + 512);
}
} // namespace
} // namespace test
} // namespace crashpad