mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-26 18:26:05 +00:00
c6b2b78429
Summary: YAML serialization was used in the Proof of Concept for simplicity. This patch replaces implements Protobuf (de) serialization of almost all types that need to be transferred over the protocol. Reviewers: sammccall Reviewed By: sammccall Subscribers: mgorny, ilya-biryukov, MaskRay, jkorous, arphaman, kadircet, usaxena95, cfe-commits Tags: #clang Differential Revision: https://reviews.llvm.org/D79862
94 lines
2.4 KiB
C++
94 lines
2.4 KiB
C++
//===--- MarshallingTests.cpp ------------------------------------*- C++-*-===//
|
|
//
|
|
// 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 "../TestTU.h"
|
|
#include "index/Serialization.h"
|
|
#include "index/remote/marshalling/Marshalling.h"
|
|
#include "llvm/Support/StringSaver.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace clang {
|
|
namespace clangd {
|
|
namespace remote {
|
|
namespace {
|
|
|
|
TEST(RemoteMarshallingTest, SymbolSerialization) {
|
|
const auto *Header = R"(
|
|
// This is a class.
|
|
class Foo {
|
|
public:
|
|
Foo();
|
|
|
|
int Bar;
|
|
private:
|
|
double Number;
|
|
};
|
|
/// This is a function.
|
|
char baz();
|
|
template <typename T>
|
|
T getT ();
|
|
)";
|
|
const auto TU = TestTU::withHeaderCode(Header);
|
|
const auto Symbols = TU.headerSymbols();
|
|
// Sanity check: there are more than 5 symbols available.
|
|
EXPECT_GE(Symbols.size(), 5UL);
|
|
llvm::BumpPtrAllocator Arena;
|
|
llvm::UniqueStringSaver Strings(Arena);
|
|
for (auto &Sym : Symbols) {
|
|
const auto ProtobufMeessage = toProtobuf(Sym);
|
|
const auto SymToProtobufAndBack = fromProtobuf(ProtobufMeessage, &Strings);
|
|
EXPECT_TRUE(SymToProtobufAndBack.hasValue());
|
|
EXPECT_EQ(toYAML(Sym), toYAML(*SymToProtobufAndBack));
|
|
}
|
|
}
|
|
|
|
TEST(RemoteMarshallingTest, ReferenceSerialization) {
|
|
TestTU TU;
|
|
TU.HeaderCode = R"(
|
|
int foo();
|
|
int GlobalVariable = 42;
|
|
class Foo {
|
|
public:
|
|
Foo();
|
|
|
|
char Symbol = 'S';
|
|
};
|
|
template <typename T>
|
|
T getT() { return T(); }
|
|
)";
|
|
TU.Code = R"(
|
|
int foo() {
|
|
++GlobalVariable;
|
|
|
|
Foo foo = Foo();
|
|
if (foo.Symbol - 'a' == 42) {
|
|
foo.Symbol = 'b';
|
|
}
|
|
|
|
const auto bar = getT<Foo>();
|
|
}
|
|
)";
|
|
const auto References = TU.headerRefs();
|
|
llvm::BumpPtrAllocator Arena;
|
|
llvm::UniqueStringSaver Strings(Arena);
|
|
// Sanity check: there are more than 5 references available.
|
|
EXPECT_GE(References.numRefs(), 5UL);
|
|
for (const auto &SymbolWithRefs : References) {
|
|
for (const auto &Ref : SymbolWithRefs.second) {
|
|
const auto RefToProtobufAndBack = fromProtobuf(toProtobuf(Ref), &Strings);
|
|
EXPECT_TRUE(RefToProtobufAndBack.hasValue());
|
|
EXPECT_EQ(toYAML(Ref), toYAML(*RefToProtobufAndBack));
|
|
}
|
|
}
|
|
} // namespace
|
|
|
|
} // namespace
|
|
} // namespace remote
|
|
} // namespace clangd
|
|
} // namespace clang
|