mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 20:30:14 +01:00
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>
66 lines
2.1 KiB
C++
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
|