carbon-lang/toolchain/source/source_buffer_test.cpp

// 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