mirror of
https://github.com/danbulant/oxc
synced 2026-05-24 12:21:58 +00:00
chore(query): move tests to their own file (#643)
This commit is contained in:
u9g
GitHub
parent
453edd27eb
commit
5b0449507e
2 changed files with 208 additions and 212 deletions
|
|
@ -9,215 +9,4 @@ mod vertex;
|
|||
pub use adapter::{schema, Adapter};
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
use std::{collections::BTreeMap, rc::Rc, sync::Arc};
|
||||
|
||||
use oxc_allocator::Allocator;
|
||||
use oxc_parser::Parser;
|
||||
use oxc_semantic::SemanticBuilder;
|
||||
use oxc_span::SourceType;
|
||||
use trustfall::{execute_query, FieldValue, TryIntoStruct};
|
||||
|
||||
use crate::{adapter::schema, Adapter};
|
||||
|
||||
#[test]
|
||||
fn test_path_query() {
|
||||
let code = "const apple = 1;";
|
||||
|
||||
let allocator = Allocator::default();
|
||||
let source_type =
|
||||
SourceType::default().with_module(true).with_jsx(true).with_typescript(true);
|
||||
let ret = Parser::new(&allocator, code, source_type).parse();
|
||||
let program = allocator.alloc(ret.program);
|
||||
let semantic_ret =
|
||||
SemanticBuilder::new(code, source_type).with_trivias(&ret.trivias).build(program);
|
||||
|
||||
let adapter = Adapter {
|
||||
path_components: vec![Some("index".to_string())],
|
||||
semantic: Rc::new(semantic_ret.semantic),
|
||||
};
|
||||
|
||||
let args: BTreeMap<Arc<str>, FieldValue> = BTreeMap::new();
|
||||
|
||||
let adapter = Arc::from(&adapter);
|
||||
|
||||
#[allow(clippy::items_after_statements)]
|
||||
#[derive(Debug, PartialOrd, Ord, PartialEq, Eq, serde::Deserialize)]
|
||||
struct Output {
|
||||
name: String,
|
||||
is_first: bool,
|
||||
is_last: bool,
|
||||
}
|
||||
|
||||
let mut results: Vec<Output> = execute_query(
|
||||
schema(),
|
||||
adapter,
|
||||
r#"
|
||||
query {
|
||||
File {
|
||||
last_path_part {
|
||||
name @output
|
||||
is_first @output
|
||||
is_last @output
|
||||
}
|
||||
}
|
||||
}
|
||||
"#,
|
||||
args,
|
||||
)
|
||||
.expect("to successfully execute the query")
|
||||
.map(|row| row.try_into_struct().expect("shape mismatch"))
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
results.sort_unstable();
|
||||
|
||||
assert_eq!(vec![Output { name: "index".into(), is_first: true, is_last: true },], results);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_variable_query() {
|
||||
let code = "const apple = 1;";
|
||||
|
||||
let allocator = Allocator::default();
|
||||
let source_type =
|
||||
SourceType::default().with_module(true).with_jsx(true).with_typescript(true);
|
||||
let ret = Parser::new(&allocator, code, source_type).parse();
|
||||
let program = allocator.alloc(ret.program);
|
||||
let semantic_ret =
|
||||
SemanticBuilder::new(code, source_type).with_trivias(&ret.trivias).build(program);
|
||||
|
||||
let adapter = Adapter {
|
||||
path_components: vec![Some("index".to_string())],
|
||||
semantic: Rc::new(semantic_ret.semantic),
|
||||
};
|
||||
|
||||
let args: BTreeMap<Arc<str>, FieldValue> = BTreeMap::new();
|
||||
|
||||
let adapter = Arc::from(&adapter);
|
||||
|
||||
#[allow(clippy::items_after_statements)]
|
||||
#[derive(Debug, PartialOrd, Ord, PartialEq, Eq, serde::Deserialize)]
|
||||
struct Output {
|
||||
assignment_to_variable_name: String,
|
||||
span_asmt_type_start: u64,
|
||||
span_asmt_type_end: u64,
|
||||
span_start: u64,
|
||||
span_end: u64,
|
||||
__typename: String,
|
||||
}
|
||||
|
||||
let mut results: Vec<Output> = execute_query(
|
||||
schema(),
|
||||
adapter,
|
||||
r#"
|
||||
query {
|
||||
File {
|
||||
variable_declaration {
|
||||
left {
|
||||
assignment_to_variable_name @output
|
||||
span_asmt_type_: span {
|
||||
start @output
|
||||
end @output
|
||||
}
|
||||
}
|
||||
|
||||
__typename @output
|
||||
|
||||
span_: span {
|
||||
start @output
|
||||
end @output
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
"#,
|
||||
args,
|
||||
)
|
||||
.expect("to successfully execute the query")
|
||||
.map(|row| row.try_into_struct().expect("shape mismatch"))
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
results.sort_unstable();
|
||||
|
||||
assert_eq!(
|
||||
vec![Output {
|
||||
assignment_to_variable_name: "apple".to_owned(),
|
||||
span_asmt_type_start: 6,
|
||||
span_asmt_type_end: 11,
|
||||
span_start: 6,
|
||||
span_end: 15,
|
||||
__typename: "VariableDeclarationAST".to_owned()
|
||||
}],
|
||||
results
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parent_query() {
|
||||
let code = "interface MyGreatInterface { myGreatProperty: number }";
|
||||
|
||||
let allocator = Allocator::default();
|
||||
let source_type =
|
||||
SourceType::default().with_module(true).with_jsx(true).with_typescript(true);
|
||||
let ret = Parser::new(&allocator, code, source_type).parse();
|
||||
let program = allocator.alloc(ret.program);
|
||||
let semantic_ret =
|
||||
SemanticBuilder::new(code, source_type).with_trivias(&ret.trivias).build(program);
|
||||
|
||||
let adapter = Adapter {
|
||||
path_components: vec![Some("index".to_string())],
|
||||
semantic: Rc::new(semantic_ret.semantic),
|
||||
};
|
||||
|
||||
let args: BTreeMap<Arc<str>, FieldValue> = BTreeMap::new();
|
||||
|
||||
let adapter = Arc::from(&adapter);
|
||||
|
||||
#[allow(clippy::items_after_statements)]
|
||||
#[derive(Debug, PartialOrd, Ord, PartialEq, Eq, serde::Deserialize)]
|
||||
struct Output {
|
||||
type_: String,
|
||||
tn1: String,
|
||||
tn2: String,
|
||||
}
|
||||
|
||||
let mut results: Vec<Output> = execute_query(
|
||||
schema(),
|
||||
adapter,
|
||||
r#"
|
||||
query {
|
||||
File {
|
||||
ast_node {
|
||||
... on TypeAnnotationAST {
|
||||
type {
|
||||
type_: str @output
|
||||
}
|
||||
parent {
|
||||
tn1: __typename @output
|
||||
parent {
|
||||
tn2: __typename @output
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
"#,
|
||||
args,
|
||||
)
|
||||
.expect("to successfully execute the query")
|
||||
.map(|row| row.try_into_struct().expect("shape mismatch"))
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
results.sort_unstable();
|
||||
|
||||
assert_eq!(
|
||||
vec![Output {
|
||||
type_: "number".to_owned(),
|
||||
tn1: "ASTNode".to_owned(),
|
||||
tn2: "InterfaceAST".to_owned()
|
||||
}],
|
||||
results
|
||||
);
|
||||
}
|
||||
}
|
||||
mod tests;
|
||||
|
|
|
|||
207
crates/oxc_query/src/tests.rs
Normal file
207
crates/oxc_query/src/tests.rs
Normal file
|
|
@ -0,0 +1,207 @@
|
|||
use std::{collections::BTreeMap, rc::Rc, sync::Arc};
|
||||
|
||||
use oxc_allocator::Allocator;
|
||||
use oxc_parser::Parser;
|
||||
use oxc_semantic::SemanticBuilder;
|
||||
use oxc_span::SourceType;
|
||||
use trustfall::{execute_query, FieldValue, TryIntoStruct};
|
||||
|
||||
use crate::{adapter::schema, Adapter};
|
||||
|
||||
#[test]
|
||||
fn test_path_query() {
|
||||
let code = "const apple = 1;";
|
||||
|
||||
let allocator = Allocator::default();
|
||||
let source_type = SourceType::default().with_module(true).with_jsx(true).with_typescript(true);
|
||||
let ret = Parser::new(&allocator, code, source_type).parse();
|
||||
let program = allocator.alloc(ret.program);
|
||||
let semantic_ret =
|
||||
SemanticBuilder::new(code, source_type).with_trivias(&ret.trivias).build(program);
|
||||
|
||||
let adapter = Adapter {
|
||||
path_components: vec![Some("index".to_string())],
|
||||
semantic: Rc::new(semantic_ret.semantic),
|
||||
};
|
||||
|
||||
let args: BTreeMap<Arc<str>, FieldValue> = BTreeMap::new();
|
||||
|
||||
let adapter = Arc::from(&adapter);
|
||||
|
||||
#[allow(clippy::items_after_statements)]
|
||||
#[derive(Debug, PartialOrd, Ord, PartialEq, Eq, serde::Deserialize)]
|
||||
struct Output {
|
||||
name: String,
|
||||
is_first: bool,
|
||||
is_last: bool,
|
||||
}
|
||||
|
||||
let mut results: Vec<Output> = execute_query(
|
||||
schema(),
|
||||
adapter,
|
||||
r#"
|
||||
query {
|
||||
File {
|
||||
last_path_part {
|
||||
name @output
|
||||
is_first @output
|
||||
is_last @output
|
||||
}
|
||||
}
|
||||
}
|
||||
"#,
|
||||
args,
|
||||
)
|
||||
.expect("to successfully execute the query")
|
||||
.map(|row| row.try_into_struct().expect("shape mismatch"))
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
results.sort_unstable();
|
||||
|
||||
assert_eq!(vec![Output { name: "index".into(), is_first: true, is_last: true },], results);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_variable_query() {
|
||||
let code = "const apple = 1;";
|
||||
|
||||
let allocator = Allocator::default();
|
||||
let source_type = SourceType::default().with_module(true).with_jsx(true).with_typescript(true);
|
||||
let ret = Parser::new(&allocator, code, source_type).parse();
|
||||
let program = allocator.alloc(ret.program);
|
||||
let semantic_ret =
|
||||
SemanticBuilder::new(code, source_type).with_trivias(&ret.trivias).build(program);
|
||||
|
||||
let adapter = Adapter {
|
||||
path_components: vec![Some("index".to_string())],
|
||||
semantic: Rc::new(semantic_ret.semantic),
|
||||
};
|
||||
|
||||
let args: BTreeMap<Arc<str>, FieldValue> = BTreeMap::new();
|
||||
|
||||
let adapter = Arc::from(&adapter);
|
||||
|
||||
#[allow(clippy::items_after_statements)]
|
||||
#[derive(Debug, PartialOrd, Ord, PartialEq, Eq, serde::Deserialize)]
|
||||
struct Output {
|
||||
assignment_to_variable_name: String,
|
||||
span_asmt_type_start: u64,
|
||||
span_asmt_type_end: u64,
|
||||
span_start: u64,
|
||||
span_end: u64,
|
||||
__typename: String,
|
||||
}
|
||||
|
||||
let mut results: Vec<Output> = execute_query(
|
||||
schema(),
|
||||
adapter,
|
||||
r#"
|
||||
query {
|
||||
File {
|
||||
variable_declaration {
|
||||
left {
|
||||
assignment_to_variable_name @output
|
||||
span_asmt_type_: span {
|
||||
start @output
|
||||
end @output
|
||||
}
|
||||
}
|
||||
|
||||
__typename @output
|
||||
|
||||
span_: span {
|
||||
start @output
|
||||
end @output
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
"#,
|
||||
args,
|
||||
)
|
||||
.expect("to successfully execute the query")
|
||||
.map(|row| row.try_into_struct().expect("shape mismatch"))
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
results.sort_unstable();
|
||||
|
||||
assert_eq!(
|
||||
vec![Output {
|
||||
assignment_to_variable_name: "apple".to_owned(),
|
||||
span_asmt_type_start: 6,
|
||||
span_asmt_type_end: 11,
|
||||
span_start: 6,
|
||||
span_end: 15,
|
||||
__typename: "VariableDeclarationAST".to_owned()
|
||||
}],
|
||||
results
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parent_query() {
|
||||
let code = "interface MyGreatInterface { myGreatProperty: number }";
|
||||
|
||||
let allocator = Allocator::default();
|
||||
let source_type = SourceType::default().with_module(true).with_jsx(true).with_typescript(true);
|
||||
let ret = Parser::new(&allocator, code, source_type).parse();
|
||||
let program = allocator.alloc(ret.program);
|
||||
let semantic_ret =
|
||||
SemanticBuilder::new(code, source_type).with_trivias(&ret.trivias).build(program);
|
||||
|
||||
let adapter = Adapter {
|
||||
path_components: vec![Some("index".to_string())],
|
||||
semantic: Rc::new(semantic_ret.semantic),
|
||||
};
|
||||
|
||||
let args: BTreeMap<Arc<str>, FieldValue> = BTreeMap::new();
|
||||
|
||||
let adapter = Arc::from(&adapter);
|
||||
|
||||
#[allow(clippy::items_after_statements)]
|
||||
#[derive(Debug, PartialOrd, Ord, PartialEq, Eq, serde::Deserialize)]
|
||||
struct Output {
|
||||
type_: String,
|
||||
tn1: String,
|
||||
tn2: String,
|
||||
}
|
||||
|
||||
let mut results: Vec<Output> = execute_query(
|
||||
schema(),
|
||||
adapter,
|
||||
r#"
|
||||
query {
|
||||
File {
|
||||
ast_node {
|
||||
... on TypeAnnotationAST {
|
||||
type {
|
||||
type_: str @output
|
||||
}
|
||||
parent {
|
||||
tn1: __typename @output
|
||||
parent {
|
||||
tn2: __typename @output
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
"#,
|
||||
args,
|
||||
)
|
||||
.expect("to successfully execute the query")
|
||||
.map(|row| row.try_into_struct().expect("shape mismatch"))
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
results.sort_unstable();
|
||||
|
||||
assert_eq!(
|
||||
vec![Output {
|
||||
type_: "number".to_owned(),
|
||||
tn1: "ASTNode".to_owned(),
|
||||
tn2: "InterfaceAST".to_owned()
|
||||
}],
|
||||
results
|
||||
);
|
||||
}
|
||||
Loading…
Reference in a new issue