66 lines
1.5 KiB
Rust
66 lines
1.5 KiB
Rust
use crate::{
|
|
parser::types::VariableDefinition,
|
|
validation::visitor::{Visitor, VisitorContext},
|
|
Positioned,
|
|
};
|
|
|
|
#[derive(Default)]
|
|
pub struct VariablesAreInputTypes;
|
|
|
|
impl<'a> Visitor<'a> for VariablesAreInputTypes {
|
|
fn enter_variable_definition(
|
|
&mut self,
|
|
ctx: &mut VisitorContext<'a>,
|
|
variable_definition: &'a Positioned<VariableDefinition>,
|
|
) {
|
|
if let Some(ty) = ctx
|
|
.registry
|
|
.concrete_type_by_parsed_type(&variable_definition.node.var_type.node)
|
|
{
|
|
if !ty.is_input() {
|
|
ctx.report_error(
|
|
vec![variable_definition.pos],
|
|
format!(
|
|
"Variable \"{}\" cannot be of non-input type \"{}\"",
|
|
variable_definition.node.name.node,
|
|
ty.name()
|
|
),
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
pub fn factory() -> VariablesAreInputTypes {
|
|
VariablesAreInputTypes
|
|
}
|
|
|
|
#[test]
|
|
fn input_types_are_valid() {
|
|
expect_passes_rule!(
|
|
factory,
|
|
r#"
|
|
query Foo($a: String, $b: [Boolean!]!, $c: ComplexInput) {
|
|
field(a: $a, b: $b, c: $c)
|
|
}
|
|
"#,
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn output_types_are_invalid() {
|
|
expect_fails_rule!(
|
|
factory,
|
|
r#"
|
|
query Foo($a: Dog, $b: [[CatOrDog!]]!, $c: Pet) {
|
|
field(a: $a, b: $b, c: $c)
|
|
}
|
|
"#,
|
|
);
|
|
}
|
|
}
|