use std::fmt::Write as FmtWrite;
use crate::e2e::codegen::field_skip::{FieldSkip, nested_wildcard_skip_line};
use crate::e2e::escape::escape_rust;
use crate::e2e::field_access::FieldResolver;
use crate::e2e::fixture::Assertion;
use super::assertion_helpers::{
render_count_equals_assertion, render_count_min_assertion, render_equals_assertion, render_gte_assertion,
render_is_empty_assertion, render_method_result_assertion, render_not_empty_assertion,
};
use super::assertion_synthetic::{
numeric_literal, render_chunks_have_content, render_chunks_have_embeddings, render_chunks_have_heading_context,
render_embedding_dimensions, render_embedding_quality, render_embeddings_assertion,
render_first_chunk_starts_with_heading, render_keywords_assertion, render_keywords_count_assertion,
tree_field_access_expr, value_to_rust_string,
};
fn is_optional_scalar_field(assertion: &Assertion, is_unwrapped: bool, field_resolver: &FieldResolver) -> bool {
assertion.field.as_ref().is_some_and(|f| {
let resolved = field_resolver.resolve(f);
let is_opt = !is_unwrapped && field_resolver.is_optional(resolved);
let is_arr = field_resolver.is_array(resolved);
is_opt && !is_arr
})
}
fn containment_predicate(field_access: &str, expected: &str, field_is_enum: bool, field_is_collection: bool) -> String {
if field_is_enum {
format!("format!(\"{{:?}}\", {field_access}).to_lowercase().contains(&{expected}.to_lowercase())")
} else if field_is_collection {
format!(
"{field_access}.iter().any(|item| serde_json::to_value(item).ok().is_some_and(|value| match value {{ serde_json::Value::String(text) => text == {expected}, serde_json::Value::Object(fields) => fields.get(\"name\").and_then(serde_json::Value::as_str).is_some_and(|text| text == {expected}), _ => false }}))"
)
} else {
format!("{field_access}.contains({expected})")
}
}
fn containment_message(field_is_enum: bool, field_is_collection: bool) -> &'static str {
if !field_is_enum && field_is_collection {
"expected collection item name"
} else {
"expected to contain"
}
}
#[allow(clippy::too_many_arguments)]
pub fn render_assertion(
out: &mut String,
assertion: &Assertion,
result_var: &str,
module: &str,
dep_name: &str,
is_error_context: bool,
unwrapped_fields: &[(String, String)], field_resolver: &FieldResolver,
result_is_tree: bool,
result_is_simple: bool,
result_is_vec: bool,
result_is_option: bool,
returns_result: bool,
streaming_item_type: Option<&str>,
) {
render_assertion_with_streaming(
out,
assertion,
result_var,
module,
dep_name,
is_error_context,
unwrapped_fields,
field_resolver,
result_is_tree,
result_is_simple,
result_is_vec,
result_is_option,
returns_result,
streaming_item_type,
false,
)
}
#[allow(clippy::too_many_arguments)]
pub fn render_assertion_with_streaming(
out: &mut String,
assertion: &Assertion,
result_var: &str,
module: &str,
dep_name: &str,
is_error_context: bool,
unwrapped_fields: &[(String, String)], field_resolver: &FieldResolver,
result_is_tree: bool,
result_is_simple: bool,
result_is_vec: bool,
result_is_option: bool,
returns_result: bool,
streaming_item_type: Option<&str>,
_is_streaming: bool,
) {
let has_field = assertion.field.as_ref().is_some_and(|f| !f.is_empty());
if result_is_vec && has_field && !is_error_context {
let _ = writeln!(out, " for r in &{result_var} {{");
render_assertion(
out,
assertion,
"r",
module,
dep_name,
is_error_context,
unwrapped_fields,
field_resolver,
result_is_tree,
result_is_simple,
false, result_is_option,
returns_result,
streaming_item_type,
);
let _ = writeln!(out, " }}");
return;
}
if result_is_option && !is_error_context {
let assertion_type = assertion.assertion_type.as_str();
if !has_field && (assertion_type == "is_empty" || assertion_type == "not_empty") {
let check = if assertion_type == "is_empty" {
"is_none"
} else {
"is_some"
};
let _ = writeln!(
out,
" assert!({result_var}.{check}(), \"expected Option to be {check}\");"
);
return;
}
let _ = writeln!(
out,
" let r = {result_var}.as_ref().expect(\"Option<T> should be Some\");"
);
render_assertion(
out,
assertion,
"r",
module,
dep_name,
is_error_context,
unwrapped_fields,
field_resolver,
result_is_tree,
result_is_simple,
result_is_vec,
false, returns_result,
streaming_item_type,
);
return;
}
if let Some(f) = &assertion.field {
match f.as_str() {
"chunks_have_content" => {
render_chunks_have_content(out, result_var, assertion.assertion_type.as_str());
return;
}
"chunks_have_embeddings" => {
render_chunks_have_embeddings(out, result_var, assertion.assertion_type.as_str());
return;
}
"chunks_have_heading_context" => {
render_chunks_have_heading_context(out, result_var, assertion.assertion_type.as_str());
return;
}
"first_chunk_starts_with_heading" => {
render_first_chunk_starts_with_heading(out, result_var, assertion.assertion_type.as_str());
return;
}
"embeddings" => {
render_embeddings_assertion(out, result_var, assertion);
return;
}
"embedding_dimensions" => {
render_embedding_dimensions(out, result_var, assertion);
return;
}
"embeddings_valid" | "embeddings_finite" | "embeddings_non_zero" | "embeddings_normalized" => {
render_embedding_quality(out, result_var, f, assertion.assertion_type.as_str());
return;
}
"keywords" => {
render_keywords_assertion(out, result_var, assertion);
return;
}
"keywords_count" => {
render_keywords_count_assertion(out, result_var, assertion);
return;
}
_ => {}
}
}
if let Some(f) = &assertion.field
&& !f.is_empty()
&& crate::e2e::codegen::streaming_assertions::is_streaming_virtual_field(f)
{
if let Some(expr) =
crate::e2e::codegen::streaming_assertions::StreamingFieldResolver::accessor_with_streaming_context(
f,
"rust",
"chunks",
Some(dep_name),
streaming_item_type,
)
{
match assertion.assertion_type.as_str() {
"count_min" => {
if let Some(val) = &assertion.value
&& let Some(n) = val.as_u64()
{
let expr_for_len = if field_resolver.is_optional(f) {
format!("{expr}.as_ref().map_or(0, |v| v.len())")
} else {
format!("{expr}.len()")
};
let _ = writeln!(
out,
" assert!({expr_for_len} >= {n} as usize, \"expected >= {n} chunks\");"
);
} else {
panic!(
"Rust e2e generator: streaming field '{f}' assertion 'count_min' requires a numeric value in the fixture, got {:?}",
assertion.value
);
}
}
"count_equals" => {
if let Some(val) = &assertion.value
&& let Some(n) = val.as_u64()
{
let expr_for_len = if field_resolver.is_optional(f) {
format!("{expr}.as_ref().map_or(0, |v| v.len())")
} else {
format!("{expr}.len()")
};
let _ = writeln!(
out,
" assert_eq!({expr_for_len}, {n} as usize, \"expected exactly {n} chunks\");"
);
} else {
panic!(
"Rust e2e generator: streaming field '{f}' assertion 'count_equals' requires a numeric value in the fixture, got {:?}",
assertion.value
);
}
}
"equals" => {
if let Some(serde_json::Value::String(s)) = &assertion.value {
let escaped = crate::e2e::escape::escape_rust(s);
let _ = writeln!(out, " assert_eq!({expr}, \"{escaped}\");");
} else if let Some(val) = &assertion.value {
let lit = super::assertion_synthetic::numeric_literal(val);
let _ = writeln!(out, " assert_eq!({expr}, {lit});");
} else {
panic!(
"Rust e2e generator: streaming field '{f}' assertion 'equals' requires a string or numeric value in the fixture, got {:?}",
assertion.value
);
}
}
"not_empty" => {
let check_expr = if field_resolver.is_optional(f) {
format!("{expr}.as_ref().is_some_and(|v| !v.is_empty())")
} else {
format!("!{expr}.is_empty()")
};
let _ = writeln!(out, " assert!({check_expr}, \"expected non-empty\");");
}
"is_empty" => {
let check_expr = if field_resolver.is_optional(f) {
format!("{expr}.as_ref().is_none_or(|v| v.is_empty())")
} else {
format!("{expr}.is_empty()")
};
let _ = writeln!(out, " assert!({check_expr}, \"expected empty\");");
}
"is_true" => {
let _ = writeln!(out, " assert!({expr}, \"expected true\");");
}
"is_false" => {
let _ = writeln!(out, " assert!(!{expr}, \"expected false\");");
}
"greater_than" => {
if let Some(val) = &assertion.value {
let lit = super::assertion_synthetic::numeric_literal(val);
let _ = writeln!(out, " assert!({expr} > {lit}, \"expected > {lit}\");");
} else {
panic!(
"Rust e2e generator: streaming field '{f}' assertion 'greater_than' requires a numeric value in the fixture, got {:?}",
assertion.value
);
}
}
"greater_than_or_equal" => {
if let Some(val) = &assertion.value {
let lit = super::assertion_synthetic::numeric_literal(val);
let _ = writeln!(out, " assert!({expr} >= {lit}, \"expected >= {lit}\");");
} else {
panic!(
"Rust e2e generator: streaming field '{f}' assertion 'greater_than_or_equal' requires a numeric value in the fixture, got {:?}",
assertion.value
);
}
}
"contains" => {
if let Some(serde_json::Value::String(s)) = &assertion.value {
let escaped = crate::e2e::escape::escape_rust(s);
let _ = writeln!(
out,
" assert!({expr}.contains(\"{escaped}\"), \"expected to contain: {escaped}\");"
);
} else {
panic!(
"Rust e2e generator: streaming field '{f}' assertion 'contains' requires a string value in the fixture, got {:?}",
assertion.value
);
}
}
other => {
panic!("Rust e2e generator: unsupported assertion type '{other}' on streaming field '{f}'");
}
}
} else {
panic!(
"Rust e2e generator: streaming field '{f}' has no accessor for context (streaming_item_type={streaming_item_type:?}); check the streaming adapter configuration"
);
}
return;
}
if let Some(f) = &assertion.field
&& !f.is_empty()
{
if f.starts_with("error.") && !is_error_context {
let _ = writeln!(
out,
" // skipped: {}",
FieldSkip::NotAvailableOnResultType.message(f)
);
return;
}
if !f.starts_with("error.") && !result_is_simple && !field_resolver.is_valid_for_result(f) {
let _ = writeln!(
out,
" // skipped: {}",
FieldSkip::NotAvailableOnResultType.message(f)
);
return;
}
}
let is_unwrapped = assertion
.field
.as_ref()
.is_some_and(|f| unwrapped_fields.iter().any(|(ff, _)| ff == f));
let has_field = assertion.field.as_ref().is_some_and(|f| !f.is_empty());
let is_field_assertion = !matches!(assertion.assertion_type.as_str(), "error" | "not_error");
let is_error_field = assertion.field.as_ref().is_some_and(|f| f.starts_with("error."));
let effective_result_var =
if has_field && is_error_context && returns_result && is_field_assertion && !is_error_field {
format!("{result_var}_ok.as_ref().unwrap()")
} else {
result_var.to_string()
};
if let Some(f) = assertion.field.as_deref()
&& !f.is_empty()
&& !f.starts_with("error.")
&& !result_is_simple
&& !result_is_tree
&& !is_unwrapped
&& f != result_var
&& let Some((array_part, elem_part)) = field_resolver.wildcard_split(f)
{
render_rust_wildcard_assertion(
out,
assertion,
f,
&array_part,
&elem_part,
&effective_result_var,
field_resolver,
);
return;
}
let field_access = match &assertion.field {
Some(f) if !f.is_empty() => {
if let Some((_, local_var)) = unwrapped_fields.iter().find(|(ff, _)| ff == f) {
local_var.clone()
} else if result_is_simple && !f.starts_with("error.") {
effective_result_var.clone()
} else if f == result_var {
effective_result_var.clone()
} else if result_is_tree {
tree_field_access_expr(f, &effective_result_var, module)
} else if let Some(sub) = f.strip_prefix("error.") {
let err_accessor = field_resolver.accessor_for_error(sub, "rust", "__err");
format!("{{ let __err = {result_var}.as_ref().err().unwrap(); {err_accessor} }}")
} else {
field_resolver.accessor(f, "rust", &effective_result_var)
}
}
_ => effective_result_var,
};
let field_is_enum = assertion
.field
.as_deref()
.is_some_and(|field| field_resolver.is_enum(field));
let field_is_collection = assertion
.field
.as_deref()
.is_some_and(|field| field_resolver.is_array(field) || field_resolver.is_collection_root(field))
|| result_is_vec;
match assertion.assertion_type.as_str() {
"error" => {
let _ = writeln!(out, " assert!({result_var}.is_err(), \"expected call to fail\");");
if let Some(serde_json::Value::String(msg)) = &assertion.value {
let escaped = escape_rust(msg);
let _ = writeln!(
out,
" {{ let __e = {result_var}.as_ref().err().unwrap(); assert!(format!(\"{{:?}}\", __e).contains(\"{escaped}\") || __e.to_string().contains(\"{escaped}\"), \"error message mismatch\"); }}"
);
}
}
"not_error" => {
}
"equals" => {
render_equals_assertion(out, assertion, &field_access, is_unwrapped, field_resolver);
}
"contains" => {
if let Some(val) = &assertion.value {
let expected = value_to_rust_string(val);
let predicate = containment_predicate(&field_access, &expected, field_is_enum, field_is_collection);
let message = containment_message(field_is_enum, field_is_collection);
let _ = writeln!(out, " assert!({predicate}, \"{message}: {{}}\", {expected});");
}
}
"contains_all" => {
if let Some(values) = &assertion.values {
for val in values {
let expected = value_to_rust_string(val);
let predicate = containment_predicate(&field_access, &expected, field_is_enum, field_is_collection);
let message = containment_message(field_is_enum, field_is_collection);
let _ = writeln!(out, " assert!({predicate}, \"{message}: {{}}\", {expected});");
}
}
}
"not_contains" => {
for val in assertion.expected_values() {
let expected = value_to_rust_string(val);
let predicate = containment_predicate(&field_access, &expected, field_is_enum, field_is_collection);
let _ = writeln!(
out,
" assert!(!{predicate}, \"expected NOT to contain: {{}}\", {expected});"
);
}
}
"not_empty" => {
render_not_empty_assertion(
out,
assertion,
&field_access,
result_var,
result_is_option,
is_unwrapped,
field_resolver,
);
}
"is_empty" => {
render_is_empty_assertion(out, assertion, &field_access, is_unwrapped, field_resolver);
}
"contains_any" => {
if let Some(values) = &assertion.values {
let checks: Vec<String> = values
.iter()
.map(|v| {
let expected = value_to_rust_string(v);
containment_predicate(&field_access, &expected, field_is_enum, field_is_collection)
})
.collect();
let joined = checks.join(" || ");
let _ = writeln!(
out,
" assert!({joined}, \"expected to contain at least one of the specified values\");"
);
}
}
"greater_than" => {
if let Some(val) = &assertion.value {
if val.as_f64().is_some_and(|n| n < 0.0) {
let _ = writeln!(
out,
" // skipped: greater_than with negative value is always true for unsigned types"
);
} else if val.as_u64() == Some(0) {
if field_access.ends_with(".len()") {
let base = field_access.strip_suffix(".len()").unwrap();
let _ = writeln!(out, " assert!(!{base}.is_empty(), \"expected > 0\");");
} else if is_optional_scalar_field(assertion, is_unwrapped, field_resolver) {
let _ = writeln!(out, " assert!({field_access}.unwrap_or(0) > 0, \"expected > 0\");");
} else {
let _ = writeln!(out, " assert!({field_access} > 0, \"expected > 0\");");
}
} else {
let lit = numeric_literal(val);
if is_optional_scalar_field(assertion, is_unwrapped, field_resolver) {
let default_literal = if lit.contains("_f64") || lit.contains('.') {
"0.0"
} else {
"0"
};
let _ = writeln!(
out,
" assert!({field_access}.unwrap_or({default_literal}) > {lit}, \"expected > {lit}\");"
);
} else {
let _ = writeln!(out, " assert!({field_access} > {lit}, \"expected > {lit}\");");
}
}
}
}
"less_than" => {
if let Some(val) = &assertion.value {
let lit = numeric_literal(val);
if is_optional_scalar_field(assertion, is_unwrapped, field_resolver) {
let default_literal = if lit.contains("_f64") || lit.contains('.') {
"0.0"
} else {
"0"
};
let _ = writeln!(
out,
" assert!({field_access}.unwrap_or({default_literal}) < {lit}, \"expected < {lit}\");"
);
} else {
let _ = writeln!(out, " assert!({field_access} < {lit}, \"expected < {lit}\");");
}
}
}
"greater_than_or_equal" => {
render_gte_assertion(out, assertion, &field_access, is_unwrapped, field_resolver);
}
"less_than_or_equal" => {
if let Some(val) = &assertion.value {
let lit = numeric_literal(val);
if is_optional_scalar_field(assertion, is_unwrapped, field_resolver) {
let default_literal = if lit.contains("_f64") || lit.contains('.') {
"0.0"
} else {
"0"
};
let _ = writeln!(
out,
" assert!({field_access}.unwrap_or({default_literal}) <= {lit}, \"expected <= {lit}\");"
);
} else {
let _ = writeln!(out, " assert!({field_access} <= {lit}, \"expected <= {lit}\");");
}
}
}
"starts_with" => {
if let Some(val) = &assertion.value {
let expected = value_to_rust_string(val);
let _ = writeln!(
out,
" assert!({field_access}.starts_with({expected}), \"expected to start with: {{}}\", {expected});"
);
}
}
"ends_with" => {
if let Some(val) = &assertion.value {
let expected = value_to_rust_string(val);
let _ = writeln!(
out,
" assert!({field_access}.ends_with({expected}), \"expected to end with: {{}}\", {expected});"
);
}
}
"min_length" => {
if let Some(val) = &assertion.value
&& let Some(n) = val.as_u64()
{
if n == 1 {
let _ = writeln!(
out,
" assert!(!{field_access}.is_empty(), \"expected length >= 1, got {{}}\", {field_access}.len());"
);
} else {
let _ = writeln!(
out,
" assert!({field_access}.len() >= {n}, \"expected length >= {n}, got {{}}\", {field_access}.len());"
);
}
}
}
"max_length" => {
if let Some(val) = &assertion.value
&& let Some(n) = val.as_u64()
{
let _ = writeln!(
out,
" assert!({field_access}.len() <= {n}, \"expected length <= {n}, got {{}}\", {field_access}.len());"
);
}
}
"count_min" => {
render_count_min_assertion(out, assertion, &field_access, is_unwrapped, field_resolver);
}
"count_equals" => {
render_count_equals_assertion(out, assertion, &field_access, is_unwrapped, field_resolver);
}
"is_true" => {
if is_optional_scalar_field(assertion, is_unwrapped, field_resolver) {
let _ = writeln!(out, " assert!({field_access}.is_some(), \"expected true (Some)\");");
} else {
let _ = writeln!(out, " assert!({field_access}, \"expected true\");");
}
}
"is_false" => {
if is_optional_scalar_field(assertion, is_unwrapped, field_resolver) {
let _ = writeln!(out, " assert!({field_access}.is_none(), \"expected false (None)\");");
} else {
let _ = writeln!(out, " assert!(!{field_access}, \"expected false\");");
}
}
"method_result" => {
render_method_result_assertion(out, assertion, &field_access, result_is_tree, module);
}
other => {
panic!("Rust e2e generator: unsupported assertion type: {other}");
}
}
}
fn rust_wildcard_any(array_accessor: &str, array_is_optional: bool, element_predicate: &str) -> String {
if array_is_optional {
format!("{array_accessor}.as_ref().is_some_and(|v| v.iter().any(|e| {element_predicate}))")
} else {
format!("{array_accessor}.iter().any(|e| {element_predicate})")
}
}
fn render_rust_wildcard_assertion(
out: &mut String,
assertion: &Assertion,
field: &str,
array_part: &str,
elem_part: &str,
result_var: &str,
field_resolver: &FieldResolver,
) {
if let Some(line) = nested_wildcard_skip_line(" ", "//", field, elem_part) {
let _ = writeln!(out, "{line}");
return;
}
let array_accessor = if array_part.is_empty() {
result_var.to_string()
} else {
field_resolver.accessor(array_part, "rust", result_var)
};
let elem_accessor = if elem_part.is_empty() {
"e".to_string()
} else {
field_resolver.accessor(elem_part, "rust", "e")
};
let array_is_optional = !array_part.is_empty() && field_resolver.is_optional(array_part);
let escaped_field = escape_rust(field);
match assertion.assertion_type.as_str() {
"contains" | "contains_all" | "not_contains" => {
let negate = assertion.assertion_type == "not_contains";
let values: Vec<&serde_json::Value> = if assertion.assertion_type == "contains" {
assertion.value.iter().collect()
} else {
assertion.expected_values()
};
for val in values {
let expected = value_to_rust_string(val);
let pattern = if val.is_string() {
expected.clone()
} else {
format!("&{expected}.to_string()")
};
let predicate = rust_wildcard_any(
&array_accessor,
array_is_optional,
&format!("{elem_accessor}.to_string().contains({pattern})"),
);
if negate {
let _ = writeln!(
out,
" assert!(!{predicate}, \"expected no element of {escaped_field} to contain: {{}}\", {expected});"
);
} else {
let _ = writeln!(
out,
" assert!({predicate}, \"expected some element of {escaped_field} to contain: {{}}\", {expected});"
);
}
}
}
"not_empty" => {
let predicate = rust_wildcard_any(
&array_accessor,
array_is_optional,
&format!("!{elem_accessor}.to_string().is_empty()"),
);
let _ = writeln!(
out,
" assert!({predicate}, \"expected some element of {escaped_field} to be non-empty\");"
);
}
other => {
let _ = writeln!(
out,
" // skipped: unsupported traversal assertion '{other}' on '{field}'"
);
}
}
}
#[cfg(test)]
mod tests {
use std::collections::{HashMap, HashSet};
use super::*;
use crate::e2e::field_access::FieldResolver;
use crate::e2e::fixture::Assertion;
fn empty_resolver() -> FieldResolver {
FieldResolver::new(
&HashMap::new(),
&HashSet::new(),
&HashSet::new(),
&HashSet::new(),
&HashSet::new(),
)
}
fn make_assertion(assertion_type: &str, field: Option<&str>, value: Option<serde_json::Value>) -> Assertion {
Assertion {
assertion_type: assertion_type.to_string(),
field: field.map(|s| s.to_string()),
value,
..Default::default()
}
}
fn array_resolver(array_field: &str) -> FieldResolver {
FieldResolver::new(
&HashMap::new(),
&HashSet::new(),
&HashSet::new(),
&HashSet::from([array_field.to_string()]),
&HashSet::new(),
)
}
fn render_field_contains(resolver: &FieldResolver, field: &str, value: &str) -> String {
let assertion = make_assertion("contains", Some(field), Some(serde_json::json!(value)));
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
resolver,
false,
false,
false,
false,
false,
None,
);
out
}
#[test]
fn rust_wildcard_contains_iterates_every_element() {
let out = render_field_contains(&array_resolver("links"), "links[].link_type", "external");
assert!(out.contains("result.links.iter().any(|e|"), "got: {out}");
assert!(out.contains("e.link_type.to_string().contains("), "got: {out}");
assert!(!out.contains("[0]"), "wildcard must not pin element 0, got: {out}");
}
#[test]
fn rust_explicit_index_still_pins_element_zero() {
let out = render_field_contains(&array_resolver("links"), "links[0].link_type", "external");
assert!(out.contains("result.links[0].link_type"), "got: {out}");
assert!(
!out.contains(".iter().any("),
"explicit index must not become a traversal, got: {out}"
);
}
#[test]
fn rust_wildcard_match_in_second_element_is_not_missed() {
let out = render_field_contains(&array_resolver("links"), "links[].link_type", "canonical");
assert!(out.contains(".iter().any("), "got: {out}");
assert!(!out.contains("links[0]"), "got: {out}");
assert!(!out.contains("links[1]"), "predicate must be index-free, got: {out}");
}
#[test]
fn nested_wildcard_should_emit_a_visible_skip_rather_than_an_index_zero_check() {
let out = render_field_contains(&array_resolver("pages"), "pages[].links[].url", "example.test");
assert_eq!(
out, " // skipped: nested array-wildcard field 'pages[].links[].url' not supported\n",
"got: {out}"
);
}
#[test]
fn rust_wildcard_optional_array_guards_with_is_some_and() {
let resolver = FieldResolver::new(
&HashMap::new(),
&HashSet::from(["links".to_string()]),
&HashSet::new(),
&HashSet::from(["links".to_string()]),
&HashSet::new(),
);
let out = render_field_contains(&resolver, "links[].link_type", "external");
assert!(out.contains(".as_ref().is_some_and(|v| v.iter().any(|e|"), "got: {out}");
}
#[test]
fn rust_ir_reachable_field_absent_from_result_fields_is_not_skipped() {
let reachable: HashSet<String> = ["data".to_string()].into_iter().collect();
let resolver = FieldResolver::new(
&HashMap::new(),
&HashSet::new(),
&HashSet::new(),
&HashSet::new(),
&HashSet::new(),
)
.with_ir_fields(reachable, HashSet::new());
let assertion = make_assertion("equals", Some("data"), Some(serde_json::json!("hello")));
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(!out.contains("skipped"), "got: {out}");
}
#[test]
fn rust_ir_excluded_field_present_in_result_fields_is_still_skipped() {
let result_fields: HashSet<String> = ["internal_diagnostics".to_string()].into_iter().collect();
let excluded: HashSet<String> = ["internal_diagnostics".to_string()].into_iter().collect();
let resolver = FieldResolver::new(
&HashMap::new(),
&HashSet::new(),
&result_fields,
&HashSet::new(),
&HashSet::new(),
)
.with_ir_fields(HashSet::new(), excluded);
let assertion = make_assertion("equals", Some("internal_diagnostics"), Some(serde_json::json!("hello")));
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(out.contains("skipped"), "got: {out}");
}
#[test]
fn render_assertion_error_type_emits_is_err_check() {
let resolver = empty_resolver();
let assertion = make_assertion("error", None, None);
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
true,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(out.contains("is_err()"), "got: {out}");
}
#[test]
fn render_contains_assertion_uses_raw_string_content() {
let resolver = empty_resolver();
let assertion = make_assertion("contains", None, Some(serde_json::json!("line\n\"quoted\"\\path")));
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"sample",
"sample",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(out.contains("result.contains("), "got: {out}");
assert!(!out.contains("format!(\"{{:?}}\""), "got: {out}");
}
#[test]
fn render_not_contains_emits_each_plural_value() {
let resolver = empty_resolver();
let assertion = Assertion {
assertion_type: "not_contains".into(),
field: Some("content".into()),
values: Some(vec![
serde_json::json!("unsafe markup"),
serde_json::json!("unsafe handler"),
]),
..Default::default()
};
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"sample",
"sample",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(out.contains("unsafe markup"), "got: {out}");
assert!(out.contains("unsafe handler"), "got: {out}");
assert_eq!(out.matches("assert!(!").count(), 2, "got: {out}");
}
#[test]
fn render_assertion_vec_result_wraps_in_for_loop() {
let resolver = empty_resolver();
let assertion = make_assertion("not_empty", Some("content"), None);
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
true,
false,
false,
None,
);
assert!(out.contains("for r in"), "got: {out}");
}
#[test]
fn render_assertion_not_empty_bare_result_uses_is_empty() {
let resolver = empty_resolver();
let assertion = make_assertion("not_empty", None, None);
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(out.contains("is_empty()"), "got: {out}");
}
#[test]
fn render_assertion_min_length_one_uses_is_empty_not_len_ge_one() {
let resolver = empty_resolver();
let assertion = make_assertion("min_length", Some("content"), Some(serde_json::Value::from(1u64)));
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(
out.contains("is_empty()"),
"min_length 1 should use !is_empty(); got: {out}"
);
assert!(
!out.contains("len() >= 1"),
"min_length 1 must not emit len() >= 1 (clippy::len_zero); got: {out}"
);
}
#[test]
fn render_assertion_min_length_two_still_uses_len_ge() {
let resolver = empty_resolver();
let assertion = make_assertion("min_length", Some("content"), Some(serde_json::Value::from(2u64)));
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(
out.contains("len() >= 2"),
"min_length 2 should emit len() >= 2; got: {out}"
);
}
#[test]
fn contains_uses_declared_collection_and_enum_types() {
let result_fields = HashSet::from(["cookies".to_string(), "link_type".to_string()]);
let array_fields = HashSet::from(["cookies".to_string()]);
let resolver = FieldResolver::new(
&HashMap::new(),
&HashSet::new(),
&result_fields,
&array_fields,
&HashSet::new(),
)
.with_enum_fields(HashSet::from(["link_type".to_string()]));
let mut assertions = String::new();
for (field, expected) in [("cookies", "domain_cookie"), ("link_type", "anchor")] {
render_assertion(
&mut assertions,
&make_assertion("contains", Some(field), Some(serde_json::json!(expected))),
"result",
"sample",
"sample",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
}
assert_eq!(assertions.matches("format!(\"{:?}\"").count(), 1, "got: {assertions}");
assert!(assertions.contains("result.cookies.iter().any"), "got: {assertions}");
assert!(assertions.contains("fields.get(\"name\")"), "got: {assertions}");
}
#[test]
fn contains_uses_the_effective_result_type() {
let resolver = FieldResolver::new(
&HashMap::new(),
&HashSet::new(),
&HashSet::new(),
&HashSet::from(["items".to_string()]),
&HashSet::new(),
);
let assertion = make_assertion("contains", Some("items"), Some(serde_json::json!("needle")));
let mut scalar = String::new();
render_assertion(
&mut scalar,
&assertion,
"result",
"sample",
"sample",
false,
&[],
&resolver,
false,
true,
false,
false,
false,
None,
);
assert!(!scalar.contains("format!(\"{:?}\""), "got: {scalar}");
let mut vector = String::new();
render_assertion(
&mut vector,
&make_assertion("contains", None, Some(serde_json::json!("needle"))),
"result",
"sample",
"sample",
false,
&[],
&resolver,
false,
true,
true,
false,
false,
None,
);
assert!(vector.contains("result.iter().any"), "got: {vector}");
assert!(!vector.contains("format!(\"{:?}\""), "got: {vector}");
}
const CONTAINMENT_OPERATORS: [&str; 4] = ["contains", "contains_all", "not_contains", "contains_any"];
fn render_containment(operator: &str, field: &str, expected: &str, resolver: &FieldResolver) -> String {
let assertion = Assertion {
assertion_type: operator.to_string(),
field: Some(field.to_string()),
value: Some(serde_json::json!(expected)),
values: Some(vec![serde_json::json!(expected)]),
..Default::default()
};
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"sample",
"sample",
false,
&[],
resolver,
false,
false,
false,
false,
false,
None,
);
out
}
#[test]
fn every_containment_operator_uses_the_enum_predicate_on_an_enum_field() {
let resolver = empty_resolver().with_enum_fields(HashSet::from(["link_type".to_string()]));
for operator in CONTAINMENT_OPERATORS {
let rendered = render_containment(operator, "link_type", "anchor", &resolver);
assert!(
rendered.contains("format!(\"{:?}\", result.link_type).to_lowercase()"),
"`{operator}` must compare an enum field through its Debug form, or the generated \
test cannot compile — an enum has no inherent `contains`; got: {rendered}"
);
assert!(
!rendered.contains("result.link_type.contains("),
"`{operator}` still calls `contains` directly on an enum field; got: {rendered}"
);
}
}
#[test]
fn every_containment_operator_uses_the_collection_predicate_on_a_collection_field() {
let resolver = FieldResolver::new(
&HashMap::new(),
&HashSet::new(),
&HashSet::from(["cookies".to_string()]),
&HashSet::from(["cookies".to_string()]),
&HashSet::new(),
);
for operator in CONTAINMENT_OPERATORS {
let rendered = render_containment(operator, "cookies", "session", &resolver);
assert!(
rendered.contains("result.cookies.iter().any("),
"`{operator}` must match a collection field element-wise, or it compares a whole \
element against a name; got: {rendered}"
);
assert!(
rendered.contains("fields.get(\"name\")"),
"`{operator}` must accept an object element matched by its `name`; got: {rendered}"
);
}
}
#[test]
fn every_containment_operator_emits_parseable_rust() {
let resolver = FieldResolver::new(
&HashMap::new(),
&HashSet::new(),
&HashSet::from(["cookies".to_string()]),
&HashSet::from(["cookies".to_string()]),
&HashSet::new(),
)
.with_enum_fields(HashSet::from(["link_type".to_string()]));
for operator in CONTAINMENT_OPERATORS {
for (field, expected) in [("link_type", "anchor"), ("cookies", "session"), ("content", "plain")] {
let body = render_containment(operator, field, expected, &resolver);
let unit = format!("fn generated() {{\n{body}}}\n");
syn::parse_file(&unit).unwrap_or_else(|error| {
panic!("`{operator}` on `{field}` must emit parseable Rust: {error}\n{unit}")
});
}
}
}
#[test]
fn contains_emits_unchanged_bytes_for_every_field_kind() {
let resolver = FieldResolver::new(
&HashMap::new(),
&HashSet::new(),
&HashSet::from(["content".to_string(), "link_type".to_string(), "cookies".to_string()]),
&HashSet::from(["cookies".to_string()]),
&HashSet::new(),
)
.with_enum_fields(HashSet::from(["link_type".to_string()]));
let expected = [
(
"content",
"needle",
" assert!(result.content.contains(r#\"needle\"#), \"expected to contain: {}\", r#\"needle\"#);\n",
),
(
"link_type",
"anchor",
" assert!(format!(\"{:?}\", result.link_type).to_lowercase().contains(&r#\"anchor\"#.to_lowercase()), \"expected to contain: {}\", r#\"anchor\"#);\n",
),
(
"cookies",
"session",
" assert!(result.cookies.iter().any(|item| serde_json::to_value(item).ok().is_some_and(|value| match value { serde_json::Value::String(text) => text == r#\"session\"#, serde_json::Value::Object(fields) => fields.get(\"name\").and_then(serde_json::Value::as_str).is_some_and(|text| text == r#\"session\"#), _ => false })), \"expected collection item name: {}\", r#\"session\"#);\n",
),
];
for (field, value, want) in expected {
let assertion = make_assertion("contains", Some(field), Some(serde_json::json!(value)));
let mut got = String::new();
render_assertion(
&mut got,
&assertion,
"result",
"sample",
"sample",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert_eq!(got, want, "`contains` on `{field}` changed its emitted bytes");
}
}
const ENUM_PREDICATE: &str = r#"format!("{:?}", kind).to_lowercase().contains(&"anchor".to_lowercase())"#;
const COLLECTION_PREDICATE: &str = r#"items.iter().any(|item| serde_json::to_value(item).ok().is_some_and(|value| match value { serde_json::Value::String(text) => text == "needle", serde_json::Value::Object(fields) => fields.get("name").and_then(serde_json::Value::as_str).is_some_and(|text| text == "needle"), _ => false }))"#;
#[test]
fn the_enum_predicate_is_valid_rust_against_a_real_enum() {
#[derive(Debug)]
enum SampleKind {
Anchor,
}
let kind = SampleKind::Anchor;
assert!(format!("{:?}", kind).to_lowercase().contains(&"anchor".to_lowercase()));
assert_eq!(containment_predicate("kind", "\"anchor\"", true, false), ENUM_PREDICATE);
}
#[test]
fn the_collection_predicate_is_valid_rust_against_a_real_collection() {
#[derive(serde::Serialize)]
struct SampleItem {
name: String,
}
let items = [SampleItem {
name: "needle".to_string(),
}];
assert!(
items
.iter()
.any(|item| serde_json::to_value(item).ok().is_some_and(|value| match value {
serde_json::Value::String(text) => text == "needle",
serde_json::Value::Object(fields) => fields
.get("name")
.and_then(serde_json::Value::as_str)
.is_some_and(|text| text == "needle"),
_ => false,
}))
);
assert_eq!(
containment_predicate("items", "\"needle\"", false, true),
COLLECTION_PREDICATE
);
}
#[test]
fn simple_result_enum_contains_uses_enum_value() {
let resolver = empty_resolver().with_enum_fields(HashSet::from(["link_type".to_string()]));
let mut output = String::new();
render_assertion(
&mut output,
&make_assertion("contains", Some("link_type"), Some(serde_json::json!("anchor"))),
"result.links[0].link_type",
"sample",
"sample",
false,
&[],
&resolver,
false,
true,
false,
false,
false,
None,
);
assert!(
output.contains("format!(\"{:?}\", result.links[0].link_type)"),
"got: {output}"
);
assert!(!output.contains("link_type.contains"), "got: {output}");
}
#[test]
#[should_panic(expected = "streaming field 'chunks' assertion 'count_min' requires a numeric")]
fn streaming_count_min_without_value_fails_loudly() {
let resolver = empty_resolver();
let assertion = make_assertion("count_min", Some("chunks"), None);
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
}
#[test]
#[should_panic(expected = "unsupported assertion type 'bogus_type' on streaming field 'chunks'")]
fn streaming_assertion_unknown_type_fails_loudly() {
let resolver = empty_resolver();
let assertion = make_assertion("bogus_type", Some("chunks"), None);
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
}
#[test]
#[should_panic(expected = "streaming field 'stream.has_page_event' has no accessor for context")]
fn streaming_field_without_accessor_fails_loudly() {
let resolver = empty_resolver();
let assertion = make_assertion("is_true", Some("stream.has_page_event"), None);
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
}
#[test]
fn streaming_not_empty_still_renders_real_assertion() {
let resolver = empty_resolver();
let assertion = make_assertion("not_empty", Some("chunks"), None);
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(out.contains("assert!"), "got: {out}");
assert!(out.contains("expected non-empty"), "got: {out}");
}
#[test]
fn streaming_count_min_with_value_still_renders_real_assertion() {
let resolver = empty_resolver();
let assertion = make_assertion("count_min", Some("chunks"), Some(serde_json::json!(3)));
let mut out = String::new();
render_assertion(
&mut out,
&assertion,
"result",
"my_mod",
"dep",
false,
&[],
&resolver,
false,
false,
false,
false,
false,
None,
);
assert!(out.contains("assert!(chunks.len() >= 3 as usize"), "got: {out}");
}
}