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
|
//===--- LRTableTest.cpp - ---------------------------------------*- C++-*-===//
//
// 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 "clang-pseudo/grammar/LRTable.h"
#include "clang-pseudo/grammar/Grammar.h"
#include "clang/Basic/TokenKinds.h"
#include "llvm/Testing/Support/SupportHelpers.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include <vector>
namespace clang {
namespace pseudo {
namespace {
using llvm::ValueIs;
using testing::ElementsAre;
using StateID = LRTable::StateID;
TEST(LRTable, Builder) {
std::vector<std::string> GrammarDiags;
Grammar G = Grammar::parseBNF(R"bnf(
_ := expr # rule 0
expr := term # rule 1
expr := expr + term # rule 2
term := IDENTIFIER # rule 3
)bnf",
GrammarDiags);
EXPECT_THAT(GrammarDiags, testing::IsEmpty());
SymbolID Term = *G.findNonterminal("term");
SymbolID Eof = tokenSymbol(tok::eof);
SymbolID Identifier = tokenSymbol(tok::identifier);
SymbolID Plus = tokenSymbol(tok::plus);
LRTable::Builder B(G);
// eof IDENT term
// +-------+----+-------+------+
// |state0 | | s0 | |
// |state1 | | | g3 |
// |state2 | | | |
// +-------+----+-------+------+-------
B.Transition[{StateID{0}, Identifier}] = StateID{0};
B.Transition[{StateID{1}, Term}] = StateID{3};
B.Reduce[StateID{0}].insert(RuleID{0});
B.Reduce[StateID{1}].insert(RuleID{2});
B.Reduce[StateID{2}].insert(RuleID{1});
LRTable T = std::move(B).build();
EXPECT_EQ(T.getShiftState(0, Eof), llvm::None);
EXPECT_THAT(T.getShiftState(0, Identifier), ValueIs(0));
EXPECT_THAT(T.getReduceRules(0), ElementsAre(0));
EXPECT_EQ(T.getShiftState(1, Eof), llvm::None);
EXPECT_EQ(T.getShiftState(1, Identifier), llvm::None);
EXPECT_THAT(T.getGoToState(1, Term), ValueIs(3));
EXPECT_THAT(T.getReduceRules(1), ElementsAre(2));
// Verify the behaivor for other non-available-actions terminals.
SymbolID Int = tokenSymbol(tok::kw_int);
EXPECT_EQ(T.getShiftState(2, Int), llvm::None);
// Check follow sets.
EXPECT_TRUE(T.canFollow(Term, Plus));
EXPECT_TRUE(T.canFollow(Term, Eof));
EXPECT_FALSE(T.canFollow(Term, Int));
}
} // namespace
} // namespace pseudo
} // namespace clang
|
