158 lines · cpp
1//===- Caching.cpp --------------------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "llvm/Support/Caching.h"10#include "llvm/Support/Error.h"11#include "llvm/Support/MemoryBuffer.h"12#include "llvm/Support/Path.h"13#include "llvm/Testing/Support/Error.h"14#include "gtest/gtest.h"15 16using namespace llvm;17 18#define ASSERT_NO_ERROR(x) \19 if (std::error_code ASSERT_NO_ERROR_ec = x) { \20 SmallString<128> MessageStorage; \21 raw_svector_ostream Message(MessageStorage); \22 Message << #x ": did not return errc::success.\n" \23 << "error number: " << ASSERT_NO_ERROR_ec.value() << "\n" \24 << "error message: " << ASSERT_NO_ERROR_ec.message() << "\n"; \25 GTEST_FATAL_FAILURE_(MessageStorage.c_str()); \26 } else { \27 }28 29char data[] = "some data";30 31TEST(Caching, Normal) {32 SmallString<256> CacheDir;33 sys::fs::createUniquePath("llvm_test_cache-%%%%%%", CacheDir, true);34 35 sys::fs::remove_directories(CacheDir.str());36 37 std::unique_ptr<MemoryBuffer> CachedBuffer;38 auto AddBuffer = [&CachedBuffer](unsigned Task, const Twine &ModuleName,39 std::unique_ptr<MemoryBuffer> M) {40 CachedBuffer = std::move(M);41 };42 auto CacheOrErr =43 localCache("LLVMTestCache", "LLVMTest", CacheDir, AddBuffer);44 ASSERT_TRUE(bool(CacheOrErr));45 46 FileCache &Cache = *CacheOrErr;47 48 {49 auto AddStreamOrErr = Cache(1, "foo", "");50 ASSERT_TRUE(bool(AddStreamOrErr));51 52 AddStreamFn &AddStream = *AddStreamOrErr;53 ASSERT_TRUE(AddStream);54 55 auto FileOrErr = AddStream(1, "");56 ASSERT_TRUE(bool(FileOrErr));57 58 CachedFileStream *CFS = FileOrErr->get();59 (*CFS->OS).write(data, sizeof(data));60 ASSERT_THAT_ERROR(CFS->commit(), Succeeded());61 }62 63 {64 auto AddStreamOrErr = Cache(1, "foo", "");65 ASSERT_TRUE(bool(AddStreamOrErr));66 67 AddStreamFn &AddStream = *AddStreamOrErr;68 ASSERT_FALSE(AddStream);69 70 ASSERT_TRUE(CachedBuffer->getBufferSize() == sizeof(data));71 StringRef readData = CachedBuffer->getBuffer();72 ASSERT_TRUE(memcmp(data, readData.data(), sizeof(data)) == 0);73 }74 75 ASSERT_NO_ERROR(sys::fs::remove_directories(CacheDir.str()));76}77 78TEST(Caching, WriteAfterCommit) {79 SmallString<256> CacheDir;80 sys::fs::createUniquePath("llvm_test_cache-%%%%%%", CacheDir, true);81 82 sys::fs::remove_directories(CacheDir.str());83 84 std::unique_ptr<MemoryBuffer> CachedBuffer;85 auto AddBuffer = [&CachedBuffer](unsigned Task, const Twine &ModuleName,86 std::unique_ptr<MemoryBuffer> M) {87 CachedBuffer = std::move(M);88 };89 auto CacheOrErr =90 localCache("LLVMTestCache", "LLVMTest", CacheDir, AddBuffer);91 ASSERT_TRUE(bool(CacheOrErr));92 93 FileCache &Cache = *CacheOrErr;94 95 auto AddStreamOrErr = Cache(1, "foo", "");96 ASSERT_TRUE(bool(AddStreamOrErr));97 98 AddStreamFn &AddStream = *AddStreamOrErr;99 ASSERT_TRUE(AddStream);100 101 auto FileOrErr = AddStream(1, "");102 ASSERT_TRUE(bool(FileOrErr));103 104 CachedFileStream *CFS = FileOrErr->get();105 (*CFS->OS).write(data, sizeof(data));106 ASSERT_THAT_ERROR(CFS->commit(), Succeeded());107 108 EXPECT_DEATH(109 { (*CFS->OS).write(data, sizeof(data)); }, "")110 << "Write after commit did not cause abort";111 112 ASSERT_NO_ERROR(sys::fs::remove_directories(CacheDir.str()));113}114 115TEST(Caching, NoCommit) {116 SmallString<256> CacheDir;117 sys::fs::createUniquePath("llvm_test_cache-%%%%%%", CacheDir, true);118 119 sys::fs::remove_directories(CacheDir.str());120 121 std::unique_ptr<MemoryBuffer> CachedBuffer;122 auto AddBuffer = [&CachedBuffer](unsigned Task, const Twine &ModuleName,123 std::unique_ptr<MemoryBuffer> M) {124 CachedBuffer = std::move(M);125 };126 auto CacheOrErr =127 localCache("LLVMTestCache", "LLVMTest", CacheDir, AddBuffer);128 ASSERT_TRUE(bool(CacheOrErr));129 130 FileCache &Cache = *CacheOrErr;131 132 auto AddStreamOrErr = Cache(1, "foo", "");133 ASSERT_TRUE(bool(AddStreamOrErr));134 135 AddStreamFn &AddStream = *AddStreamOrErr;136 ASSERT_TRUE(AddStream);137 138 auto FileOrErr = AddStream(1, "");139 ASSERT_TRUE(bool(FileOrErr));140 141 CachedFileStream *CFS = FileOrErr->get();142 (*CFS->OS).write(data, sizeof(data));143 ASSERT_THAT_ERROR(CFS->commit(), Succeeded());144 145 EXPECT_DEATH(146 {147 auto FileOrErr = AddStream(1, "");148 ASSERT_TRUE(bool(FileOrErr));149 150 CachedFileStream *CFS = FileOrErr->get();151 (*CFS->OS).write(data, sizeof(data));152 },153 "")154 << "destruction without commit did not cause error";155 156 ASSERT_NO_ERROR(sys::fs::remove_directories(CacheDir.str()));157}158