https://github.com/ashgti updated https://github.com/llvm/llvm-project/pull/139937
>From 6f947e38ad4f744754cf13c1094c4e5e3fd249b6 Mon Sep 17 00:00:00 2001 From: John Harrison <harj...@google.com> Date: Wed, 14 May 2025 10:31:40 -0700 Subject: [PATCH 1/4] [lldb-dap] Setup DAP for unit testing. This is a very simple case that currently only validates we can create a DAP instance and send a message over the transport layer. More in-depth tests will require additional helpers and possibly refactors of DAP to make it more testable, however this is some ground work to have basic support for unit tests. --- lldb/tools/lldb-dap/DAP.h | 3 +- lldb/unittests/DAP/CMakeLists.txt | 1 + lldb/unittests/DAP/DAPTest.cpp | 63 +++++++++++++++++++++++++++++++ 3 files changed, 66 insertions(+), 1 deletion(-) create mode 100644 lldb/unittests/DAP/DAPTest.cpp diff --git a/lldb/tools/lldb-dap/DAP.h b/lldb/tools/lldb-dap/DAP.h index c2e4c2dea582e..2ff66d1cd0182 100644 --- a/lldb/tools/lldb-dap/DAP.h +++ b/lldb/tools/lldb-dap/DAP.h @@ -226,7 +226,8 @@ struct DAP { /// \param[in] default_repl_mode /// Default repl mode behavior, as configured by the binary. /// \param[in] pre_init_commands - /// LLDB commands to execute as soon as the debugger instance is allocaed. + /// LLDB commands to execute as soon as the debugger instance is + /// allocated. /// \param[in] transport /// Transport for this debug session. DAP(Log *log, const ReplMode default_repl_mode, diff --git a/lldb/unittests/DAP/CMakeLists.txt b/lldb/unittests/DAP/CMakeLists.txt index 110733e93b192..6074e9b872c49 100644 --- a/lldb/unittests/DAP/CMakeLists.txt +++ b/lldb/unittests/DAP/CMakeLists.txt @@ -3,6 +3,7 @@ add_lldb_unittest(DAPTests LLDBUtilsTest.cpp TransportTest.cpp ProtocolTypesTest.cpp + DAPTest.cpp LINK_LIBS lldbDAP diff --git a/lldb/unittests/DAP/DAPTest.cpp b/lldb/unittests/DAP/DAPTest.cpp new file mode 100644 index 0000000000000..9d2a9b944678e --- /dev/null +++ b/lldb/unittests/DAP/DAPTest.cpp @@ -0,0 +1,63 @@ +//===-- DAPTest.cpp -------------------------------------------------===// +// +// 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 "DAP.h" +#include "Protocol/ProtocolBase.h" +#include "Transport.h" +#include "lldb/Host/File.h" +#include "lldb/Host/Pipe.h" +#include "llvm/Testing/Support/Error.h" +#include "gtest/gtest.h" +#include <chrono> +#include <memory> +#include <optional> + +using namespace llvm; +using namespace lldb; +using namespace lldb_dap; +using namespace lldb_dap::protocol; +using lldb_private::File; +using lldb_private::NativeFile; +using lldb_private::Pipe; + +class DAPTest : public testing::Test { +protected: + Pipe input; + Pipe output; + std::unique_ptr<Transport> toDAP; + std::unique_ptr<Transport> fromDAP; + + void SetUp() override { + ASSERT_THAT_ERROR(input.CreateNew(false).ToError(), Succeeded()); + ASSERT_THAT_ERROR(output.CreateNew(false).ToError(), Succeeded()); + toDAP = std::make_unique<Transport>( + "toDAP", nullptr, + std::make_shared<NativeFile>(input.GetReadFileDescriptor(), + File::eOpenOptionReadOnly, + NativeFile::Unowned), + std::make_shared<NativeFile>(output.GetWriteFileDescriptor(), + File::eOpenOptionWriteOnly, + NativeFile::Unowned)); + fromDAP = std::make_unique<Transport>( + "fromDAP", nullptr, + std::make_shared<NativeFile>(output.GetReadFileDescriptor(), + File::eOpenOptionReadOnly, + NativeFile::Unowned), + std::make_shared<NativeFile>(input.GetWriteFileDescriptor(), + File::eOpenOptionWriteOnly, + NativeFile::Unowned)); + } +}; + +TEST_F(DAPTest, SendProtocolMessages) { + DAP dap{nullptr, ReplMode::Auto, {}, *toDAP}; + dap.Send(Event{"my-event", std::nullopt}); + ASSERT_THAT_EXPECTED(fromDAP->Read(std::chrono::milliseconds(1)), + HasValue(testing::VariantWith<Event>(testing::FieldsAre( + /*event=*/"my-event", /*body=*/std::nullopt)))); +} >From 8bc4880a338dd0d0f2daa64b77c8bdbf567270b2 Mon Sep 17 00:00:00 2001 From: John Harrison <harj...@google.com> Date: Wed, 14 May 2025 10:46:51 -0700 Subject: [PATCH 2/4] Adjusting the naming of variables. --- lldb/unittests/DAP/DAPTest.cpp | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/lldb/unittests/DAP/DAPTest.cpp b/lldb/unittests/DAP/DAPTest.cpp index 9d2a9b944678e..8e8c2ea3ccbb0 100644 --- a/lldb/unittests/DAP/DAPTest.cpp +++ b/lldb/unittests/DAP/DAPTest.cpp @@ -29,22 +29,22 @@ class DAPTest : public testing::Test { protected: Pipe input; Pipe output; - std::unique_ptr<Transport> toDAP; - std::unique_ptr<Transport> fromDAP; + std::unique_ptr<Transport> to_dap; + std::unique_ptr<Transport> from_dap; void SetUp() override { ASSERT_THAT_ERROR(input.CreateNew(false).ToError(), Succeeded()); ASSERT_THAT_ERROR(output.CreateNew(false).ToError(), Succeeded()); - toDAP = std::make_unique<Transport>( - "toDAP", nullptr, + to_dap = std::make_unique<Transport>( + "to_dap", nullptr, std::make_shared<NativeFile>(input.GetReadFileDescriptor(), File::eOpenOptionReadOnly, NativeFile::Unowned), std::make_shared<NativeFile>(output.GetWriteFileDescriptor(), File::eOpenOptionWriteOnly, NativeFile::Unowned)); - fromDAP = std::make_unique<Transport>( - "fromDAP", nullptr, + from_dap = std::make_unique<Transport>( + "from_dap", nullptr, std::make_shared<NativeFile>(output.GetReadFileDescriptor(), File::eOpenOptionReadOnly, NativeFile::Unowned), @@ -55,9 +55,9 @@ class DAPTest : public testing::Test { }; TEST_F(DAPTest, SendProtocolMessages) { - DAP dap{nullptr, ReplMode::Auto, {}, *toDAP}; + DAP dap{nullptr, ReplMode::Auto, {}, *to_dap}; dap.Send(Event{"my-event", std::nullopt}); - ASSERT_THAT_EXPECTED(fromDAP->Read(std::chrono::milliseconds(1)), + ASSERT_THAT_EXPECTED(from_dap->Read(std::chrono::milliseconds(1)), HasValue(testing::VariantWith<Event>(testing::FieldsAre( /*event=*/"my-event", /*body=*/std::nullopt)))); } >From 41f8f41a6cae98c6e0fe703f5b34f9f3f4acba9a Mon Sep 17 00:00:00 2001 From: John Harrison <harj...@google.com> Date: Wed, 14 May 2025 10:51:12 -0700 Subject: [PATCH 3/4] Adding inline comments to improve readability. --- lldb/unittests/DAP/DAPTest.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lldb/unittests/DAP/DAPTest.cpp b/lldb/unittests/DAP/DAPTest.cpp index 8e8c2ea3ccbb0..e23fb32eebea9 100644 --- a/lldb/unittests/DAP/DAPTest.cpp +++ b/lldb/unittests/DAP/DAPTest.cpp @@ -55,8 +55,9 @@ class DAPTest : public testing::Test { }; TEST_F(DAPTest, SendProtocolMessages) { - DAP dap{nullptr, ReplMode::Auto, {}, *to_dap}; - dap.Send(Event{"my-event", std::nullopt}); + DAP dap{/*log=*/nullptr, /*default_repl_mode=*/ReplMode::Auto, + /*pre_init_commands=*/{}, /*transport=*/*to_dap}; + dap.Send(Event{/*event=*/"my-event", /*body=*/std::nullopt}); ASSERT_THAT_EXPECTED(from_dap->Read(std::chrono::milliseconds(1)), HasValue(testing::VariantWith<Event>(testing::FieldsAre( /*event=*/"my-event", /*body=*/std::nullopt)))); >From e6dfa922161b70228a38f2bcd52776efb64ae6c1 Mon Sep 17 00:00:00 2001 From: John Harrison <harj...@google.com> Date: Wed, 14 May 2025 13:07:10 -0700 Subject: [PATCH 4/4] Add some helper test bases classes and a Handler test as well. --- lldb/unittests/DAP/CMakeLists.txt | 6 +- lldb/unittests/DAP/DAPTest.cpp | 46 +++--------- .../Handler/DisconnectRequestHandlerTest.cpp | 34 +++++++++ lldb/unittests/DAP/TestBase.cpp | 70 +++++++++++++++++++ lldb/unittests/DAP/TestBase.h | 46 ++++++++++++ lldb/unittests/DAP/TransportTest.cpp | 24 +++---- 6 files changed, 174 insertions(+), 52 deletions(-) create mode 100644 lldb/unittests/DAP/Handler/DisconnectRequestHandlerTest.cpp create mode 100644 lldb/unittests/DAP/TestBase.cpp create mode 100644 lldb/unittests/DAP/TestBase.h diff --git a/lldb/unittests/DAP/CMakeLists.txt b/lldb/unittests/DAP/CMakeLists.txt index 6074e9b872c49..4b3447ae3310a 100644 --- a/lldb/unittests/DAP/CMakeLists.txt +++ b/lldb/unittests/DAP/CMakeLists.txt @@ -1,9 +1,11 @@ add_lldb_unittest(DAPTests + DAPTest.cpp + Handler/DisconnectRequestHandlerTest.cpp JSONUtilsTest.cpp LLDBUtilsTest.cpp - TransportTest.cpp ProtocolTypesTest.cpp - DAPTest.cpp + TestBase.cpp + TransportTest.cpp LINK_LIBS lldbDAP diff --git a/lldb/unittests/DAP/DAPTest.cpp b/lldb/unittests/DAP/DAPTest.cpp index e23fb32eebea9..5fb6bf7e564ab 100644 --- a/lldb/unittests/DAP/DAPTest.cpp +++ b/lldb/unittests/DAP/DAPTest.cpp @@ -1,4 +1,4 @@ -//===-- DAPTest.cpp -------------------------------------------------===// +//===-- DAPTest.cpp -------------------------------------------------------===// // // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // See https://llvm.org/LICENSE.txt for license information. @@ -8,9 +8,8 @@ #include "DAP.h" #include "Protocol/ProtocolBase.h" +#include "TestBase.h" #include "Transport.h" -#include "lldb/Host/File.h" -#include "lldb/Host/Pipe.h" #include "llvm/Testing/Support/Error.h" #include "gtest/gtest.h" #include <chrono> @@ -20,43 +19,18 @@ using namespace llvm; using namespace lldb; using namespace lldb_dap; +using namespace lldb_dap_tests; using namespace lldb_dap::protocol; -using lldb_private::File; -using lldb_private::NativeFile; -using lldb_private::Pipe; -class DAPTest : public testing::Test { -protected: - Pipe input; - Pipe output; - std::unique_ptr<Transport> to_dap; - std::unique_ptr<Transport> from_dap; - - void SetUp() override { - ASSERT_THAT_ERROR(input.CreateNew(false).ToError(), Succeeded()); - ASSERT_THAT_ERROR(output.CreateNew(false).ToError(), Succeeded()); - to_dap = std::make_unique<Transport>( - "to_dap", nullptr, - std::make_shared<NativeFile>(input.GetReadFileDescriptor(), - File::eOpenOptionReadOnly, - NativeFile::Unowned), - std::make_shared<NativeFile>(output.GetWriteFileDescriptor(), - File::eOpenOptionWriteOnly, - NativeFile::Unowned)); - from_dap = std::make_unique<Transport>( - "from_dap", nullptr, - std::make_shared<NativeFile>(output.GetReadFileDescriptor(), - File::eOpenOptionReadOnly, - NativeFile::Unowned), - std::make_shared<NativeFile>(input.GetWriteFileDescriptor(), - File::eOpenOptionWriteOnly, - NativeFile::Unowned)); - } -}; +class DAPTest : public TransportBase {}; TEST_F(DAPTest, SendProtocolMessages) { - DAP dap{/*log=*/nullptr, /*default_repl_mode=*/ReplMode::Auto, - /*pre_init_commands=*/{}, /*transport=*/*to_dap}; + DAP dap{ + /*log=*/nullptr, + /*default_repl_mode=*/ReplMode::Auto, + /*pre_init_commands=*/{}, + /*transport=*/*to_dap, + }; dap.Send(Event{/*event=*/"my-event", /*body=*/std::nullopt}); ASSERT_THAT_EXPECTED(from_dap->Read(std::chrono::milliseconds(1)), HasValue(testing::VariantWith<Event>(testing::FieldsAre( diff --git a/lldb/unittests/DAP/Handler/DisconnectRequestHandlerTest.cpp b/lldb/unittests/DAP/Handler/DisconnectRequestHandlerTest.cpp new file mode 100644 index 0000000000000..0ef763574e8df --- /dev/null +++ b/lldb/unittests/DAP/Handler/DisconnectRequestHandlerTest.cpp @@ -0,0 +1,34 @@ +//===-- DisconnectRequestHandlerTest.cpp ----------------------------------===// +// +// 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 "DAP.h" +#include "Handler/RequestHandler.h" +#include "Protocol/ProtocolBase.h" +#include "TestBase.h" +#include "llvm/Testing/Support/Error.h" +#include "gtest/gtest.h" +#include <memory> +#include <optional> + +using namespace llvm; +using namespace lldb; +using namespace lldb_dap; +using namespace lldb_dap_tests; +using namespace lldb_dap::protocol; + +class DisconnectRequestHandlerTest : public DAPTestBase {}; + +TEST_F(DisconnectRequestHandlerTest, DisconnectingTriggersTerminated) { + DisconnectRequestHandler handler(*dap); + ASSERT_THAT_ERROR(handler.Run(std::nullopt), Succeeded()); + EXPECT_TRUE(dap->disconnecting); + std::vector<Message> messages = DrainOutput(); + EXPECT_THAT(messages, + testing::Contains(testing::VariantWith<Event>(testing::FieldsAre( + /*event=*/"terminated", /*body=*/std::nullopt)))); +} diff --git a/lldb/unittests/DAP/TestBase.cpp b/lldb/unittests/DAP/TestBase.cpp new file mode 100644 index 0000000000000..08dda62e217c3 --- /dev/null +++ b/lldb/unittests/DAP/TestBase.cpp @@ -0,0 +1,70 @@ +//===-- TestBase.cpp -------------------------------------------------===// +// +// 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 "TestBase.h" +#include "Protocol/ProtocolBase.h" +#include "lldb/Host/File.h" +#include "lldb/Host/Pipe.h" +#include "llvm/Testing/Support/Error.h" + +using namespace llvm; +using namespace lldb; +using namespace lldb_dap; +using namespace lldb_dap::protocol; +using namespace lldb_dap_tests; +using lldb_private::File; +using lldb_private::NativeFile; +using lldb_private::Pipe; + +void PipeBase::SetUp() { + ASSERT_THAT_ERROR(input.CreateNew(false).ToError(), Succeeded()); + ASSERT_THAT_ERROR(output.CreateNew(false).ToError(), Succeeded()); +} + +void TransportBase::SetUp() { + PipeBase::SetUp(); + to_dap = std::make_unique<Transport>( + "to_dap", nullptr, + std::make_shared<NativeFile>(input.GetReadFileDescriptor(), + File::eOpenOptionReadOnly, + NativeFile::Unowned), + std::make_shared<NativeFile>(output.GetWriteFileDescriptor(), + File::eOpenOptionWriteOnly, + NativeFile::Unowned)); + from_dap = std::make_unique<Transport>( + "from_dap", nullptr, + std::make_shared<NativeFile>(output.GetReadFileDescriptor(), + File::eOpenOptionReadOnly, + NativeFile::Unowned), + std::make_shared<NativeFile>(input.GetWriteFileDescriptor(), + File::eOpenOptionWriteOnly, + NativeFile::Unowned)); +} + +void DAPTestBase::SetUp() { + TransportBase::SetUp(); + dap = std::make_unique<DAP>( + /*log=*/nullptr, + /*default_repl_mode=*/ReplMode::Auto, + /*pre_init_commands=*/std::vector<std::string>(), + /*transport=*/*to_dap); +} + +std::vector<Message> DAPTestBase::DrainOutput() { + std::vector<Message> msgs; + output.CloseWriteFileDescriptor(); + while (true) { + Expected<Message> next = from_dap->Read(std::chrono::milliseconds(1)); + if (!next) { + consumeError(next.takeError()); + break; + } + msgs.push_back(*next); + } + return msgs; +} \ No newline at end of file diff --git a/lldb/unittests/DAP/TestBase.h b/lldb/unittests/DAP/TestBase.h new file mode 100644 index 0000000000000..7f8b2fb5121c6 --- /dev/null +++ b/lldb/unittests/DAP/TestBase.h @@ -0,0 +1,46 @@ +//===-- TestBase.cpp ------------------------------------------------------===// +// +// 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 "DAP.h" +#include "Protocol/ProtocolBase.h" +#include "Transport.h" +#include "lldb/Host/Pipe.h" +#include "gtest/gtest.h" + +namespace lldb_dap_tests { + +/// A base class for tests that need a pair of pipes for communication. +class PipeBase : public testing::Test { +protected: + lldb_private::Pipe input; + lldb_private::Pipe output; + + void SetUp() override; +}; + +/// A base class for tests that need transport configured. +class TransportBase : public PipeBase { +protected: + std::unique_ptr<lldb_dap::Transport> to_dap; + std::unique_ptr<lldb_dap::Transport> from_dap; + + void SetUp() override; +}; + +class DAPTestBase : public TransportBase { +protected: + std::unique_ptr<lldb_dap::DAP> dap; + + void SetUp() override; + + /// Closes the DAP output pipe and returns the remaining protocol messages in + /// the buffer. + std::vector<lldb_dap::protocol::Message> DrainOutput(); +}; + +} // namespace lldb_dap_tests diff --git a/lldb/unittests/DAP/TransportTest.cpp b/lldb/unittests/DAP/TransportTest.cpp index 5c77b4bb26343..e6dab42e30941 100644 --- a/lldb/unittests/DAP/TransportTest.cpp +++ b/lldb/unittests/DAP/TransportTest.cpp @@ -8,10 +8,10 @@ #include "Transport.h" #include "Protocol/ProtocolBase.h" +#include "TestBase.h" #include "lldb/Host/File.h" #include "lldb/Host/Pipe.h" #include "llvm/ADT/StringRef.h" -#include "llvm/Support/FormatVariadic.h" #include "llvm/Testing/Support/Error.h" #include "gtest/gtest.h" #include <chrono> @@ -21,20 +21,18 @@ using namespace llvm; using namespace lldb; using namespace lldb_dap; +using namespace lldb_dap_tests; using namespace lldb_dap::protocol; using lldb_private::File; using lldb_private::NativeFile; using lldb_private::Pipe; -class TransportTest : public testing::Test { +class TransportTest : public PipeBase { protected: - Pipe input; - Pipe output; std::unique_ptr<Transport> transport; void SetUp() override { - ASSERT_THAT_ERROR(input.CreateNew(false).ToError(), Succeeded()); - ASSERT_THAT_ERROR(output.CreateNew(false).ToError(), Succeeded()); + PipeBase::SetUp(); transport = std::make_unique<Transport>( "stdio", nullptr, std::make_shared<NativeFile>(input.GetReadFileDescriptor(), @@ -44,13 +42,6 @@ class TransportTest : public testing::Test { File::eOpenOptionWriteOnly, NativeFile::Unowned)); } - - void Write(StringRef json) { - std::string message = - formatv("Content-Length: {0}\r\n\r\n{1}", json.size(), json).str(); - ASSERT_THAT_EXPECTED(input.Write(message.data(), message.size()), - Succeeded()); - } }; TEST_F(TransportTest, MalformedRequests) { @@ -65,7 +56,12 @@ TEST_F(TransportTest, MalformedRequests) { } TEST_F(TransportTest, Read) { - Write(R"json({"seq": 1, "type": "request", "command": "abc"})json"); + std::string json = + R"json({"seq": 1, "type": "request", "command": "abc"})json"; + std::string message = + formatv("Content-Length: {0}\r\n\r\n{1}", json.size(), json).str(); + ASSERT_THAT_EXPECTED(input.Write(message.data(), message.size()), + Succeeded()); ASSERT_THAT_EXPECTED( transport->Read(std::chrono::milliseconds(1)), HasValue(testing::VariantWith<Request>(testing::FieldsAre( _______________________________________________ lldb-commits mailing list lldb-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/lldb-commits