1//===-- LineEditor.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-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "llvm/LineEditor/LineEditor.h"
10#include "llvm/Support/FileSystem.h"
11#include "llvm/Support/Path.h"
12#include "gtest/gtest.h"
13
14using namespace llvm;
15
16class LineEditorTest : public testing::Test {
17public:
18 SmallString<64> HistPath;
19 LineEditor *LE;
20
21 LineEditorTest() {
22 init();
23 }
24
25 void init() {
26 sys::fs::createTemporaryFile(Prefix: "temp", Suffix: "history", ResultPath&: HistPath);
27 ASSERT_FALSE(HistPath.empty());
28 LE = new LineEditor("test", HistPath);
29 }
30
31 ~LineEditorTest() override {
32 delete LE;
33 sys::fs::remove(path: HistPath.str());
34 }
35};
36
37TEST_F(LineEditorTest, HistorySaveLoad) {
38 LE->saveHistory();
39 LE->loadHistory();
40}
41
42struct TestListCompleter {
43 std::vector<LineEditor::Completion> Completions;
44
45 TestListCompleter(const std::vector<LineEditor::Completion> &Completions)
46 : Completions(Completions) {}
47
48 std::vector<LineEditor::Completion> operator()(StringRef Buffer,
49 size_t Pos) const {
50 EXPECT_TRUE(Buffer.empty());
51 EXPECT_EQ(0u, Pos);
52 return Completions;
53 }
54};
55
56TEST_F(LineEditorTest, ListCompleters) {
57 std::vector<LineEditor::Completion> Comps;
58
59 Comps.push_back(x: LineEditor::Completion("foo", "int foo()"));
60 LE->setListCompleter(TestListCompleter(Comps));
61 LineEditor::CompletionAction CA = LE->getCompletionAction(Buffer: "", Pos: 0);
62 EXPECT_EQ(LineEditor::CompletionAction::AK_Insert, CA.Kind);
63 EXPECT_EQ("foo", CA.Text);
64
65 Comps.push_back(x: LineEditor::Completion("bar", "int bar()"));
66 LE->setListCompleter(TestListCompleter(Comps));
67 CA = LE->getCompletionAction(Buffer: "", Pos: 0);
68 EXPECT_EQ(LineEditor::CompletionAction::AK_ShowCompletions, CA.Kind);
69 ASSERT_EQ(2u, CA.Completions.size());
70 ASSERT_EQ("int foo()", CA.Completions[0]);
71 ASSERT_EQ("int bar()", CA.Completions[1]);
72
73 Comps.clear();
74 Comps.push_back(x: LineEditor::Completion("fee", "int fee()"));
75 Comps.push_back(x: LineEditor::Completion("fi", "int fi()"));
76 Comps.push_back(x: LineEditor::Completion("foe", "int foe()"));
77 Comps.push_back(x: LineEditor::Completion("fum", "int fum()"));
78 LE->setListCompleter(TestListCompleter(Comps));
79 CA = LE->getCompletionAction(Buffer: "", Pos: 0);
80 EXPECT_EQ(LineEditor::CompletionAction::AK_Insert, CA.Kind);
81 EXPECT_EQ("f", CA.Text);
82}
83

source code of llvm/unittests/LineEditor/LineEditor.cpp