internal: add tests for extra parser entry points

This commit is contained in:
Aleksey Kladov 2021-12-29 18:23:34 +03:00
parent 355a4bdb88
commit 3f5fc05d66
4 changed files with 46 additions and 1 deletions

View file

@ -22,6 +22,7 @@ pub struct Output {
error: Vec<String>,
}
#[derive(Debug)]
pub enum Step<'a> {
Token { kind: SyntaxKind, n_input_tokens: u8 },
Enter { kind: SyntaxKind },

View file

@ -16,6 +16,7 @@ use crate::{
SyntaxKind::{self, *},
};
#[derive(Debug)]
pub enum StrStep<'a> {
Token { kind: SyntaxKind, text: &'a str },
Enter { kind: SyntaxKind },
@ -75,7 +76,8 @@ impl<'a> LexedStr<'a> {
builder.eat_trivias();
(builder.sink)(StrStep::Exit);
}
State::PendingEnter | State::Normal => unreachable!(),
State::PendingEnter => (),
State::Normal => unreachable!(),
}
let is_eof = builder.pos == builder.lexed.len();

View file

@ -1,4 +1,5 @@
mod sourcegen_inline_tests;
mod entries;
use std::{
fmt::Write,

View file

@ -0,0 +1,41 @@
use crate::{LexedStr, PrefixEntryPoint, StrStep};
#[test]
fn vis() {
check_prefix(PrefixEntryPoint::Vis, "pub(crate) fn foo() {}", "pub(crate)");
check_prefix(PrefixEntryPoint::Vis, "fn foo() {}", "");
check_prefix(PrefixEntryPoint::Vis, "pub(fn foo() {}", "pub");
check_prefix(PrefixEntryPoint::Vis, "pub(crate fn foo() {}", "pub(crate");
check_prefix(PrefixEntryPoint::Vis, "crate fn foo() {}", "crate");
}
#[test]
fn block() {
check_prefix(PrefixEntryPoint::Block, "{}, 92", "{}");
check_prefix(PrefixEntryPoint::Block, "{, 92)", "{, 92)");
check_prefix(PrefixEntryPoint::Block, "()", "");
}
#[test]
fn stmt() {
check_prefix(PrefixEntryPoint::Stmt, "92; fn", "92");
check_prefix(PrefixEntryPoint::Stmt, "let _ = 92; 1", "let _ = 92");
check_prefix(PrefixEntryPoint::Stmt, "pub fn f() {} = 92", "pub fn f() {}");
check_prefix(PrefixEntryPoint::Stmt, ";;;", ";");
check_prefix(PrefixEntryPoint::Stmt, "+", "+");
check_prefix(PrefixEntryPoint::Stmt, "@", "@");
check_prefix(PrefixEntryPoint::Stmt, "loop {} - 1", "loop {}");
}
fn check_prefix(entry: PrefixEntryPoint, input: &str, prefix: &str) {
let lexed = LexedStr::new(input);
let input = lexed.to_input();
let output = entry.parse(&input);
let mut buf = String::new();
lexed.intersperse_trivia(&output, &mut |step| match step {
StrStep::Token { kind: _, text } => buf.push_str(text),
_ => (),
});
assert_eq!(buf.trim(), prefix)
}