aboutsummaryrefslogtreecommitdiff
path: root/lldb/unittests/DAP/FifoFilesTest.cpp
blob: bbc1b608e91bd6ab33d050868e37e311b8417d9c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
//===-- FifoFilesTest.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 "FifoFiles.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Testing/Support/Error.h"
#include "gtest/gtest.h"
#include <chrono>
#include <thread>

using namespace lldb_dap;
using namespace llvm;

namespace {

std::string MakeTempFifoPath() {
  llvm::SmallString<128> temp_path;
  llvm::sys::fs::createUniquePath("lldb-dap-fifo-%%%%%%", temp_path,
                                  /*MakeAbsolute=*/true);
  return temp_path.str().str();
}

} // namespace

TEST(FifoFilesTest, CreateAndDestroyFifoFile) {
  std::string fifo_path = MakeTempFifoPath();
  auto fifo = CreateFifoFile(fifo_path);
  EXPECT_THAT_EXPECTED(fifo, llvm::Succeeded());

  // File should exist.
  EXPECT_TRUE(llvm::sys::fs::exists(fifo_path));

  // Destructor should remove the file.
  fifo->reset();
  EXPECT_FALSE(llvm::sys::fs::exists(fifo_path));
}

TEST(FifoFilesTest, SendAndReceiveJSON) {
  std::string fifo_path = MakeTempFifoPath();
  auto fifo = CreateFifoFile(fifo_path);
  EXPECT_THAT_EXPECTED(fifo, llvm::Succeeded());

  FifoFileIO writer(fifo_path, "writer");
  FifoFileIO reader(fifo_path, "reader");

  llvm::json::Object obj;
  obj["foo"] = "bar";
  obj["num"] = 42;

  // Writer thread.
  std::thread writer_thread([&]() {
    EXPECT_THAT_ERROR(writer.SendJSON(llvm::json::Value(std::move(obj)),
                                      std::chrono::milliseconds(500)),
                      llvm::Succeeded());
  });

  // Reader thread.
  std::thread reader_thread([&]() {
    auto result = reader.ReadJSON(std::chrono::milliseconds(500));
    EXPECT_THAT_EXPECTED(result, llvm::Succeeded());
    auto *read_obj = result->getAsObject();

    ASSERT_NE(read_obj, nullptr);
    EXPECT_EQ((*read_obj)["foo"].getAsString(), "bar");
    EXPECT_EQ((*read_obj)["num"].getAsInteger(), 42);
  });

  writer_thread.join();
  reader_thread.join();
}

TEST(FifoFilesTest, ReadTimeout) {
  std::string fifo_path = MakeTempFifoPath();
  auto fifo = CreateFifoFile(fifo_path);
  EXPECT_THAT_EXPECTED(fifo, llvm::Succeeded());

  FifoFileIO reader(fifo_path, "reader");

  // No writer, should timeout.
  auto result = reader.ReadJSON(std::chrono::milliseconds(100));
  EXPECT_THAT_EXPECTED(result, llvm::Failed());
}

TEST(FifoFilesTest, WriteTimeout) {
  std::string fifo_path = MakeTempFifoPath();
  auto fifo = CreateFifoFile(fifo_path);
  EXPECT_THAT_EXPECTED(fifo, llvm::Succeeded());

  FifoFileIO writer(fifo_path, "writer");

  // No reader, should timeout.
  llvm::json::Object obj;
  obj["foo"] = "bar";
  EXPECT_THAT_ERROR(writer.SendJSON(llvm::json::Value(std::move(obj)),
                                    std::chrono::milliseconds(100)),
                    llvm::Failed());
}