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