unittests/LineEditor/LineEditor.cpp

   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
  14 using namespace llvm;
  15
  16 class LineEditorTest : public testing::Test {
  17 public:
  18   SmallString<64> HistPath;
  19   LineEditor *LE;
  20
  21   LineEditorTest() {
  22     init();
  23   }
  24
  25   void init() {
  26     sys::fs::createTemporaryFile("temp", "history", HistPath);
  27     ASSERT_FALSE(HistPath.empty());
  28     LE = new LineEditor("test", HistPath);
  29   }
  30
  31   ~LineEditorTest() override {
  32     delete LE;
  33     sys::fs::remove(HistPath.str());
  34   }
  35 };
  36
  37 TEST_F(LineEditorTest, HistorySaveLoad) {
  38   LE->saveHistory();
  39   LE->loadHistory();
  40 }
  41
  42 struct 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
  56 TEST_F(LineEditorTest, ListCompleters) {
  57   std::vector<LineEditor::Completion> Comps;
  58
  59   Comps.push_back(LineEditor::Completion("foo", "int foo()"));
  60   LE->setListCompleter(TestListCompleter(Comps));
  61   LineEditor::CompletionAction CA = LE->getCompletionAction("", 0);
  62   EXPECT_EQ(LineEditor::CompletionAction::AK_Insert, CA.Kind);
  63   EXPECT_EQ("foo", CA.Text);
  64
  65   Comps.push_back(LineEditor::Completion("bar", "int bar()"));
  66   LE->setListCompleter(TestListCompleter(Comps));
  67   CA = LE->getCompletionAction("", 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(LineEditor::Completion("fee", "int fee()"));
  75   Comps.push_back(LineEditor::Completion("fi", "int fi()"));
  76   Comps.push_back(LineEditor::Completion("foe", "int foe()"));
  77   Comps.push_back(LineEditor::Completion("fum", "int fum()"));
  78   LE->setListCompleter(TestListCompleter(Comps));
  79   CA = LE->getCompletionAction("", 0);
  80   EXPECT_EQ(LineEditor::CompletionAction::AK_Insert, CA.Kind);
  81   EXPECT_EQ("f", CA.Text);
  82 }