mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-19 19:45:40 -04:00
8bef5cd49a
Change MemoryBufferCache to InMemoryModuleCache, moving it from Basic to Serialization. Another patch will start using it to manage module build more explicitly, but this is split out because it's mostly mechanical. Because of the move to Serialization we can no longer abuse the Preprocessor to forward it to the ASTReader. Besides the rename and file move, that means Preprocessor::Preprocessor has one fewer parameter and ASTReader::ASTReader has one more. llvm-svn: 355777
94 lines
2.9 KiB
C++
94 lines
2.9 KiB
C++
//===- InMemoryModuleCacheTest.cpp - InMemoryModuleCache tests ------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Serialization/InMemoryModuleCache.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
using namespace clang;
|
|
|
|
namespace {
|
|
|
|
std::unique_ptr<MemoryBuffer> getBuffer(int I) {
|
|
SmallVector<char, 8> Bytes;
|
|
raw_svector_ostream(Bytes) << "data:" << I;
|
|
return MemoryBuffer::getMemBuffer(StringRef(Bytes.data(), Bytes.size()), "",
|
|
/* RequiresNullTerminator = */ false);
|
|
}
|
|
|
|
TEST(InMemoryModuleCacheTest, addBuffer) {
|
|
auto B1 = getBuffer(1);
|
|
auto B2 = getBuffer(2);
|
|
auto B3 = getBuffer(3);
|
|
auto *RawB1 = B1.get();
|
|
auto *RawB2 = B2.get();
|
|
auto *RawB3 = B3.get();
|
|
|
|
// Add a few buffers.
|
|
InMemoryModuleCache Cache;
|
|
EXPECT_EQ(RawB1, &Cache.addBuffer("1", std::move(B1)));
|
|
EXPECT_EQ(RawB2, &Cache.addBuffer("2", std::move(B2)));
|
|
EXPECT_EQ(RawB3, &Cache.addBuffer("3", std::move(B3)));
|
|
EXPECT_EQ(RawB1, Cache.lookupBuffer("1"));
|
|
EXPECT_EQ(RawB2, Cache.lookupBuffer("2"));
|
|
EXPECT_EQ(RawB3, Cache.lookupBuffer("3"));
|
|
EXPECT_FALSE(Cache.isBufferFinal("1"));
|
|
EXPECT_FALSE(Cache.isBufferFinal("2"));
|
|
EXPECT_FALSE(Cache.isBufferFinal("3"));
|
|
|
|
// Remove the middle buffer.
|
|
EXPECT_FALSE(Cache.tryToRemoveBuffer("2"));
|
|
EXPECT_EQ(nullptr, Cache.lookupBuffer("2"));
|
|
EXPECT_FALSE(Cache.isBufferFinal("2"));
|
|
|
|
// Replace the middle buffer.
|
|
B2 = getBuffer(2);
|
|
RawB2 = B2.get();
|
|
EXPECT_EQ(RawB2, &Cache.addBuffer("2", std::move(B2)));
|
|
|
|
// Check that nothing is final.
|
|
EXPECT_FALSE(Cache.isBufferFinal("1"));
|
|
EXPECT_FALSE(Cache.isBufferFinal("2"));
|
|
EXPECT_FALSE(Cache.isBufferFinal("3"));
|
|
}
|
|
|
|
TEST(InMemoryModuleCacheTest, finalizeCurrentBuffers) {
|
|
// Add a buffer.
|
|
InMemoryModuleCache Cache;
|
|
auto B1 = getBuffer(1);
|
|
auto *RawB1 = B1.get();
|
|
Cache.addBuffer("1", std::move(B1));
|
|
ASSERT_FALSE(Cache.isBufferFinal("1"));
|
|
|
|
// Finalize it.
|
|
Cache.finalizeCurrentBuffers();
|
|
EXPECT_TRUE(Cache.isBufferFinal("1"));
|
|
EXPECT_TRUE(Cache.tryToRemoveBuffer("1"));
|
|
EXPECT_EQ(RawB1, Cache.lookupBuffer("1"));
|
|
EXPECT_TRUE(Cache.isBufferFinal("1"));
|
|
|
|
// Repeat.
|
|
auto B2 = getBuffer(2);
|
|
auto *RawB2 = B2.get();
|
|
Cache.addBuffer("2", std::move(B2));
|
|
EXPECT_FALSE(Cache.isBufferFinal("2"));
|
|
|
|
Cache.finalizeCurrentBuffers();
|
|
EXPECT_TRUE(Cache.isBufferFinal("1"));
|
|
EXPECT_TRUE(Cache.isBufferFinal("2"));
|
|
EXPECT_TRUE(Cache.tryToRemoveBuffer("1"));
|
|
EXPECT_TRUE(Cache.tryToRemoveBuffer("2"));
|
|
EXPECT_EQ(RawB1, Cache.lookupBuffer("1"));
|
|
EXPECT_EQ(RawB2, Cache.lookupBuffer("2"));
|
|
EXPECT_TRUE(Cache.isBufferFinal("1"));
|
|
EXPECT_TRUE(Cache.isBufferFinal("2"));
|
|
}
|
|
|
|
} // namespace
|