traxys/tree-sitter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

rust: add handling of arbitrary predicate operators

Browse source
This commit is contained in:
Max Brunsfeld 2020-03-13 13:02:34 -07:00
parent 6e2df06dc2
commit 0457736766
2 changed files with 65 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

20
cli/src/tests/query_test.rs
View file

@ -2,7 +2,8 @@ use super::helpers::allocations;
use super::helpers::fixtures::get_language;
use std::fmt::Write;
use tree_sitter::{
Node, Parser, Query, QueryCapture, QueryCursor, QueryError, QueryMatch, QueryProperty,
Node, Parser, Query, QueryCapture, QueryCursor, QueryError, QueryMatch, QueryPredicate,
QueryPredicateArg, QueryProperty,
};
#[test]
@ -948,7 +949,7 @@ fn test_query_captures_with_text_conditions() {
}
#[test]
fn test_query_captures_with_set_properties() {
fn test_query_captures_with_predicates() {
allocations::record(|| {
let language = get_language("javascript");
@ -957,7 +958,8 @@ fn test_query_captures_with_set_properties() {
r#"
((call_expression (identifier) @foo)
(set! name something)
(set! cool))
(set! cool)
(something! @foo omg))
((property_identifier) @bar
(is? cool)
@ -972,6 +974,16 @@ fn test_query_captures_with_set_properties() {
QueryProperty::new("cool", None, None),
]
);
assert_eq!(
query.general_predicates(0),
&[QueryPredicate {
operator: "something!".to_string().into_boxed_str(),
args: vec![
QueryPredicateArg::Capture(0),
QueryPredicateArg::String("omg".to_string().into_boxed_str()),
],
},]
);
assert_eq!(query.property_settings(1), &[]);
assert_eq!(query.property_predicates(0), &[]);
assert_eq!(
@ -985,7 +997,7 @@ fn test_query_captures_with_set_properties() {
}
#[test]
fn test_query_captures_with_set_quoted_properties() {
fn test_query_captures_with_quoted_predicate_args() {
allocations::record(|| {
let language = get_language("javascript");