1use std::fmt::Write as FmtWrite;
4
5use crate::config::E2eConfig;
6use crate::escape::sanitize_filename;
7use crate::field_access::FieldResolver;
8use crate::fixture::{Fixture, FixtureGroup};
9
10use super::args::{emit_rust_visitor_method, render_rust_arg, resolve_visitor_trait};
11use super::assertions::render_assertion;
12use super::http::render_http_test_function;
13use super::mock_server::render_mock_server_setup;
14
15pub(super) fn resolve_function_name_for_call(call_config: &crate::config::CallConfig) -> String {
16 call_config
17 .overrides
18 .get("rust")
19 .and_then(|o| o.function.clone())
20 .unwrap_or_else(|| call_config.function.clone())
21}
22
23pub(super) fn resolve_module(e2e_config: &E2eConfig, dep_name: &str) -> String {
24 resolve_module_for_call(&e2e_config.call, dep_name)
25}
26
27pub(super) fn resolve_module_for_call(call_config: &crate::config::CallConfig, dep_name: &str) -> String {
28 let overrides = call_config.overrides.get("rust");
31 overrides
32 .and_then(|o| o.crate_name.clone())
33 .or_else(|| overrides.and_then(|o| o.module.clone()))
34 .unwrap_or_else(|| dep_name.to_string())
35}
36
37pub(super) fn is_skipped(fixture: &Fixture, language: &str) -> bool {
38 fixture.skip.as_ref().is_some_and(|s| s.should_skip(language))
39}
40
41pub fn render_test_file(
42 category: &str,
43 fixtures: &[&Fixture],
44 e2e_config: &E2eConfig,
45 dep_name: &str,
46 needs_mock_server: bool,
47) -> String {
48 let mut out = String::new();
49 out.push_str(&alef_core::hash::header(alef_core::hash::CommentStyle::DoubleSlash));
50 let _ = writeln!(out, "//! E2e tests for category: {category}");
51 let _ = writeln!(out);
52
53 let module = resolve_module(e2e_config, dep_name);
54 let field_resolver = FieldResolver::new(
55 &e2e_config.fields,
56 &e2e_config.fields_optional,
57 &e2e_config.result_fields,
58 &e2e_config.fields_array,
59 &e2e_config.fields_method_calls,
60 );
61
62 let file_has_http = fixtures.iter().any(|f| f.http.is_some());
64 let file_has_call_based = fixtures.iter().any(|f| {
67 if f.mock_response.is_some() {
68 return true;
69 }
70 if f.http.is_none() && f.mock_response.is_none() {
71 let call_config = e2e_config.resolve_call_for_fixture(f.call.as_deref(), &f.input);
72 let fn_name = resolve_function_name_for_call(call_config);
73 return !fn_name.is_empty();
74 }
75 false
76 });
77
78 let rust_call_override = e2e_config.call.overrides.get("rust");
83 let client_factory = rust_call_override.and_then(|o| o.client_factory.as_deref());
84
85 if file_has_call_based && client_factory.is_none() {
87 let mut imported: std::collections::BTreeSet<(String, String)> = std::collections::BTreeSet::new();
88 for fixture in fixtures.iter().filter(|f| {
89 if f.mock_response.is_some() {
90 return true;
91 }
92 if f.http.is_none() && f.mock_response.is_none() {
93 let call_config = e2e_config.resolve_call_for_fixture(f.call.as_deref(), &f.input);
94 let fn_name = resolve_function_name_for_call(call_config);
95 return !fn_name.is_empty();
96 }
97 false
98 }) {
99 let call_config = e2e_config.resolve_call_for_fixture(fixture.call.as_deref(), &fixture.input);
100 let fn_name = resolve_function_name_for_call(call_config);
101 let mod_name = resolve_module_for_call(call_config, dep_name);
102 imported.insert((mod_name, fn_name));
103 }
104 let mut by_module: std::collections::BTreeMap<String, Vec<String>> = std::collections::BTreeMap::new();
106 for (mod_name, fn_name) in &imported {
107 by_module.entry(mod_name.clone()).or_default().push(fn_name.clone());
108 }
109 for (mod_name, fns) in &by_module {
110 if fns.len() == 1 {
111 let _ = writeln!(out, "use {mod_name}::{};", fns[0]);
112 } else {
113 let joined = fns.join(", ");
114 let _ = writeln!(out, "use {mod_name}::{{{joined}}};");
115 }
116 }
117 }
118
119 if file_has_http {
121 let _ = writeln!(out, "use {module}::{{App, RequestContext}};");
122 }
123
124 let has_handle_args = e2e_config.call.args.iter().any(|a| a.arg_type == "handle");
126 if has_handle_args {
127 let _ = writeln!(out, "use {module}::CrawlConfig;");
128 }
129 for arg in &e2e_config.call.args {
130 if arg.arg_type == "handle" {
131 use heck::ToSnakeCase;
132 let constructor_name = format!("create_{}", arg.name.to_snake_case());
133 let _ = writeln!(out, "use {module}::{constructor_name};");
134 }
135 }
136
137 if client_factory.is_some() && file_has_call_based {
140 let trait_imports: Vec<String> = e2e_config
141 .call
142 .overrides
143 .get("rust")
144 .map(|o| o.trait_imports.clone())
145 .unwrap_or_default();
146 for trait_name in &trait_imports {
147 let _ = writeln!(out, "use {module}::{trait_name};");
148 }
149 }
150
151 let file_needs_mock = needs_mock_server && fixtures.iter().any(|f| f.mock_response.is_some());
153 if file_needs_mock {
154 let _ = writeln!(out, "mod mock_server;");
155 let _ = writeln!(out, "use mock_server::{{MockRoute, MockServer}};");
156 }
157
158 let file_needs_visitor = fixtures.iter().any(|f| f.visitor.is_some());
165 if file_needs_visitor {
166 let visitor_trait = resolve_visitor_trait(rust_call_override).unwrap_or_else(|| {
167 panic!(
168 "category '{}': fixture declares a visitor block but \
169 `[e2e.call.overrides.rust] visitor_trait` is not configured",
170 category
171 )
172 });
173 let _ = writeln!(
174 out,
175 "use {module}::visitor::{{{visitor_trait}, NodeContext, VisitResult}};"
176 );
177 }
178
179 if file_has_call_based {
184 let rust_options_type = e2e_config
185 .call
186 .overrides
187 .get("rust")
188 .and_then(|o| o.options_type.as_deref());
189 if let Some(opts_type) = rust_options_type {
190 let has_json_object_arg = e2e_config.call.args.iter().any(|a| a.arg_type == "json_object");
193 if has_json_object_arg {
194 let _ = writeln!(out, "use {module}::{opts_type};");
195 }
196 }
197 }
198
199 if file_has_call_based {
203 let mut element_types: std::collections::BTreeSet<String> = std::collections::BTreeSet::new();
204 for fixture in fixtures.iter().filter(|f| {
205 if f.mock_response.is_some() {
206 return true;
207 }
208 if f.http.is_none() && f.mock_response.is_none() {
209 let call_config = e2e_config.resolve_call_for_fixture(f.call.as_deref(), &f.input);
210 let fn_name = resolve_function_name_for_call(call_config);
211 return !fn_name.is_empty();
212 }
213 false
214 }) {
215 let call_config = e2e_config.resolve_call_for_fixture(fixture.call.as_deref(), &fixture.input);
216 for arg in &call_config.args {
217 if arg.arg_type == "json_object" {
218 if let Some(ref elem_type) = arg.element_type {
219 element_types.insert(elem_type.clone());
220 }
221 }
222 }
223 }
224 for elem_type in &element_types {
225 let _ = writeln!(out, "use {module}::{elem_type};");
226 }
227 }
228
229 let _ = writeln!(out);
230
231 for fixture in fixtures {
232 render_test_function(&mut out, fixture, e2e_config, dep_name, &field_resolver, client_factory);
233 let _ = writeln!(out);
234 }
235
236 if !out.ends_with('\n') {
237 out.push('\n');
238 }
239 out
240}
241
242pub fn render_test_function(
243 out: &mut String,
244 fixture: &Fixture,
245 e2e_config: &E2eConfig,
246 dep_name: &str,
247 field_resolver: &FieldResolver,
248 client_factory: Option<&str>,
249) {
250 if fixture.http.is_some() {
252 render_http_test_function(out, fixture, dep_name);
253 return;
254 }
255
256 if fixture.http.is_none() && fixture.mock_response.is_none() {
262 let call_config = e2e_config.resolve_call_for_fixture(fixture.call.as_deref(), &fixture.input);
263 let resolved_fn_name = resolve_function_name_for_call(call_config);
264 if resolved_fn_name.is_empty() {
265 let fn_name = crate::escape::sanitize_ident(&fixture.id);
266 let description = &fixture.description;
267 let _ = writeln!(out, "#[tokio::test]");
268 let _ = writeln!(out, "async fn test_{fn_name}() {{");
269 let _ = writeln!(out, " // {description}");
270 let _ = writeln!(
271 out,
272 " // TODO: implement when a callable API is available for this fixture type."
273 );
274 let _ = writeln!(out, "}}");
275 return;
276 }
277 }
279
280 let fn_name = crate::escape::sanitize_ident(&fixture.id);
281 let description = &fixture.description;
282 let call_config = e2e_config.resolve_call_for_fixture(fixture.call.as_deref(), &fixture.input);
283 let function_name = resolve_function_name_for_call(call_config);
284 let module = resolve_module_for_call(call_config, dep_name);
285 let result_var = &call_config.result_var;
286 let has_mock = fixture.mock_response.is_some();
287
288 let rust_overrides = call_config.overrides.get("rust");
290
291 let returns_result = rust_overrides
294 .and_then(|o| o.returns_result)
295 .unwrap_or(if client_factory.is_some() {
296 true
297 } else {
298 call_config.returns_result
299 });
300
301 let is_async = call_config.r#async || has_mock;
303 if is_async {
304 let _ = writeln!(out, "#[tokio::test]");
305 let _ = writeln!(out, "async fn test_{fn_name}() {{");
306 } else {
307 let _ = writeln!(out, "#[test]");
308 let _ = writeln!(out, "fn test_{fn_name}() {{");
309 }
310 let _ = writeln!(out, " // {description}");
311
312 if has_mock {
315 render_mock_server_setup(out, fixture, e2e_config);
316 }
317
318 let has_error_assertion = fixture.assertions.iter().any(|a| a.assertion_type == "error");
320
321 let wrap_options_in_some = rust_overrides.is_some_and(|o| o.wrap_options_in_some);
323 let extra_args: Vec<String> = rust_overrides.map(|o| o.extra_args.clone()).unwrap_or_default();
324 let options_type: Option<String> = rust_overrides.and_then(|o| o.options_type.clone());
328
329 let visitor_via_options = fixture.visitor.is_some() && rust_overrides.is_none_or(|o| o.visitor_function.is_none());
334
335 let mut arg_exprs: Vec<String> = Vec::new();
337 let mut options_arg_name: Option<String> = None;
339 for arg in &call_config.args {
340 let value = crate::codegen::resolve_field(&fixture.input, &arg.field);
341 let var_name = &arg.name;
342 let (mut bindings, expr) = render_rust_arg(
343 var_name,
344 value,
345 &arg.arg_type,
346 arg.optional,
347 &module,
348 &fixture.id,
349 if has_mock {
350 Some("mock_server.url.as_str()")
351 } else {
352 None
353 },
354 arg.owned,
355 arg.element_type.as_deref(),
356 );
357 if arg.arg_type == "json_object" {
361 if let Some(ref opts_type) = options_type {
362 bindings = bindings
363 .into_iter()
364 .map(|b| {
365 let prefix = format!("let {var_name} = ");
367 if b.starts_with(&prefix) {
368 format!("let {var_name}: {opts_type} = {}", &b[prefix.len()..])
369 } else {
370 b
371 }
372 })
373 .collect();
374 }
375 }
376 if visitor_via_options && arg.arg_type == "json_object" {
379 options_arg_name = Some(var_name.clone());
380 bindings = bindings
381 .into_iter()
382 .map(|b| {
383 let prefix = format!("let {var_name}");
385 if b.starts_with(&prefix) {
386 format!("let mut {}", &b[4..])
387 } else {
388 b
389 }
390 })
391 .collect();
392 }
393 for binding in &bindings {
394 let _ = writeln!(out, " {binding}");
395 }
396 let final_expr = if wrap_options_in_some && arg.arg_type == "json_object" {
400 if visitor_via_options {
401 let name = if let Some(rest) = expr.strip_prefix('&') {
404 rest.to_string()
405 } else {
406 expr.clone()
407 };
408 format!("Some({name})")
409 } else if let Some(rest) = expr.strip_prefix('&') {
410 format!("Some({rest}.clone())")
411 } else {
412 format!("Some({expr})")
413 }
414 } else {
415 expr
416 };
417 arg_exprs.push(final_expr);
418 }
419
420 if let Some(visitor_spec) = &fixture.visitor {
422 let visitor_trait = resolve_visitor_trait(rust_overrides)
425 .expect("visitor_trait must be set in [e2e.call.overrides.rust] when a fixture declares a visitor block");
426 let _ = writeln!(out, " #[derive(Debug)]");
428 let _ = writeln!(out, " struct _TestVisitor;");
429 let _ = writeln!(out, " impl {visitor_trait} for _TestVisitor {{");
430 for (method_name, action) in &visitor_spec.callbacks {
431 emit_rust_visitor_method(out, method_name, action);
432 }
433 let _ = writeln!(out, " }}");
434 let _ = writeln!(
435 out,
436 " let visitor = std::rc::Rc::new(std::cell::RefCell::new(_TestVisitor));"
437 );
438 if visitor_via_options {
439 let opts_name = options_arg_name.as_deref().unwrap_or("options");
441 let _ = writeln!(out, " {opts_name}.visitor = Some(visitor);");
442 } else {
443 arg_exprs.push("Some(visitor)".to_string());
445 }
446 } else {
447 arg_exprs.extend(extra_args);
450 }
451
452 let args_str = arg_exprs.join(", ");
453
454 let await_suffix = if is_async { ".await" } else { "" };
455
456 let call_expr = if let Some(factory) = client_factory {
460 let base_url_arg = if has_mock {
461 "Some(mock_server.url.clone())"
462 } else {
463 "None"
464 };
465 let _ = writeln!(
466 out,
467 " let client = {module}::{factory}(\"test-key\".to_string(), {base_url_arg}, None, None, None).unwrap();"
468 );
469 format!("client.{function_name}({args_str})")
470 } else {
471 format!("{function_name}({args_str})")
472 };
473
474 let result_is_tree = call_config.result_var == "tree";
475 let result_is_simple = call_config.result_is_simple || rust_overrides.is_some_and(|o| o.result_is_simple);
479 let result_is_vec = rust_overrides.is_some_and(|o| o.result_is_vec);
482 let result_is_option = call_config.result_is_option || rust_overrides.is_some_and(|o| o.result_is_option);
485
486 if has_error_assertion {
487 let _ = writeln!(out, " let {result_var} = {call_expr}{await_suffix};");
488 let has_non_error_assertions = fixture
490 .assertions
491 .iter()
492 .any(|a| !matches!(a.assertion_type.as_str(), "error" | "not_error"));
493 if returns_result && has_non_error_assertions {
496 let _ = writeln!(out, " let {result_var}_ok = {result_var}.as_ref().ok();");
498 }
499 for assertion in &fixture.assertions {
501 render_assertion(
502 out,
503 assertion,
504 result_var,
505 &module,
506 dep_name,
507 true,
508 &[],
509 field_resolver,
510 result_is_tree,
511 result_is_simple,
512 false,
513 false,
514 returns_result,
515 );
516 }
517 let _ = writeln!(out, "}}");
518 return;
519 }
520
521 let has_not_error = fixture.assertions.iter().any(|a| a.assertion_type == "not_error");
523
524 let has_usable_assertion = fixture.assertions.iter().any(|a| {
528 if a.assertion_type == "not_error" || a.assertion_type == "error" {
529 return false;
530 }
531 if a.assertion_type == "method_result" {
532 let supported_checks = [
535 "equals",
536 "is_true",
537 "is_false",
538 "greater_than_or_equal",
539 "count_min",
540 "is_error",
541 "contains",
542 "not_empty",
543 "is_empty",
544 ];
545 let check = a.check.as_deref().unwrap_or("is_true");
546 if a.method.is_none() || !supported_checks.contains(&check) {
547 return false;
548 }
549 }
550 match &a.field {
551 Some(f) if !f.is_empty() => field_resolver.is_valid_for_result(f),
552 _ => true,
553 }
554 });
555
556 let result_binding = if has_usable_assertion {
557 result_var.to_string()
558 } else {
559 "_".to_string()
560 };
561
562 let has_field_access = fixture
566 .assertions
567 .iter()
568 .any(|a| a.field.as_ref().is_some_and(|f| !f.is_empty()));
569 let only_emptiness_checks = !has_field_access
570 && fixture.assertions.iter().all(|a| {
571 matches!(
572 a.assertion_type.as_str(),
573 "is_empty" | "is_false" | "not_empty" | "is_true" | "not_error"
574 )
575 });
576
577 let unwrap_suffix = if returns_result {
578 ".expect(\"should succeed\")"
579 } else {
580 ""
581 };
582 if !returns_result || (only_emptiness_checks && !has_not_error) {
583 let _ = writeln!(out, " let {result_binding} = {call_expr}{await_suffix};");
586 } else if has_not_error || !fixture.assertions.is_empty() {
587 let _ = writeln!(
588 out,
589 " let {result_binding} = {call_expr}{await_suffix}{unwrap_suffix};"
590 );
591 } else {
592 let _ = writeln!(out, " let {result_binding} = {call_expr}{await_suffix};");
593 }
594
595 let string_assertion_types = [
601 "equals",
602 "contains",
603 "contains_all",
604 "contains_any",
605 "not_contains",
606 "starts_with",
607 "ends_with",
608 "min_length",
609 "max_length",
610 "matches_regex",
611 ];
612 let mut unwrapped_fields: Vec<(String, String)> = Vec::new(); if !result_is_vec {
614 for assertion in &fixture.assertions {
615 if let Some(f) = &assertion.field {
616 if !f.is_empty()
617 && string_assertion_types.contains(&assertion.assertion_type.as_str())
618 && !unwrapped_fields.iter().any(|(ff, _)| ff == f)
619 {
620 let is_string_assertion = assertion.value.as_ref().is_none_or(|v| v.is_string());
623 if !is_string_assertion {
624 continue;
625 }
626 if let Some((binding, local_var)) = field_resolver.rust_unwrap_binding(f, result_var) {
627 let _ = writeln!(out, " {binding}");
628 unwrapped_fields.push((f.clone(), local_var));
629 }
630 }
631 }
632 }
633 }
634
635 for assertion in &fixture.assertions {
637 if assertion.assertion_type == "not_error" {
638 continue;
640 }
641 render_assertion(
642 out,
643 assertion,
644 result_var,
645 &module,
646 dep_name,
647 false,
648 &unwrapped_fields,
649 field_resolver,
650 result_is_tree,
651 result_is_simple,
652 result_is_vec,
653 result_is_option,
654 returns_result,
655 );
656 }
657
658 let _ = writeln!(out, "}}");
659}
660
661pub fn collect_test_filenames(groups: &[FixtureGroup]) -> Vec<String> {
663 groups
664 .iter()
665 .filter(|g| !g.fixtures.is_empty())
666 .map(|g| format!("{}_test.rs", sanitize_filename(&g.category)))
667 .collect()
668}
669
670#[cfg(test)]
671mod tests {
672 use super::*;
673
674 #[test]
675 fn resolve_module_for_call_prefers_crate_name_override() {
676 use crate::config::CallConfig;
677 use std::collections::HashMap;
678 let mut overrides = HashMap::new();
679 overrides.insert(
680 "rust".to_string(),
681 crate::config::CallOverride {
682 crate_name: Some("custom_crate".to_string()),
683 module: Some("ignored_module".to_string()),
684 ..Default::default()
685 },
686 );
687 let call = CallConfig {
688 overrides,
689 ..Default::default()
690 };
691 let result = resolve_module_for_call(&call, "dep_name");
692 assert_eq!(result, "custom_crate");
693 }
694}