blob: b380756ea3061877a1855b07fcb2a69f70fca620 [file] [log] [blame] [edit]
// Copyright 2020 Google LLC
//
// 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
//
// https://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 "fuzzing/replay/test_file_buffer.h"
#include <cstdlib>
#include <string>
#include "absl/strings/str_cat.h"
#include "fuzzing/replay/file_util.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"
namespace fuzzing {
namespace {
TEST(TestFileBufferTest, EmptyBuffer) {
TestFileBuffer buffer(1024);
EXPECT_THAT(buffer.last_test(), testing::IsEmpty());
}
TEST(TestFileBufferTest, ReadsFileSuccessfully) {
const std::string test_file =
absl::StrCat(getenv("TEST_TMPDIR"), "/successful_test.txt");
ASSERT_TRUE(SetFileContents(test_file, "123456789").ok());
TestFileBuffer buffer(1024);
const absl::Status status = buffer.ReadFile(test_file);
EXPECT_TRUE(status.ok());
EXPECT_EQ(buffer.last_test(), "123456789");
}
TEST(TestFileBufferTest, FailsOnMissingFile) {
TestFileBuffer buffer(1024);
const absl::Status status = buffer.ReadFile("missing_path");
EXPECT_FALSE(status.ok());
EXPECT_THAT(buffer.last_test(), testing::IsEmpty());
}
TEST(TestFileBufferTest, TruncatesTooLargeFile) {
const std::string test_file =
absl::StrCat(getenv("TEST_TMPDIR"), "/truncated_test.txt");
ASSERT_TRUE(SetFileContents(test_file, "123456789").ok());
TestFileBuffer buffer(4);
const absl::Status status = buffer.ReadFile(test_file);
EXPECT_EQ(status.code(), absl::StatusCode::kResourceExhausted);
EXPECT_EQ(buffer.last_test(), "1234");
}
TEST(TestFileBufferTest, OverridesLastTest) {
const std::string first_test_file =
absl::StrCat(getenv("TEST_TMPDIR"), "/override_test_first.txt");
ASSERT_TRUE(SetFileContents(first_test_file, "123456789").ok());
const std::string second_test_file =
absl::StrCat(getenv("TEST_TMPDIR"), "/override_test_second.txt");
ASSERT_TRUE(SetFileContents(second_test_file, "ABCDEF").ok());
TestFileBuffer buffer(1024);
EXPECT_TRUE(buffer.ReadFile(first_test_file).ok());
EXPECT_EQ(buffer.last_test(), "123456789");
EXPECT_TRUE(buffer.ReadFile(second_test_file).ok());
EXPECT_EQ(buffer.last_test(), "ABCDEF");
EXPECT_FALSE(buffer.ReadFile("third_test_file_missing").ok());
EXPECT_THAT(buffer.last_test(), testing::IsEmpty());
}
} // namespace
} // namespace fuzzing