TheRock/compiler/amd-llvm/lldb/unittests/DAP/Handler/DisconnectTest.cpp

//===-- DisconnectTest.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 "lldb/API/SBDefines.h"
#include "lldb/lldb-enumerations.h"
#include "llvm/Testing/Support/Error.h"
#include "gmock/gmock.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;
using testing::_;

class DisconnectRequestHandlerTest : public DAPTestBase {};

TEST_F(DisconnectRequestHandlerTest, DisconnectTriggersTerminated) {
  DisconnectRequestHandler handler(*dap);
  ASSERT_THAT_ERROR(handler.Run(std::nullopt), Succeeded());
  EXPECT_CALL(client, Received(IsEvent("terminated", _)));
  Run();
}

// Is flaky on Linux, see https://github.com/llvm/llvm-project/issues/154763.
#if LLDB_ENABLE_PYTHON && !defined(__linux__)
TEST_F(DisconnectRequestHandlerTest, DisconnectTriggersTerminateCommands) {
  CreateDebugger();

  if (!GetDebuggerSupportsTarget("X86"))
    GTEST_SKIP() << "Unsupported platform";

  LoadCore();

  DisconnectRequestHandler handler(*dap);

  dap->configuration.terminateCommands = {"?script print(1)",
                                          "script print(2)"};
  EXPECT_EQ(dap->target.GetProcess().GetState(), lldb::eStateStopped);
  ASSERT_THAT_ERROR(handler.Run(std::nullopt), Succeeded());
  EXPECT_CALL(client, Received(Output("1\n")));
  EXPECT_CALL(client, Received(Output("2\n"))).Times(2);
  EXPECT_CALL(client, Received(Output("(lldb) script print(2)\n")));
  EXPECT_CALL(client, Received(Output("Running terminateCommands:\n")));
  EXPECT_CALL(client, Received(IsEvent("terminated", _)));
  Run();
}
#endif