77 lines
2.2 KiB
Rust
77 lines
2.2 KiB
Rust
use super::super::error;
|
|
use super::super::error::Result;
|
|
use lazy_static::lazy_static;
|
|
use regex::Regex;
|
|
use std::collections::hash_map::HashMap;
|
|
use std::fs;
|
|
use tree_sitter::Point;
|
|
|
|
// TODO: It would be cooler to do this with a comments query rather than with a regex
|
|
// directly.
|
|
lazy_static! {
|
|
static ref METADATA_REGEX: Regex = Regex::new(r#"(\w+): ([^\s,]+), (\d+), (\d+)"#).unwrap();
|
|
}
|
|
|
|
#[derive(Debug, Eq, PartialEq)]
|
|
pub struct CaptureInfo {
|
|
pub name: String,
|
|
pub position: Point,
|
|
}
|
|
|
|
#[derive(Debug, Eq, PartialEq)]
|
|
struct Assertion {
|
|
position: Point,
|
|
capture_class: String,
|
|
capture_type: String,
|
|
}
|
|
|
|
impl From<regex::Captures<'_>> for Assertion {
|
|
fn from(re: regex::Captures) -> Assertion {
|
|
Assertion {
|
|
capture_class: re.get(1).unwrap().as_str().to_string(),
|
|
capture_type: re.get(2).unwrap().as_str().to_string(),
|
|
position: Point {
|
|
row: re
|
|
.get(3)
|
|
.iter()
|
|
.flat_map(|m| m.as_str().parse::<usize>())
|
|
.next()
|
|
.unwrap(),
|
|
column: re
|
|
.get(4)
|
|
.iter()
|
|
.flat_map(|m| m.as_str().parse::<usize>())
|
|
.next()
|
|
.unwrap(),
|
|
},
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn assert_expected_captures(captures: Vec<CaptureInfo>, path: String) -> Result<()> {
|
|
let contents = fs::read_to_string(path)?;
|
|
|
|
let assertions: Vec<Assertion> = METADATA_REGEX
|
|
.captures_iter(&contents)
|
|
.map(|c| Assertion::from(c))
|
|
.collect();
|
|
|
|
let per_position_index: HashMap<Point, &Assertion> =
|
|
assertions.iter().map(|a| (a.position, a)).collect();
|
|
|
|
for capture in &captures {
|
|
let oFound = per_position_index.get(&capture.position);
|
|
if oFound.is_none() {
|
|
continue;
|
|
}
|
|
let found = oFound.unwrap();
|
|
let joined = format!("{}.{}", found.capture_class, found.capture_type);
|
|
if joined != capture.name && capture.name != "name" {
|
|
Err(error::Error::new(format!(
|
|
"Assertion failed: at {}, found {}, expected {}",
|
|
capture.position, capture.name, joined
|
|
)))?
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|