Files
carbon-lang/toolchain/source/source_buffer_test.cpp
T
Jon Ross-PerkinsandRichard Smith a93e621488 Add vfs support to toolchain. (#2888)
This adds vfs support to the toolchain, allowing Driver to take in-memory inputs in tests. As a consequence, I'm simplifying SourceBuffer: rather than allowing tests to pass in their own memory buffer, I'm using InMemoryFileSystem to push for greater consistency with production code. This does hit a quirk where I need to be careful about null terminator handling because fuzzer imports don't always have one, but that's probably more robust anyways.

Co-authored-by: Richard Smith <richard@metafoo.co.uk>
2023-06-12 13:19:57 -07:00

66 lines
2.1 KiB
C++

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#include "toolchain/source/source_buffer.h"
#include <gtest/gtest.h>
#include "llvm/ADT/SmallString.h"
#include "llvm/Support/VirtualFileSystem.h"
#include "llvm/Support/raw_ostream.h"
namespace Carbon::Testing {
namespace {
static constexpr llvm::StringLiteral TestFileName = "test.carbon";
TEST(SourceBufferTest, MissingFile) {
llvm::vfs::InMemoryFileSystem fs;
auto buffer = SourceBuffer::CreateFromFile(fs, TestFileName);
EXPECT_FALSE(buffer.ok());
}
TEST(SourceBufferTest, SimpleFile) {
llvm::vfs::InMemoryFileSystem fs;
CARBON_CHECK(fs.addFile(TestFileName, /*ModificationTime=*/0,
llvm::MemoryBuffer::getMemBuffer("Hello World")));
auto buffer = SourceBuffer::CreateFromFile(fs, TestFileName);
ASSERT_TRUE(buffer.ok()) << "Error message: " << buffer.error();
EXPECT_EQ(TestFileName, buffer->filename());
EXPECT_EQ("Hello World", buffer->text());
}
TEST(SourceBufferTest, NoNull) {
llvm::vfs::InMemoryFileSystem fs;
static constexpr char NoNull[3] = {'a', 'b', 'c'};
CARBON_CHECK(fs.addFile(
TestFileName, /*ModificationTime=*/0,
llvm::MemoryBuffer::getMemBuffer(llvm::StringRef(NoNull, sizeof(NoNull)),
/*BufferName=*/"",
/*RequiresNullTerminator=*/false)));
auto buffer = SourceBuffer::CreateFromFile(fs, TestFileName);
ASSERT_TRUE(buffer.ok()) << "Error message: " << buffer.error();
EXPECT_EQ(TestFileName, buffer->filename());
EXPECT_EQ("abc", buffer->text());
}
TEST(SourceBufferTest, EmptyFile) {
llvm::vfs::InMemoryFileSystem fs;
CARBON_CHECK(fs.addFile(TestFileName, /*ModificationTime=*/0,
llvm::MemoryBuffer::getMemBuffer("")));
auto buffer = SourceBuffer::CreateFromFile(fs, TestFileName);
ASSERT_TRUE(buffer.ok()) << "Error message: " << buffer.error();
EXPECT_EQ(TestFileName, buffer->filename());
EXPECT_EQ("", buffer->text());
}
} // namespace
} // namespace Carbon::Testing