1use crate::config::E2eConfig;
4use crate::escape::{escape_elixir, sanitize_filename, sanitize_ident};
5use crate::field_access::FieldResolver;
6use crate::fixture::{Assertion, CallbackAction, Fixture, FixtureGroup, HttpFixture, ValidationErrorExpectation};
7use alef_core::backend::GeneratedFile;
8use alef_core::config::ResolvedCrateConfig;
9use alef_core::hash::{self, CommentStyle};
10use alef_core::template_versions as tv;
11use anyhow::Result;
12use heck::ToSnakeCase;
13use std::collections::HashMap;
14use std::fmt::Write as FmtWrite;
15use std::path::PathBuf;
16
17use super::E2eCodegen;
18use super::client;
19
20pub struct ElixirCodegen;
22
23impl E2eCodegen for ElixirCodegen {
24 fn generate(
25 &self,
26 groups: &[FixtureGroup],
27 e2e_config: &E2eConfig,
28 config: &ResolvedCrateConfig,
29 _type_defs: &[alef_core::ir::TypeDef],
30 _enums: &[alef_core::ir::EnumDef],
31 ) -> Result<Vec<GeneratedFile>> {
32 let lang = self.language_name();
33 let output_base = PathBuf::from(e2e_config.effective_output()).join(lang);
34
35 let mut files = Vec::new();
36
37 let call = &e2e_config.call;
39 let overrides = call.overrides.get(lang);
40 let raw_module = overrides
41 .and_then(|o| o.module.as_ref())
42 .cloned()
43 .unwrap_or_else(|| call.module.clone());
44 let module_path = if raw_module.contains('.') || raw_module.chars().next().is_some_and(|c| c.is_uppercase()) {
48 raw_module.clone()
49 } else {
50 elixir_module_name(&raw_module)
51 };
52 let base_function_name = overrides
53 .and_then(|o| o.function.as_ref())
54 .cloned()
55 .unwrap_or_else(|| call.function.clone());
56 let function_name =
61 if call.r#async && !base_function_name.ends_with("_async") && !base_function_name.ends_with("_stream") {
62 format!("{base_function_name}_async")
63 } else {
64 base_function_name
65 };
66 let options_type = overrides.and_then(|o| o.options_type.clone());
67 let options_default_fn = overrides.and_then(|o| o.options_via.clone());
68 let empty_enum_fields = HashMap::new();
69 let enum_fields = overrides.map(|o| &o.enum_fields).unwrap_or(&empty_enum_fields);
70 let handle_struct_type = overrides.and_then(|o| o.handle_struct_type.clone());
71 let empty_atom_fields = std::collections::HashSet::new();
72 let handle_atom_list_fields = overrides
73 .map(|o| &o.handle_atom_list_fields)
74 .unwrap_or(&empty_atom_fields);
75 let result_var = &call.result_var;
76
77 let has_http_tests = groups.iter().any(|g| g.fixtures.iter().any(|f| f.is_http_test()));
79 let has_nif_tests = groups.iter().any(|g| g.fixtures.iter().any(|f| !f.is_http_test()));
80 let has_mock_server_tests = groups.iter().any(|g| {
82 g.fixtures.iter().any(|f| {
83 if f.needs_mock_server() {
84 return true;
85 }
86 let cc = e2e_config.resolve_call_for_fixture(
87 f.call.as_deref(),
88 &f.id,
89 &f.resolved_category(),
90 &f.tags,
91 &f.input,
92 );
93 let elixir_override = cc
94 .overrides
95 .get("elixir")
96 .or_else(|| e2e_config.call.overrides.get("elixir"));
97 elixir_override.and_then(|o| o.client_factory.as_deref()).is_some()
98 })
99 });
100
101 let pkg_ref = e2e_config.resolve_package(lang);
103 let pkg_dep_ref = if has_nif_tests {
104 match e2e_config.dep_mode {
105 crate::config::DependencyMode::Local => pkg_ref
106 .as_ref()
107 .and_then(|p| p.path.as_deref())
108 .unwrap_or("../../packages/elixir")
109 .to_string(),
110 crate::config::DependencyMode::Registry => pkg_ref
111 .as_ref()
112 .and_then(|p| p.version.clone())
113 .or_else(|| config.resolved_version())
114 .unwrap_or_else(|| "0.1.0".to_string()),
115 }
116 } else {
117 String::new()
118 };
119
120 let pkg_atom = config.elixir_app_name();
129 files.push(GeneratedFile {
130 path: output_base.join("mix.exs"),
131 content: render_mix_exs(
132 &pkg_atom,
133 &pkg_dep_ref,
134 e2e_config.dep_mode,
135 has_http_tests,
136 has_nif_tests,
137 ),
138 generated_header: false,
139 });
140
141 files.push(GeneratedFile {
143 path: output_base.join("lib").join("e2e_elixir.ex"),
144 content: "defmodule E2eElixir do\n @moduledoc false\nend\n".to_string(),
145 generated_header: false,
146 });
147
148 files.push(GeneratedFile {
150 path: output_base.join("test").join("test_helper.exs"),
151 content: render_test_helper(has_http_tests || has_mock_server_tests),
152 generated_header: false,
153 });
154
155 for group in groups {
157 let active: Vec<&Fixture> = group
158 .fixtures
159 .iter()
160 .filter(|f| super::should_include_fixture(f, lang, e2e_config))
161 .collect();
162
163 if active.is_empty() {
164 continue;
165 }
166
167 let filename = format!("{}_test.exs", sanitize_filename(&group.category));
168 let content = render_test_file(
169 &group.category,
170 &active,
171 e2e_config,
172 &module_path,
173 &function_name,
174 result_var,
175 &e2e_config.call.args,
176 options_type.as_deref(),
177 options_default_fn.as_deref(),
178 enum_fields,
179 handle_struct_type.as_deref(),
180 handle_atom_list_fields,
181 &config.adapters,
182 );
183 files.push(GeneratedFile {
184 path: output_base.join("test").join(filename),
185 content,
186 generated_header: true,
187 });
188 }
189
190 Ok(files)
191 }
192
193 fn language_name(&self) -> &'static str {
194 "elixir"
195 }
196}
197
198fn render_test_helper(has_http_tests: bool) -> String {
199 if has_http_tests {
200 r#"ExUnit.start()
201
202# Spawn mock-server binary and set MOCK_SERVER_URL for all tests.
203mock_server_bin = Path.expand("../../rust/target/release/mock-server", __DIR__)
204fixtures_dir = Path.expand("../../../fixtures", __DIR__)
205
206if File.exists?(mock_server_bin) do
207 port = Port.open({:spawn_executable, mock_server_bin}, [
208 :binary,
209 # Use a large line buffer (default 1024 truncates `MOCK_SERVERS={...}` lines for
210 # fixture sets with many host-root routes, splitting them into `:noeol` chunks
211 # that the prefix-match clauses below would never see).
212 {:line, 65_536},
213 args: [fixtures_dir]
214 ])
215 # Read startup lines: MOCK_SERVER_URL= then MOCK_SERVERS= (always emitted, possibly `{}`).
216 # The standalone mock-server prints noisy stderr lines BEFORE the stdout sentinels;
217 # selective receive ignores anything that doesn't match the two prefix patterns.
218 # Each iteration only halts after the MOCK_SERVERS= line is processed.
219 {url, _} =
220 Enum.reduce_while(1..16, {nil, port}, fn _, {url_acc, p} ->
221 receive do
222 {^p, {:data, {:eol, "MOCK_SERVER_URL=" <> u}}} ->
223 {:cont, {u, p}}
224
225 {^p, {:data, {:eol, "MOCK_SERVERS=" <> json_val}}} ->
226 System.put_env("MOCK_SERVERS", json_val)
227 case Jason.decode(json_val) do
228 {:ok, servers} ->
229 Enum.each(servers, fn {fid, furl} ->
230 System.put_env("MOCK_SERVER_#{String.upcase(fid)}", furl)
231 end)
232
233 _ ->
234 :ok
235 end
236
237 {:halt, {url_acc, p}}
238 after
239 30_000 ->
240 raise "mock-server startup timeout"
241 end
242 end)
243
244 if url != nil do
245 System.put_env("MOCK_SERVER_URL", url)
246 end
247end
248"#
249 .to_string()
250 } else {
251 "ExUnit.start()\n".to_string()
252 }
253}
254
255fn render_mix_exs(
256 pkg_name: &str,
257 pkg_path: &str,
258 dep_mode: crate::config::DependencyMode,
259 has_http_tests: bool,
260 has_nif_tests: bool,
261) -> String {
262 let mut out = String::new();
263 let _ = writeln!(out, "defmodule E2eElixir.MixProject do");
264 let _ = writeln!(out, " use Mix.Project");
265 let _ = writeln!(out);
266 let _ = writeln!(out, " def project do");
267 let _ = writeln!(out, " [");
268 let _ = writeln!(out, " app: :e2e_elixir,");
269 let _ = writeln!(out, " version: \"0.1.0\",");
270 let _ = writeln!(out, " elixir: \"~> 1.14\",");
271 let _ = writeln!(out, " deps: deps()");
272 let _ = writeln!(out, " ]");
273 let _ = writeln!(out, " end");
274 let _ = writeln!(out);
275 let _ = writeln!(out, " defp deps do");
276 let _ = writeln!(out, " [");
277
278 let mut deps: Vec<String> = Vec::new();
280
281 if has_nif_tests && !pkg_path.is_empty() {
283 let pkg_atom = pkg_name;
284 let nif_dep = match dep_mode {
285 crate::config::DependencyMode::Local => {
286 format!(" {{:{pkg_atom}, path: \"{pkg_path}\"}}")
287 }
288 crate::config::DependencyMode::Registry => {
289 format!(" {{:{pkg_atom}, \"{pkg_path}\"}}")
291 }
292 };
293 deps.push(nif_dep);
294 deps.push(format!(
296 " {{:rustler_precompiled, \"{rp}\"}}",
297 rp = tv::hex::RUSTLER_PRECOMPILED
298 ));
299 deps.push(format!(
304 " {{:rustler, \"{rustler}\", runtime: false}}",
305 rustler = tv::hex::RUSTLER
306 ));
307 }
308
309 if has_http_tests {
311 deps.push(format!(" {{:req, \"{req}\"}}", req = tv::hex::REQ));
312 deps.push(format!(" {{:jason, \"{jason}\"}}", jason = tv::hex::JASON));
313 }
314
315 let _ = writeln!(out, "{}", deps.join(",\n"));
316 let _ = writeln!(out, " ]");
317 let _ = writeln!(out, " end");
318 let _ = writeln!(out, "end");
319 out
320}
321
322#[allow(clippy::too_many_arguments)]
323fn render_test_file(
324 category: &str,
325 fixtures: &[&Fixture],
326 e2e_config: &E2eConfig,
327 module_path: &str,
328 function_name: &str,
329 result_var: &str,
330 args: &[crate::config::ArgMapping],
331 options_type: Option<&str>,
332 options_default_fn: Option<&str>,
333 enum_fields: &HashMap<String, String>,
334 handle_struct_type: Option<&str>,
335 handle_atom_list_fields: &std::collections::HashSet<String>,
336 adapters: &[alef_core::config::extras::AdapterConfig],
337) -> String {
338 let mut out = String::new();
339 out.push_str(&hash::header(CommentStyle::Hash));
340 let _ = writeln!(out, "# E2e tests for category: {category}");
341 let _ = writeln!(out, "defmodule E2e.{}Test do", elixir_module_name(category));
342
343 let has_http = fixtures.iter().any(|f| f.is_http_test());
345
346 let async_flag = if has_http { "true" } else { "false" };
349 let _ = writeln!(out, " use ExUnit.Case, async: {async_flag}");
350
351 if has_http {
352 let _ = writeln!(out);
353 let _ = writeln!(out, " defp mock_server_url do");
354 let _ = writeln!(
355 out,
356 " System.get_env(\"MOCK_SERVER_URL\") || \"http://localhost:8080\""
357 );
358 let _ = writeln!(out, " end");
359 }
360
361 let has_array_contains = fixtures.iter().any(|fixture| {
364 let cc = e2e_config.resolve_call_for_fixture(
365 fixture.call.as_deref(),
366 &fixture.id,
367 &fixture.resolved_category(),
368 &fixture.tags,
369 &fixture.input,
370 );
371 let fr = FieldResolver::new(
372 e2e_config.effective_fields(cc),
373 e2e_config.effective_fields_optional(cc),
374 e2e_config.effective_result_fields(cc),
375 e2e_config.effective_fields_array(cc),
376 &std::collections::HashSet::new(),
377 );
378 fixture.assertions.iter().any(|a| {
379 matches!(a.assertion_type.as_str(), "contains" | "contains_all" | "not_contains")
380 && a.field
381 .as_deref()
382 .is_some_and(|f| !f.is_empty() && fr.is_array(fr.resolve(f)))
383 })
384 });
385 if has_array_contains {
386 let _ = writeln!(out);
387 let _ = writeln!(out, " defp alef_e2e_item_texts(item) when is_binary(item), do: [item]");
388 let _ = writeln!(out, " defp alef_e2e_item_texts(item) do");
389 let _ = writeln!(out, " [:kind, :name, :signature, :path, :alias, :text, :source]");
390 let _ = writeln!(out, " |> Enum.filter(&Map.has_key?(item, &1))");
391 let _ = writeln!(out, " |> Enum.flat_map(fn attr ->");
392 let _ = writeln!(out, " case Map.get(item, attr) do");
393 let _ = writeln!(out, " nil -> []");
394 let _ = writeln!(
395 out,
396 " atom when is_atom(atom) -> [atom |> to_string() |> String.capitalize()]"
397 );
398 let _ = writeln!(out, " str -> [to_string(str)]");
399 let _ = writeln!(out, " end");
400 let _ = writeln!(out, " end)");
401 let _ = writeln!(out, " end");
402 }
403
404 let _ = writeln!(out);
405
406 for (i, fixture) in fixtures.iter().enumerate() {
407 if let Some(http) = &fixture.http {
408 render_http_test_case(&mut out, fixture, http);
409 } else {
410 render_test_case(
411 &mut out,
412 fixture,
413 e2e_config,
414 module_path,
415 function_name,
416 result_var,
417 args,
418 options_type,
419 options_default_fn,
420 enum_fields,
421 handle_struct_type,
422 handle_atom_list_fields,
423 adapters,
424 );
425 }
426 if i + 1 < fixtures.len() {
427 let _ = writeln!(out);
428 }
429 }
430
431 let _ = writeln!(out, "end");
432 out
433}
434
435const FINCH_UNSUPPORTED_METHODS: &[&str] = &["TRACE", "CONNECT"];
442
443const REQ_CONVENIENCE_METHODS: &[&str] = &["get", "post", "put", "patch", "delete", "head"];
446
447struct ElixirTestClientRenderer<'a> {
451 fixture_id: &'a str,
454 expected_status: u16,
456}
457
458impl<'a> client::TestClientRenderer for ElixirTestClientRenderer<'a> {
459 fn language_name(&self) -> &'static str {
460 "elixir"
461 }
462
463 fn render_test_open(&self, out: &mut String, fn_name: &str, description: &str, skip_reason: Option<&str>) {
469 let escaped_description = description.replace('"', "\\\"");
470 let _ = writeln!(out, " describe \"{fn_name}\" do");
471 if skip_reason.is_some() {
472 let _ = writeln!(out, " @tag :skip");
473 }
474 let _ = writeln!(out, " test \"{escaped_description}\" do");
475 }
476
477 fn render_test_close(&self, out: &mut String) {
479 let _ = writeln!(out, " end");
480 let _ = writeln!(out, " end");
481 }
482
483 fn render_call(&self, out: &mut String, ctx: &client::CallCtx<'_>) {
485 let method = ctx.method.to_lowercase();
486 let mut opts: Vec<String> = Vec::new();
487
488 if let Some(body) = ctx.body {
489 let elixir_val = json_to_elixir(body);
490 opts.push(format!("json: {elixir_val}"));
491 }
492
493 if !ctx.headers.is_empty() {
494 let header_pairs: Vec<String> = ctx
495 .headers
496 .iter()
497 .map(|(k, v)| format!("{{\"{}\", \"{}\"}}", escape_elixir(k), escape_elixir(v)))
498 .collect();
499 opts.push(format!("headers: [{}]", header_pairs.join(", ")));
500 }
501
502 if !ctx.cookies.is_empty() {
503 let cookie_str = ctx
504 .cookies
505 .iter()
506 .map(|(k, v)| format!("{k}={v}"))
507 .collect::<Vec<_>>()
508 .join("; ");
509 opts.push(format!("headers: [{{\"cookie\", \"{}\"}}]", escape_elixir(&cookie_str)));
510 }
511
512 if !ctx.query_params.is_empty() {
513 let pairs: Vec<String> = ctx
514 .query_params
515 .iter()
516 .map(|(k, v)| {
517 let val_str = match v {
518 serde_json::Value::String(s) => s.clone(),
519 other => other.to_string(),
520 };
521 format!("{{\"{}\", \"{}\"}}", escape_elixir(k), escape_elixir(&val_str))
522 })
523 .collect();
524 opts.push(format!("params: [{}]", pairs.join(", ")));
525 }
526
527 if (300..400).contains(&self.expected_status) {
530 opts.push("redirect: false".to_string());
531 }
532
533 let fixture_id = escape_elixir(self.fixture_id);
534 let url_expr = format!("\"#{{mock_server_url()}}/fixtures/{fixture_id}\"");
535
536 if REQ_CONVENIENCE_METHODS.contains(&method.as_str()) {
537 if opts.is_empty() {
538 let _ = writeln!(out, " {{:ok, response}} = Req.{method}(url: {url_expr})");
539 } else {
540 let opts_str = opts.join(", ");
541 let _ = writeln!(
542 out,
543 " {{:ok, response}} = Req.{method}(url: {url_expr}, {opts_str})"
544 );
545 }
546 } else {
547 opts.insert(0, format!("method: :{method}"));
548 opts.insert(1, format!("url: {url_expr}"));
549 let opts_str = opts.join(", ");
550 let _ = writeln!(out, " {{:ok, response}} = Req.request({opts_str})");
551 }
552 }
553
554 fn render_assert_status(&self, out: &mut String, response_var: &str, status: u16) {
555 let _ = writeln!(out, " assert {response_var}.status == {status}");
556 }
557
558 fn render_assert_header(&self, out: &mut String, response_var: &str, name: &str, expected: &str) {
563 let header_key = name.to_lowercase();
564 if header_key == "connection" {
566 return;
567 }
568 let key_lit = format!("\"{}\"", escape_elixir(&header_key));
569 let get_header_expr = format!(
570 "Enum.find_value({response_var}.headers, fn {{k, v}} -> if String.downcase(k) == {key_lit}, do: List.first(List.wrap(v)) end)"
571 );
572 match expected {
573 "<<present>>" => {
574 let _ = writeln!(out, " assert {get_header_expr} != nil");
575 }
576 "<<absent>>" => {
577 let _ = writeln!(out, " assert {get_header_expr} == nil");
578 }
579 "<<uuid>>" => {
580 let var = sanitize_ident(&header_key);
581 let _ = writeln!(out, " header_val_{var} = {get_header_expr}");
582 let _ = writeln!(
583 out,
584 " assert Regex.match?(~r/^[0-9a-f]{{8}}-[0-9a-f]{{4}}-[0-9a-f]{{4}}-[0-9a-f]{{4}}-[0-9a-f]{{12}}$/i, to_string(header_val_{var}))"
585 );
586 }
587 literal => {
588 let val_lit = format!("\"{}\"", escape_elixir(literal));
589 let _ = writeln!(out, " assert {get_header_expr} == {val_lit}");
590 }
591 }
592 }
593
594 fn render_assert_json_body(&self, out: &mut String, response_var: &str, expected: &serde_json::Value) {
599 let elixir_val = json_to_elixir(expected);
600 match expected {
601 serde_json::Value::Object(_) | serde_json::Value::Array(_) => {
602 let _ = writeln!(
603 out,
604 " body_decoded = if is_binary({response_var}.body), do: Jason.decode!({response_var}.body), else: {response_var}.body"
605 );
606 let _ = writeln!(out, " assert body_decoded == {elixir_val}");
607 }
608 _ => {
609 let _ = writeln!(out, " assert {response_var}.body == {elixir_val}");
610 }
611 }
612 }
613
614 fn render_assert_partial_body(&self, out: &mut String, response_var: &str, expected: &serde_json::Value) {
616 if let Some(obj) = expected.as_object() {
617 let _ = writeln!(
618 out,
619 " decoded_body = if is_binary({response_var}.body), do: Jason.decode!({response_var}.body), else: {response_var}.body"
620 );
621 for (key, val) in obj {
622 let key_lit = format!("\"{}\"", escape_elixir(key));
623 let elixir_val = json_to_elixir(val);
624 let _ = writeln!(out, " assert decoded_body[{key_lit}] == {elixir_val}");
625 }
626 }
627 }
628
629 fn render_assert_validation_errors(
632 &self,
633 out: &mut String,
634 response_var: &str,
635 errors: &[ValidationErrorExpectation],
636 ) {
637 for err in errors {
638 let msg_lit = format!("\"{}\"", escape_elixir(&err.msg));
639 let _ = writeln!(
640 out,
641 " assert String.contains?(Jason.encode!({response_var}.body), {msg_lit})"
642 );
643 }
644 }
645}
646
647fn render_http_test_case(out: &mut String, fixture: &Fixture, http: &HttpFixture) {
653 let method = http.request.method.to_uppercase();
654
655 if FINCH_UNSUPPORTED_METHODS.contains(&method.as_str()) {
659 let test_name = sanitize_ident(&fixture.id);
660 let test_label = fixture.id.replace('"', "\\\"");
661 let path = &http.request.path;
662 let _ = writeln!(out, " describe \"{test_name}\" do");
663 let _ = writeln!(out, " @tag :skip");
664 let _ = writeln!(out, " test \"{method} {path} - {test_label}\" do");
665 let _ = writeln!(out, " end");
666 let _ = writeln!(out, " end");
667 return;
668 }
669
670 let renderer = ElixirTestClientRenderer {
671 fixture_id: &fixture.id,
672 expected_status: http.expected_response.status_code,
673 };
674 client::http_call::render_http_test(out, &renderer, fixture);
675}
676
677#[allow(clippy::too_many_arguments)]
682fn render_test_case(
683 out: &mut String,
684 fixture: &Fixture,
685 e2e_config: &E2eConfig,
686 default_module_path: &str,
687 default_function_name: &str,
688 default_result_var: &str,
689 args: &[crate::config::ArgMapping],
690 options_type: Option<&str>,
691 options_default_fn: Option<&str>,
692 enum_fields: &HashMap<String, String>,
693 handle_struct_type: Option<&str>,
694 handle_atom_list_fields: &std::collections::HashSet<String>,
695 adapters: &[alef_core::config::extras::AdapterConfig],
696) {
697 let test_name = sanitize_ident(&fixture.id);
698 let test_label = fixture.id.replace('"', "\\\"");
699
700 if fixture.mock_response.is_none() && !fixture_has_elixir_callable(fixture, e2e_config) {
706 let _ = writeln!(out, " describe \"{test_name}\" do");
707 let _ = writeln!(out, " @tag :skip");
708 let _ = writeln!(out, " test \"{test_label}\" do");
709 let _ = writeln!(
710 out,
711 " # non-HTTP fixture: Elixir binding does not expose a callable for the configured `[e2e.call]` function"
712 );
713 let _ = writeln!(out, " :ok");
714 let _ = writeln!(out, " end");
715 let _ = writeln!(out, " end");
716 return;
717 }
718
719 let call_config = e2e_config.resolve_call_for_fixture(
721 fixture.call.as_deref(),
722 &fixture.id,
723 &fixture.resolved_category(),
724 &fixture.tags,
725 &fixture.input,
726 );
727 let call_field_resolver = FieldResolver::new(
729 e2e_config.effective_fields(call_config),
730 e2e_config.effective_fields_optional(call_config),
731 e2e_config.effective_result_fields(call_config),
732 e2e_config.effective_fields_array(call_config),
733 &std::collections::HashSet::new(),
734 );
735 let field_resolver = &call_field_resolver;
736 let lang = "elixir";
737 let call_overrides = call_config.overrides.get(lang);
738
739 let base_fn = call_overrides
742 .and_then(|o| o.function.as_ref())
743 .cloned()
744 .unwrap_or_else(|| call_config.function.clone());
745 if base_fn.starts_with("batch_extract_") {
746 let _ = writeln!(
747 out,
748 " describe \"{test_name}\" do",
749 test_name = sanitize_ident(&fixture.id)
750 );
751 let _ = writeln!(out, " @tag :skip");
752 let _ = writeln!(
753 out,
754 " test \"{test_label}\" do",
755 test_label = fixture.id.replace('"', "\\\"")
756 );
757 let _ = writeln!(
758 out,
759 " # batch functions excluded from Elixir binding: unsafe NIF tuple marshalling"
760 );
761 let _ = writeln!(out, " :ok");
762 let _ = writeln!(out, " end");
763 let _ = writeln!(out, " end");
764 return;
765 }
766
767 let (module_path, function_name, result_var) = if fixture.call.is_some() {
770 let raw_module = call_overrides
771 .and_then(|o| o.module.as_ref())
772 .cloned()
773 .unwrap_or_else(|| call_config.module.clone());
774 let resolved_module = if raw_module.contains('.') || raw_module.chars().next().is_some_and(|c| c.is_uppercase())
775 {
776 raw_module.clone()
777 } else {
778 elixir_module_name(&raw_module)
779 };
780 let resolved_fn = if call_config.r#async && !base_fn.ends_with("_async") && !base_fn.ends_with("_stream") {
781 format!("{base_fn}_async")
782 } else {
783 base_fn
784 };
785 (resolved_module, resolved_fn, call_config.result_var.clone())
786 } else {
787 (
788 default_module_path.to_string(),
789 default_function_name.to_string(),
790 default_result_var.to_string(),
791 )
792 };
793
794 let expects_error = fixture.assertions.iter().any(|a| a.assertion_type == "error");
795 let validation_creation_failure = expects_error && fixture.resolved_category() == "validation";
799
800 let (
802 effective_args,
803 effective_options_type,
804 effective_options_default_fn,
805 effective_enum_fields,
806 effective_handle_struct_type,
807 effective_handle_atom_list_fields,
808 );
809 let empty_enum_fields_local: HashMap<String, String>;
810 let empty_atom_fields_local: std::collections::HashSet<String>;
811 let (
812 resolved_args,
813 resolved_options_type,
814 resolved_options_default_fn,
815 resolved_enum_fields_ref,
816 resolved_handle_struct_type,
817 resolved_handle_atom_list_fields_ref,
818 ) = if fixture.call.is_some() {
819 let co = call_config.overrides.get(lang);
820 effective_args = call_config.args.as_slice();
821 effective_options_type = co.and_then(|o| o.options_type.as_deref());
822 effective_options_default_fn = co.and_then(|o| o.options_via.as_deref());
823 empty_enum_fields_local = HashMap::new();
824 effective_enum_fields = co.map(|o| &o.enum_fields).unwrap_or(&empty_enum_fields_local);
825 effective_handle_struct_type = co.and_then(|o| o.handle_struct_type.as_deref());
826 empty_atom_fields_local = std::collections::HashSet::new();
827 effective_handle_atom_list_fields = co
828 .map(|o| &o.handle_atom_list_fields)
829 .unwrap_or(&empty_atom_fields_local);
830 (
831 effective_args,
832 effective_options_type,
833 effective_options_default_fn,
834 effective_enum_fields,
835 effective_handle_struct_type,
836 effective_handle_atom_list_fields,
837 )
838 } else {
839 (
840 args as &[_],
841 options_type,
842 options_default_fn,
843 enum_fields,
844 handle_struct_type,
845 handle_atom_list_fields,
846 )
847 };
848
849 let test_documents_path = e2e_config.test_documents_relative_from(0);
850 let adapter_request_type: Option<String> = adapters
851 .iter()
852 .find(|a| a.name == call_config.function.as_str())
853 .and_then(|a| a.request_type.as_deref())
854 .map(|rt| rt.rsplit("::").next().unwrap_or(rt).to_string());
855 let (mut setup_lines, args_str) = build_args_and_setup(
856 &fixture.input,
857 resolved_args,
858 &module_path,
859 resolved_options_type,
860 resolved_options_default_fn,
861 resolved_enum_fields_ref,
862 fixture,
863 resolved_handle_struct_type,
864 resolved_handle_atom_list_fields_ref,
865 &test_documents_path,
866 adapter_request_type.as_deref(),
867 );
868
869 let visitor_var = fixture
871 .visitor
872 .as_ref()
873 .map(|visitor_spec| build_elixir_visitor(&mut setup_lines, visitor_spec));
874
875 let final_args = if let Some(ref visitor_var) = visitor_var {
878 let parts: Vec<&str> = args_str.split(", ").collect();
882 if parts.len() == 2 && parts[1] == "nil" {
883 format!("{}, %{{visitor: {}}}", parts[0], visitor_var)
885 } else if parts.len() == 2 {
886 setup_lines.push(format!(
888 "{} = Map.put({}, :visitor, {})",
889 parts[1], parts[1], visitor_var
890 ));
891 args_str
892 } else if parts.len() == 1 {
893 format!("{}, %{{visitor: {}}}", parts[0], visitor_var)
895 } else {
896 args_str
897 }
898 } else {
899 args_str
900 };
901
902 let client_factory = call_overrides.and_then(|o| o.client_factory.as_deref()).or_else(|| {
904 e2e_config
905 .call
906 .overrides
907 .get("elixir")
908 .and_then(|o| o.client_factory.as_deref())
909 });
910
911 let extra_args: Vec<String> = call_overrides.map(|o| o.extra_args.clone()).unwrap_or_default();
915 let final_args_with_extras = if extra_args.is_empty() {
916 final_args
917 } else if final_args.is_empty() {
918 extra_args.join(", ")
919 } else {
920 format!("{final_args}, {}", extra_args.join(", "))
921 };
922
923 let effective_args = if client_factory.is_some() {
925 if final_args_with_extras.is_empty() {
926 "client".to_string()
927 } else {
928 format!("client, {final_args_with_extras}")
929 }
930 } else {
931 final_args_with_extras
932 };
933
934 let has_mock = fixture.mock_response.is_some() || fixture.http.is_some();
938 let api_key_var_opt = fixture.env.as_ref().and_then(|e| e.api_key_var.as_deref());
939 let needs_api_key_skip = !has_mock && api_key_var_opt.is_some();
940 let needs_env_fallback = has_mock && api_key_var_opt.is_some();
943
944 let _ = writeln!(out, " describe \"{test_name}\" do");
945 let _ = writeln!(out, " test \"{test_label}\" do");
946
947 if needs_api_key_skip {
948 let api_key_var = api_key_var_opt.unwrap_or("");
949 let _ = writeln!(out, " if System.get_env(\"{api_key_var}\") in [nil, \"\"] do");
950 let _ = writeln!(out, " # {api_key_var} not set — skipping live smoke test");
951 let _ = writeln!(out, " :ok");
952 let _ = writeln!(out, " else");
953 }
954
955 if validation_creation_failure {
959 let mut emitted_error_assertion = false;
960 for line in &setup_lines {
961 if !emitted_error_assertion && line.starts_with("{:ok,") {
962 if let Some(rhs) = line.split_once('=').map(|x| x.1) {
963 let rhs = rhs.trim();
964 let _ = writeln!(out, " assert {{:error, _}} = {rhs}");
965 emitted_error_assertion = true;
966 } else {
967 let _ = writeln!(out, " {line}");
968 }
969 } else {
970 let _ = writeln!(out, " {line}");
971 }
972 }
973 if !emitted_error_assertion {
974 let _ = writeln!(
975 out,
976 " assert {{:error, _}} = {module_path}.{function_name}({effective_args})"
977 );
978 }
979 if needs_api_key_skip {
980 let _ = writeln!(out, " end");
981 }
982 let _ = writeln!(out, " end");
983 let _ = writeln!(out, " end");
984 return;
985 }
986
987 if expects_error {
994 for line in &setup_lines {
995 let _ = writeln!(out, " {line}");
996 }
997 if let Some(factory) = client_factory {
998 let fixture_id = &fixture.id;
999 let base_url_expr = if fixture.has_host_root_route() {
1000 let env_key = format!("MOCK_SERVER_{}", fixture_id.to_uppercase());
1001 format!(
1002 "(System.get_env(\"{env_key}\") || (System.get_env(\"MOCK_SERVER_URL\") || \"\") <> \"/fixtures/{fixture_id}\")"
1003 )
1004 } else {
1005 format!("(System.get_env(\"MOCK_SERVER_URL\") || \"\") <> \"/fixtures/{fixture_id}\"")
1006 };
1007 let _ = writeln!(
1008 out,
1009 " {{:ok, client}} = {module_path}.{factory}(\"test-key\", base_url: {base_url_expr})"
1010 );
1011 }
1012 let _ = writeln!(
1013 out,
1014 " assert {{:error, _}} = {module_path}.{function_name}({effective_args})"
1015 );
1016 if needs_api_key_skip {
1017 let _ = writeln!(out, " end");
1018 }
1019 let _ = writeln!(out, " end");
1020 let _ = writeln!(out, " end");
1021 return;
1022 }
1023
1024 for line in &setup_lines {
1025 let _ = writeln!(out, " {line}");
1026 }
1027
1028 if let Some(factory) = client_factory {
1030 let fixture_id = &fixture.id;
1031 if needs_env_fallback {
1032 let api_key_var = api_key_var_opt.unwrap_or("");
1035 let mock_url_expr = if fixture.has_host_root_route() {
1036 let env_key = format!("MOCK_SERVER_{}", fixture_id.to_uppercase());
1037 format!(
1038 "System.get_env(\"{env_key}\") || (System.get_env(\"MOCK_SERVER_URL\") || \"\") <> \"/fixtures/{fixture_id}\""
1039 )
1040 } else {
1041 format!("(System.get_env(\"MOCK_SERVER_URL\") || \"\") <> \"/fixtures/{fixture_id}\"")
1042 };
1043 let _ = writeln!(out, " api_key_val = System.get_env(\"{api_key_var}\")");
1044 let _ = writeln!(
1045 out,
1046 " {{api_key_val, client_opts}} = if api_key_val && api_key_val != \"\" do"
1047 );
1048 let _ = writeln!(
1049 out,
1050 " IO.puts(\"{fixture_id}: using real API ({api_key_var} is set)\")"
1051 );
1052 let _ = writeln!(out, " {{api_key_val, []}}");
1053 let _ = writeln!(out, " else");
1054 let _ = writeln!(
1055 out,
1056 " IO.puts(\"{fixture_id}: using mock server ({api_key_var} not set)\")"
1057 );
1058 let _ = writeln!(out, " {{\"test-key\", [base_url: {mock_url_expr}]}}");
1059 let _ = writeln!(out, " end");
1060 let _ = writeln!(
1061 out,
1062 " {{:ok, client}} = {module_path}.{factory}(api_key_val, client_opts)"
1063 );
1064 } else {
1065 let base_url_expr = if fixture.has_host_root_route() {
1066 let env_key = format!("MOCK_SERVER_{}", fixture_id.to_uppercase());
1067 format!(
1068 "(System.get_env(\"{env_key}\") || (System.get_env(\"MOCK_SERVER_URL\") || \"\") <> \"/fixtures/{fixture_id}\")"
1069 )
1070 } else {
1071 format!("(System.get_env(\"MOCK_SERVER_URL\") || \"\") <> \"/fixtures/{fixture_id}\"")
1072 };
1073 let _ = writeln!(
1074 out,
1075 " {{:ok, client}} = {module_path}.{factory}(\"test-key\", base_url: {base_url_expr})"
1076 );
1077 }
1078 }
1079
1080 let returns_result = call_overrides
1082 .and_then(|o| o.returns_result)
1083 .unwrap_or(call_config.returns_result || client_factory.is_some());
1084
1085 let result_is_simple = call_config.result_is_simple || call_overrides.is_some_and(|o| o.result_is_simple);
1090
1091 let is_streaming = crate::codegen::streaming_assertions::resolve_is_streaming(fixture, call_config.streaming);
1093 let chunks_var = "chunks";
1095
1096 if returns_result {
1097 let _ = writeln!(
1098 out,
1099 " {{:ok, {result_var}}} = {module_path}.{function_name}({effective_args})"
1100 );
1101 } else {
1102 let _ = writeln!(
1104 out,
1105 " {result_var} = {module_path}.{function_name}({effective_args})"
1106 );
1107 }
1108
1109 if is_streaming {
1111 if let Some(collect) = crate::codegen::streaming_assertions::StreamingFieldResolver::collect_snippet(
1112 "elixir",
1113 &result_var,
1114 chunks_var,
1115 ) {
1116 let _ = writeln!(out, " {collect}");
1117 }
1118 }
1119
1120 for assertion in &fixture.assertions {
1121 render_assertion(
1122 out,
1123 assertion,
1124 if is_streaming { chunks_var } else { &result_var },
1125 field_resolver,
1126 &module_path,
1127 e2e_config.effective_fields_enum(call_config),
1128 resolved_enum_fields_ref,
1129 result_is_simple,
1130 is_streaming,
1131 );
1132 }
1133
1134 if needs_api_key_skip {
1135 let _ = writeln!(out, " end");
1136 }
1137 let _ = writeln!(out, " end");
1138 let _ = writeln!(out, " end");
1139}
1140
1141#[allow(clippy::too_many_arguments)]
1145fn emit_elixir_batch_item_array(arr: &serde_json::Value, elem_type: &str) -> String {
1147 if let Some(items) = arr.as_array() {
1148 let item_strs: Vec<String> = items
1149 .iter()
1150 .filter_map(|item| {
1151 if let Some(obj) = item.as_object() {
1152 match elem_type {
1153 "BatchBytesItem" => {
1154 let content = obj.get("content").and_then(|v| v.as_array());
1155 let mime_type = obj.get("mime_type").and_then(|v| v.as_str()).unwrap_or("text/plain");
1156 let content_code = if let Some(arr) = content {
1157 let bytes: Vec<String> =
1158 arr.iter().filter_map(|v| v.as_u64().map(|n| n.to_string())).collect();
1159 format!("<<{}>>", bytes.join(", "))
1160 } else {
1161 "<<>>".to_string()
1162 };
1163 Some(format!(
1164 "%BatchBytesItem{{content: {}, mime_type: \"{}\"}}",
1165 content_code, mime_type
1166 ))
1167 }
1168 "BatchFileItem" => {
1169 let path = obj.get("path").and_then(|v| v.as_str()).unwrap_or("");
1170 Some(format!("%BatchFileItem{{path: \"{}\"}}", path))
1171 }
1172 _ => None,
1173 }
1174 } else {
1175 None
1176 }
1177 })
1178 .collect();
1179 format!("[{}]", item_strs.join(", "))
1180 } else {
1181 "[]".to_string()
1182 }
1183}
1184
1185#[allow(clippy::too_many_arguments)]
1186fn build_args_and_setup(
1187 input: &serde_json::Value,
1188 args: &[crate::config::ArgMapping],
1189 module_path: &str,
1190 options_type: Option<&str>,
1191 options_default_fn: Option<&str>,
1192 enum_fields: &HashMap<String, String>,
1193 fixture: &crate::fixture::Fixture,
1194 _handle_struct_type: Option<&str>,
1195 _handle_atom_list_fields: &std::collections::HashSet<String>,
1196 test_documents_path: &str,
1197 adapter_request_type: Option<&str>,
1198) -> (Vec<String>, String) {
1199 let fixture_id = &fixture.id;
1200 if args.is_empty() {
1201 let is_empty_input = match input {
1205 serde_json::Value::Null => true,
1206 serde_json::Value::Object(m) => m.is_empty(),
1207 _ => false,
1208 };
1209 if is_empty_input {
1210 return (Vec::new(), String::new());
1211 }
1212 return (Vec::new(), json_to_elixir(input));
1213 }
1214
1215 let mut setup_lines: Vec<String> = Vec::new();
1216 let mut parts: Vec<String> = Vec::new();
1217
1218 for arg in args {
1219 if arg.arg_type == "mock_url" {
1220 if fixture.has_host_root_route() {
1221 let env_key = format!("MOCK_SERVER_{}", fixture_id.to_uppercase());
1222 setup_lines.push(format!(
1223 "{} = System.get_env(\"{env_key}\") || (System.get_env(\"MOCK_SERVER_URL\") || \"\") <> \"/fixtures/{fixture_id}\"",
1224 arg.name,
1225 ));
1226 } else {
1227 setup_lines.push(format!(
1228 "{} = (System.get_env(\"MOCK_SERVER_URL\") || \"\") <> \"/fixtures/{fixture_id}\"",
1229 arg.name,
1230 ));
1231 }
1232 if let Some(req_type) = adapter_request_type {
1233 let req_var = format!("{}_req", arg.name);
1234 setup_lines.push(format!("{req_var} = %Kreuzcrawl.{req_type}{{url: {}}}", arg.name,));
1235 parts.push(req_var);
1236 } else {
1237 parts.push(arg.name.clone());
1238 }
1239 continue;
1240 }
1241
1242 if arg.arg_type == "handle" {
1243 let constructor_name = format!("create_{}", arg.name.to_snake_case());
1247 let config_value = if arg.field == "input" {
1248 input
1249 } else {
1250 let field = arg.field.strip_prefix("input.").unwrap_or(&arg.field);
1251 input.get(field).unwrap_or(&serde_json::Value::Null)
1252 };
1253 let name = &arg.name;
1254 if config_value.is_null()
1255 || config_value.is_object() && config_value.as_object().is_some_and(|o| o.is_empty())
1256 {
1257 setup_lines.push(format!("{{:ok, {name}}} = {module_path}.{constructor_name}(nil)"));
1258 } else {
1259 let json_str = serde_json::to_string(config_value).unwrap_or_else(|_| "{}".to_string());
1262 let escaped = escape_elixir(&json_str);
1263 setup_lines.push(format!("{name}_config = \"{escaped}\""));
1264 setup_lines.push(format!(
1265 "{{:ok, {name}}} = {module_path}.{constructor_name}({name}_config)",
1266 ));
1267 }
1268 parts.push(arg.name.clone());
1269 continue;
1270 }
1271
1272 let val = if arg.field == "input" {
1273 Some(input)
1274 } else {
1275 let field = arg.field.strip_prefix("input.").unwrap_or(&arg.field);
1276 input.get(field)
1277 };
1278 match val {
1279 None | Some(serde_json::Value::Null) if arg.optional => {
1280 continue;
1283 }
1284 None | Some(serde_json::Value::Null) => {
1285 let default_val = match arg.arg_type.as_str() {
1287 "string" => "\"\"".to_string(),
1288 "int" | "integer" => "0".to_string(),
1289 "float" | "number" => "0.0".to_string(),
1290 "bool" | "boolean" => "false".to_string(),
1291 _ => "nil".to_string(),
1292 };
1293 parts.push(default_val);
1294 }
1295 Some(v) => {
1296 if arg.arg_type == "file_path" {
1299 if let Some(path_str) = v.as_str() {
1300 let full_path = format!("{test_documents_path}/{path_str}");
1301 let formatted = format!("\"{}\"", escape_elixir(&full_path));
1302 if arg.optional {
1303 parts.push(format!("{}: {formatted}", arg.name));
1304 } else {
1305 parts.push(formatted);
1306 }
1307 continue;
1308 }
1309 }
1310 if arg.arg_type == "bytes" {
1313 if let Some(raw) = v.as_str() {
1314 let var_name = &arg.name;
1315 if raw.starts_with('<') || raw.starts_with('{') || raw.starts_with('[') || raw.contains(' ') {
1316 let formatted = format!("\"{}\"", escape_elixir(raw));
1318 if arg.optional {
1319 parts.push(format!("{}: {formatted}", arg.name));
1320 } else {
1321 parts.push(formatted);
1322 }
1323 } else {
1324 let first = raw.chars().next().unwrap_or('\0');
1325 let is_file_path = (first.is_ascii_alphanumeric() || first == '_')
1326 && raw
1327 .find('/')
1328 .is_some_and(|slash_pos| slash_pos > 0 && raw[slash_pos + 1..].contains('.'));
1329 if is_file_path {
1330 let full_path = format!("{test_documents_path}/{raw}");
1333 let escaped = escape_elixir(&full_path);
1334 setup_lines.push(format!("{var_name} = File.read!(\"{escaped}\")"));
1335 if arg.optional {
1336 parts.push(format!("{}: {var_name}", arg.name));
1337 } else {
1338 parts.push(var_name.to_string());
1339 }
1340 } else {
1341 setup_lines.push(format!(
1343 "{var_name} = Base.decode64!(\"{}\", padding: false)",
1344 escape_elixir(raw)
1345 ));
1346 if arg.optional {
1347 parts.push(format!("{}: {var_name}", arg.name));
1348 } else {
1349 parts.push(var_name.to_string());
1350 }
1351 }
1352 }
1353 continue;
1354 }
1355 }
1356 if arg.arg_type == "json_object" && !v.is_null() {
1358 if let (Some(_opts_type), Some(options_fn), Some(obj)) =
1359 (options_type, options_default_fn, v.as_object())
1360 {
1361 let options_var = "options";
1363 setup_lines.push(format!("{options_var} = {module_path}.{options_fn}()"));
1364
1365 for (k, vv) in obj.iter() {
1367 let snake_key = k.to_snake_case();
1368 let elixir_val = if let Some(_enum_type) = enum_fields.get(k) {
1369 if let Some(s) = vv.as_str() {
1370 let snake_val = s.to_snake_case();
1371 format!(":{snake_val}")
1373 } else {
1374 json_to_elixir(vv)
1375 }
1376 } else {
1377 json_to_elixir(vv)
1378 };
1379 setup_lines.push(format!(
1380 "{options_var} = %{{{options_var} | {snake_key}: {elixir_val}}}"
1381 ));
1382 }
1383
1384 if arg.optional {
1387 parts.push(format!("{}: {options_var}", arg.name));
1388 } else {
1389 parts.push(options_var.to_string());
1390 }
1391 continue;
1392 }
1393 if let (Some(opts_type), None, Some(obj)) = (options_type, options_default_fn, v.as_object()) {
1395 let options_var = "options";
1396 let mut field_strs = Vec::new();
1397 for (k, vv) in obj.iter() {
1398 let snake_key = k.to_snake_case();
1399 let elixir_val = if let Some(_enum_type) = enum_fields.get(k) {
1400 if let Some(s) = vv.as_str() {
1401 let snake_val = s.to_snake_case();
1402 format!(":{snake_val}")
1403 } else {
1404 json_to_elixir(vv)
1405 }
1406 } else {
1407 json_to_elixir(vv)
1408 };
1409 field_strs.push(format!("{snake_key}: {elixir_val}"));
1410 }
1411 let fields = field_strs.join(", ");
1412 setup_lines.push(format!("{options_var} = %{module_path}.{opts_type}{{{fields}}}"));
1413 if arg.optional {
1414 parts.push(format!("{}: {options_var}", arg.name));
1415 } else {
1416 parts.push(options_var.to_string());
1417 }
1418 continue;
1419 }
1420 if let Some(elem_type) = &arg.element_type {
1422 if (elem_type == "BatchBytesItem" || elem_type == "BatchFileItem") && v.is_array() {
1423 let formatted = emit_elixir_batch_item_array(v, elem_type);
1424 if arg.optional {
1425 parts.push(format!("{}: {formatted}", arg.name));
1426 } else {
1427 parts.push(formatted);
1428 }
1429 continue;
1430 }
1431 if v.is_array() {
1434 let formatted = json_to_elixir(v);
1435 if arg.optional {
1436 parts.push(format!("{}: {formatted}", arg.name));
1437 } else {
1438 parts.push(formatted);
1439 }
1440 continue;
1441 }
1442 }
1443 if !v.is_null() {
1447 let json_str = serde_json::to_string(v).unwrap_or_else(|_| "{}".to_string());
1448 let escaped = escape_elixir(&json_str);
1449 let formatted = format!("\"{escaped}\"");
1450 if arg.optional {
1451 parts.push(format!("{}: {formatted}", arg.name));
1452 } else {
1453 parts.push(formatted);
1454 }
1455 continue;
1456 }
1457 }
1458 let elixir_val = json_to_elixir(v);
1460 if arg.optional {
1461 parts.push(format!("{}: {elixir_val}", arg.name));
1462 } else {
1463 parts.push(elixir_val);
1464 }
1465 }
1466 }
1467 }
1468
1469 let mut positional_args = Vec::new();
1471 let mut keyword_args = Vec::new();
1472 for part in parts {
1473 if part.contains(": ") && !part.starts_with('"') {
1474 keyword_args.push(part);
1475 } else {
1476 positional_args.push(part);
1477 }
1478 }
1479
1480 let mut final_args = positional_args;
1481 final_args.extend(keyword_args);
1482
1483 (setup_lines, final_args.join(", "))
1484}
1485
1486fn is_numeric_expr(field_expr: &str) -> bool {
1489 field_expr.starts_with("length(")
1490}
1491
1492#[allow(clippy::too_many_arguments)]
1493fn render_assertion(
1494 out: &mut String,
1495 assertion: &Assertion,
1496 result_var: &str,
1497 field_resolver: &FieldResolver,
1498 module_path: &str,
1499 fields_enum: &std::collections::HashSet<String>,
1500 per_call_enum_fields: &HashMap<String, String>,
1501 result_is_simple: bool,
1502 is_streaming: bool,
1503) {
1504 if let Some(f) = &assertion.field {
1507 match f.as_str() {
1508 "chunks_have_content" => {
1509 let pred =
1510 format!("Enum.all?({result_var}.chunks || [], fn c -> c.content != nil and c.content != \"\" end)");
1511 match assertion.assertion_type.as_str() {
1512 "is_true" => {
1513 let _ = writeln!(out, " assert {pred}");
1514 }
1515 "is_false" => {
1516 let _ = writeln!(out, " refute {pred}");
1517 }
1518 _ => {
1519 let _ = writeln!(
1520 out,
1521 " # skipped: unsupported assertion type on synthetic field '{f}'"
1522 );
1523 }
1524 }
1525 return;
1526 }
1527 "chunks_have_embeddings" => {
1528 let pred = format!(
1529 "Enum.all?({result_var}.chunks || [], fn c -> c.embedding != nil and c.embedding != [] end)"
1530 );
1531 match assertion.assertion_type.as_str() {
1532 "is_true" => {
1533 let _ = writeln!(out, " assert {pred}");
1534 }
1535 "is_false" => {
1536 let _ = writeln!(out, " refute {pred}");
1537 }
1538 _ => {
1539 let _ = writeln!(
1540 out,
1541 " # skipped: unsupported assertion type on synthetic field '{f}'"
1542 );
1543 }
1544 }
1545 return;
1546 }
1547 "embeddings" => {
1551 match assertion.assertion_type.as_str() {
1552 "count_equals" => {
1553 if let Some(val) = &assertion.value {
1554 let ex_val = json_to_elixir(val);
1555 let _ = writeln!(out, " assert length({result_var}) == {ex_val}");
1556 }
1557 }
1558 "count_min" => {
1559 if let Some(val) = &assertion.value {
1560 let ex_val = json_to_elixir(val);
1561 let _ = writeln!(out, " assert length({result_var}) >= {ex_val}");
1562 }
1563 }
1564 "not_empty" => {
1565 let _ = writeln!(out, " assert {result_var} != []");
1566 }
1567 "is_empty" => {
1568 let _ = writeln!(out, " assert {result_var} == []");
1569 }
1570 _ => {
1571 let _ = writeln!(
1572 out,
1573 " # skipped: unsupported assertion type on synthetic field 'embeddings'"
1574 );
1575 }
1576 }
1577 return;
1578 }
1579 "embedding_dimensions" => {
1580 let expr = format!("(if {result_var} == [], do: 0, else: length(hd({result_var})))");
1581 match assertion.assertion_type.as_str() {
1582 "equals" => {
1583 if let Some(val) = &assertion.value {
1584 let ex_val = json_to_elixir(val);
1585 let _ = writeln!(out, " assert {expr} == {ex_val}");
1586 }
1587 }
1588 "greater_than" => {
1589 if let Some(val) = &assertion.value {
1590 let ex_val = json_to_elixir(val);
1591 let _ = writeln!(out, " assert {expr} > {ex_val}");
1592 }
1593 }
1594 _ => {
1595 let _ = writeln!(
1596 out,
1597 " # skipped: unsupported assertion type on synthetic field 'embedding_dimensions'"
1598 );
1599 }
1600 }
1601 return;
1602 }
1603 "embeddings_valid" | "embeddings_finite" | "embeddings_non_zero" | "embeddings_normalized" => {
1604 let pred = match f.as_str() {
1605 "embeddings_valid" => {
1606 format!("Enum.all?({result_var}, fn e -> e != [] end)")
1607 }
1608 "embeddings_finite" => {
1609 format!("Enum.all?({result_var}, fn e -> Enum.all?(e, fn v -> is_float(v) and v == v end) end)")
1610 }
1611 "embeddings_non_zero" => {
1612 format!("Enum.all?({result_var}, fn e -> Enum.any?(e, fn v -> v != 0.0 end) end)")
1613 }
1614 "embeddings_normalized" => {
1615 format!(
1616 "Enum.all?({result_var}, fn e -> n = Enum.reduce(e, 0.0, fn v, acc -> acc + v * v end); abs(n - 1.0) < 1.0e-3 end)"
1617 )
1618 }
1619 _ => unreachable!(),
1620 };
1621 match assertion.assertion_type.as_str() {
1622 "is_true" => {
1623 let _ = writeln!(out, " assert {pred}");
1624 }
1625 "is_false" => {
1626 let _ = writeln!(out, " refute {pred}");
1627 }
1628 _ => {
1629 let _ = writeln!(
1630 out,
1631 " # skipped: unsupported assertion type on synthetic field '{f}'"
1632 );
1633 }
1634 }
1635 return;
1636 }
1637 "keywords" | "keywords_count" => {
1640 let _ = writeln!(
1641 out,
1642 " # skipped: field '{f}' not available on Elixir ExtractionResult"
1643 );
1644 return;
1645 }
1646 _ => {}
1647 }
1648 }
1649
1650 if is_streaming {
1653 if let Some(f) = &assertion.field {
1654 if !f.is_empty() && crate::codegen::streaming_assertions::is_streaming_virtual_field(f) {
1655 if let Some(expr) =
1656 crate::codegen::streaming_assertions::StreamingFieldResolver::accessor(f, "elixir", result_var)
1657 {
1658 match assertion.assertion_type.as_str() {
1659 "count_min" => {
1660 if let Some(n) = assertion.value.as_ref().and_then(|v| v.as_u64()) {
1661 let _ = writeln!(out, " assert length({expr}) >= {n}");
1662 }
1663 }
1664 "count_equals" => {
1665 if let Some(n) = assertion.value.as_ref().and_then(|v| v.as_u64()) {
1666 let _ = writeln!(out, " assert length({expr}) == {n}");
1667 }
1668 }
1669 "equals" => {
1670 if let Some(serde_json::Value::String(s)) = &assertion.value {
1671 let escaped = escape_elixir(s);
1672 let _ = writeln!(out, " assert {expr} == \"{escaped}\"");
1673 } else if let Some(n) = assertion.value.as_ref().and_then(|v| v.as_u64()) {
1674 let _ = writeln!(out, " assert {expr} == {n}");
1675 }
1676 }
1677 "not_empty" => {
1678 let _ = writeln!(out, " assert {expr} != []");
1679 }
1680 "is_empty" => {
1681 let _ = writeln!(out, " assert {expr} == []");
1682 }
1683 "is_true" => {
1684 let _ = writeln!(out, " assert {expr}");
1685 }
1686 "is_false" => {
1687 let _ = writeln!(out, " refute {expr}");
1688 }
1689 "greater_than" => {
1690 if let Some(n) = assertion.value.as_ref().and_then(|v| v.as_u64()) {
1691 let _ = writeln!(out, " assert {expr} > {n}");
1692 }
1693 }
1694 "greater_than_or_equal" => {
1695 if let Some(n) = assertion.value.as_ref().and_then(|v| v.as_u64()) {
1696 let _ = writeln!(out, " assert {expr} >= {n}");
1697 }
1698 }
1699 "contains" => {
1700 if let Some(serde_json::Value::String(s)) = &assertion.value {
1701 let escaped = escape_elixir(s);
1702 let _ = writeln!(out, " assert String.contains?({expr}, \"{escaped}\")");
1703 }
1704 }
1705 _ => {
1706 let _ = writeln!(
1707 out,
1708 " # streaming field '{f}': assertion type '{}' not rendered",
1709 assertion.assertion_type
1710 );
1711 }
1712 }
1713 }
1714 return;
1715 }
1716 }
1717 }
1718
1719 if !result_is_simple {
1724 if let Some(f) = &assertion.field {
1725 if !f.is_empty() && !field_resolver.is_valid_for_result(f) {
1726 let _ = writeln!(out, " # skipped: field '{f}' not available on result type");
1727 return;
1728 }
1729 }
1730 }
1731
1732 let field_expr = if result_is_simple {
1736 result_var.to_string()
1737 } else {
1738 match &assertion.field {
1739 Some(f) if !f.is_empty() => field_resolver.accessor(f, "elixir", result_var),
1740 _ => result_var.to_string(),
1741 }
1742 };
1743
1744 let is_numeric = is_numeric_expr(&field_expr);
1747 let field_is_enum = assertion.field.as_deref().filter(|f| !f.is_empty()).is_some_and(|f| {
1754 let resolved = field_resolver.resolve(f);
1755 fields_enum.contains(f)
1756 || fields_enum.contains(resolved)
1757 || per_call_enum_fields.contains_key(f)
1758 || per_call_enum_fields.contains_key(resolved)
1759 });
1760 let coerced_field_expr = if field_is_enum {
1761 format!("to_string({field_expr})")
1762 } else {
1763 field_expr.clone()
1764 };
1765 let trimmed_field_expr = if is_numeric {
1766 field_expr.clone()
1767 } else {
1768 format!("String.trim({coerced_field_expr})")
1769 };
1770
1771 let field_is_array = assertion
1774 .field
1775 .as_deref()
1776 .filter(|f| !f.is_empty())
1777 .is_some_and(|f| field_resolver.is_array(field_resolver.resolve(f)));
1778
1779 match assertion.assertion_type.as_str() {
1780 "equals" => {
1781 if let Some(expected) = &assertion.value {
1782 let elixir_val = json_to_elixir(expected);
1783 let is_string_expected = expected.is_string();
1785 if is_string_expected && !is_numeric {
1786 let _ = writeln!(out, " assert {trimmed_field_expr} == {elixir_val}");
1787 } else if field_is_enum {
1788 let _ = writeln!(out, " assert {coerced_field_expr} == {elixir_val}");
1789 } else {
1790 let _ = writeln!(out, " assert {field_expr} == {elixir_val}");
1791 }
1792 }
1793 }
1794 "contains" => {
1795 if let Some(expected) = &assertion.value {
1796 let elixir_val = json_to_elixir(expected);
1797 if field_is_array && expected.is_string() {
1798 let _ = writeln!(
1800 out,
1801 " assert Enum.any?({field_expr}, fn item -> Enum.any?(alef_e2e_item_texts(item), &String.contains?(&1, {elixir_val})) end)"
1802 );
1803 } else {
1804 let _ = writeln!(
1806 out,
1807 " assert String.contains?(to_string({field_expr}), {elixir_val})"
1808 );
1809 }
1810 }
1811 }
1812 "contains_all" => {
1813 if let Some(values) = &assertion.values {
1814 for val in values {
1815 let elixir_val = json_to_elixir(val);
1816 if field_is_array && val.is_string() {
1817 let _ = writeln!(
1818 out,
1819 " assert Enum.any?({field_expr}, fn item -> Enum.any?(alef_e2e_item_texts(item), &String.contains?(&1, {elixir_val})) end)"
1820 );
1821 } else {
1822 let _ = writeln!(
1823 out,
1824 " assert String.contains?(to_string({field_expr}), {elixir_val})"
1825 );
1826 }
1827 }
1828 }
1829 }
1830 "not_contains" => {
1831 if let Some(expected) = &assertion.value {
1832 let elixir_val = json_to_elixir(expected);
1833 if field_is_array && expected.is_string() {
1834 let _ = writeln!(
1835 out,
1836 " refute Enum.any?({field_expr}, fn item -> Enum.any?(alef_e2e_item_texts(item), &String.contains?(&1, {elixir_val})) end)"
1837 );
1838 } else {
1839 let _ = writeln!(
1840 out,
1841 " refute String.contains?(to_string({field_expr}), {elixir_val})"
1842 );
1843 }
1844 }
1845 }
1846 "not_empty" => {
1847 let _ = writeln!(out, " assert {field_expr} != \"\"");
1848 }
1849 "is_empty" => {
1850 if is_numeric {
1851 let _ = writeln!(out, " assert {field_expr} == 0");
1853 } else {
1854 let _ = writeln!(out, " assert is_nil({field_expr}) or {trimmed_field_expr} == \"\"");
1856 }
1857 }
1858 "contains_any" => {
1859 if let Some(values) = &assertion.values {
1860 let items: Vec<String> = values.iter().map(json_to_elixir).collect();
1861 let list_str = items.join(", ");
1862 let _ = writeln!(
1863 out,
1864 " assert Enum.any?([{list_str}], fn v -> String.contains?(to_string({field_expr}), v) end)"
1865 );
1866 }
1867 }
1868 "greater_than" => {
1869 if let Some(val) = &assertion.value {
1870 let elixir_val = json_to_elixir(val);
1871 let _ = writeln!(out, " assert {field_expr} > {elixir_val}");
1872 }
1873 }
1874 "less_than" => {
1875 if let Some(val) = &assertion.value {
1876 let elixir_val = json_to_elixir(val);
1877 let _ = writeln!(out, " assert {field_expr} < {elixir_val}");
1878 }
1879 }
1880 "greater_than_or_equal" => {
1881 if let Some(val) = &assertion.value {
1882 let elixir_val = json_to_elixir(val);
1883 let _ = writeln!(out, " assert {field_expr} >= {elixir_val}");
1884 }
1885 }
1886 "less_than_or_equal" => {
1887 if let Some(val) = &assertion.value {
1888 let elixir_val = json_to_elixir(val);
1889 let _ = writeln!(out, " assert {field_expr} <= {elixir_val}");
1890 }
1891 }
1892 "starts_with" => {
1893 if let Some(expected) = &assertion.value {
1894 let elixir_val = json_to_elixir(expected);
1895 let _ = writeln!(out, " assert String.starts_with?({field_expr}, {elixir_val})");
1896 }
1897 }
1898 "ends_with" => {
1899 if let Some(expected) = &assertion.value {
1900 let elixir_val = json_to_elixir(expected);
1901 let _ = writeln!(out, " assert String.ends_with?({field_expr}, {elixir_val})");
1902 }
1903 }
1904 "min_length" => {
1905 if let Some(val) = &assertion.value {
1906 if let Some(n) = val.as_u64() {
1907 let _ = writeln!(out, " assert String.length({field_expr}) >= {n}");
1908 }
1909 }
1910 }
1911 "max_length" => {
1912 if let Some(val) = &assertion.value {
1913 if let Some(n) = val.as_u64() {
1914 let _ = writeln!(out, " assert String.length({field_expr}) <= {n}");
1915 }
1916 }
1917 }
1918 "count_min" => {
1919 if let Some(val) = &assertion.value {
1920 if let Some(n) = val.as_u64() {
1921 let _ = writeln!(out, " assert length({field_expr}) >= {n}");
1922 }
1923 }
1924 }
1925 "count_equals" => {
1926 if let Some(val) = &assertion.value {
1927 if let Some(n) = val.as_u64() {
1928 let _ = writeln!(out, " assert length({field_expr}) == {n}");
1929 }
1930 }
1931 }
1932 "is_true" => {
1933 let _ = writeln!(out, " assert {field_expr} == true");
1934 }
1935 "is_false" => {
1936 let _ = writeln!(out, " assert {field_expr} == false");
1937 }
1938 "method_result" => {
1939 if let Some(method_name) = &assertion.method {
1940 let call_expr = build_elixir_method_call(result_var, method_name, assertion.args.as_ref(), module_path);
1941 let check = assertion.check.as_deref().unwrap_or("is_true");
1942 match check {
1943 "equals" => {
1944 if let Some(val) = &assertion.value {
1945 let elixir_val = json_to_elixir(val);
1946 let _ = writeln!(out, " assert {call_expr} == {elixir_val}");
1947 }
1948 }
1949 "is_true" => {
1950 let _ = writeln!(out, " assert {call_expr} == true");
1951 }
1952 "is_false" => {
1953 let _ = writeln!(out, " assert {call_expr} == false");
1954 }
1955 "greater_than_or_equal" => {
1956 if let Some(val) = &assertion.value {
1957 let n = val.as_u64().unwrap_or(0);
1958 let _ = writeln!(out, " assert {call_expr} >= {n}");
1959 }
1960 }
1961 "count_min" => {
1962 if let Some(val) = &assertion.value {
1963 let n = val.as_u64().unwrap_or(0);
1964 let _ = writeln!(out, " assert length({call_expr}) >= {n}");
1965 }
1966 }
1967 "contains" => {
1968 if let Some(val) = &assertion.value {
1969 let elixir_val = json_to_elixir(val);
1970 let _ = writeln!(out, " assert String.contains?({call_expr}, {elixir_val})");
1971 }
1972 }
1973 "is_error" => {
1974 let _ = writeln!(out, " assert_raise RuntimeError, fn -> {call_expr} end");
1975 }
1976 other_check => {
1977 panic!("Elixir e2e generator: unsupported method_result check type: {other_check}");
1978 }
1979 }
1980 } else {
1981 panic!("Elixir e2e generator: method_result assertion missing 'method' field");
1982 }
1983 }
1984 "matches_regex" => {
1985 if let Some(expected) = &assertion.value {
1986 let elixir_val = json_to_elixir(expected);
1987 let _ = writeln!(out, " assert Regex.match?(~r/{elixir_val}/, {field_expr})");
1988 }
1989 }
1990 "not_error" => {
1991 }
1993 "error" => {
1994 }
1996 other => {
1997 panic!("Elixir e2e generator: unsupported assertion type: {other}");
1998 }
1999 }
2000}
2001
2002fn build_elixir_method_call(
2005 result_var: &str,
2006 method_name: &str,
2007 args: Option<&serde_json::Value>,
2008 module_path: &str,
2009) -> String {
2010 match method_name {
2011 "root_child_count" => format!("{module_path}.root_child_count({result_var})"),
2012 "has_error_nodes" => format!("{module_path}.tree_has_error_nodes({result_var})"),
2013 "error_count" | "tree_error_count" => format!("{module_path}.tree_error_count({result_var})"),
2014 "tree_to_sexp" => format!("{module_path}.tree_to_sexp({result_var})"),
2015 "contains_node_type" => {
2016 let node_type = args
2017 .and_then(|a| a.get("node_type"))
2018 .and_then(|v| v.as_str())
2019 .unwrap_or("");
2020 format!("{module_path}.tree_contains_node_type({result_var}, \"{node_type}\")")
2021 }
2022 "find_nodes_by_type" => {
2023 let node_type = args
2024 .and_then(|a| a.get("node_type"))
2025 .and_then(|v| v.as_str())
2026 .unwrap_or("");
2027 format!("{module_path}.find_nodes_by_type({result_var}, \"{node_type}\")")
2028 }
2029 "run_query" => {
2030 let query_source = args
2031 .and_then(|a| a.get("query_source"))
2032 .and_then(|v| v.as_str())
2033 .unwrap_or("");
2034 let language = args
2035 .and_then(|a| a.get("language"))
2036 .and_then(|v| v.as_str())
2037 .unwrap_or("");
2038 format!("{module_path}.run_query({result_var}, \"{language}\", \"{query_source}\", source)")
2039 }
2040 _ => format!("{module_path}.{method_name}({result_var})"),
2041 }
2042}
2043
2044fn elixir_module_name(category: &str) -> String {
2046 use heck::ToUpperCamelCase;
2047 category.to_upper_camel_case()
2048}
2049
2050fn json_to_elixir(value: &serde_json::Value) -> String {
2052 match value {
2053 serde_json::Value::String(s) => format!("\"{}\"", escape_elixir(s)),
2054 serde_json::Value::Bool(true) => "true".to_string(),
2055 serde_json::Value::Bool(false) => "false".to_string(),
2056 serde_json::Value::Number(n) => {
2057 let s = n.to_string().replace("e+", "e");
2061 if s.contains('e') && !s.contains('.') {
2062 s.replacen('e', ".0e", 1)
2064 } else {
2065 s
2066 }
2067 }
2068 serde_json::Value::Null => "nil".to_string(),
2069 serde_json::Value::Array(arr) => {
2070 let items: Vec<String> = arr.iter().map(json_to_elixir).collect();
2071 format!("[{}]", items.join(", "))
2072 }
2073 serde_json::Value::Object(map) => {
2074 let entries: Vec<String> = map
2075 .iter()
2076 .map(|(k, v)| format!("\"{}\" => {}", escape_elixir(k), json_to_elixir(v)))
2077 .collect();
2078 format!("%{{{}}}", entries.join(", "))
2079 }
2080 }
2081}
2082
2083fn build_elixir_visitor(setup_lines: &mut Vec<String>, visitor_spec: &crate::fixture::VisitorSpec) -> String {
2085 use std::fmt::Write as FmtWrite;
2086 let mut visitor_obj = String::new();
2087 let _ = writeln!(visitor_obj, "%{{");
2088 for (method_name, action) in &visitor_spec.callbacks {
2089 emit_elixir_visitor_method(&mut visitor_obj, method_name, action);
2090 }
2091 let _ = writeln!(visitor_obj, " }}");
2092
2093 setup_lines.push(format!("visitor = {visitor_obj}"));
2094 "visitor".to_string()
2095}
2096
2097fn emit_elixir_visitor_method(out: &mut String, method_name: &str, action: &CallbackAction) {
2099 use std::fmt::Write as FmtWrite;
2100
2101 let handle_method = format!("handle_{}", &method_name[6..]); let arg_binding = match action {
2111 CallbackAction::CustomTemplate { .. } => "args",
2112 _ => "_args",
2113 };
2114 let _ = writeln!(out, " :{handle_method} => fn({arg_binding}) ->");
2115 match action {
2116 CallbackAction::Skip => {
2117 let _ = writeln!(out, " :skip");
2118 }
2119 CallbackAction::Continue => {
2120 let _ = writeln!(out, " :continue");
2121 }
2122 CallbackAction::PreserveHtml => {
2123 let _ = writeln!(out, " :preserve_html");
2124 }
2125 CallbackAction::Custom { output } => {
2126 let escaped = escape_elixir(output);
2127 let _ = writeln!(out, " {{:custom, \"{escaped}\"}}");
2128 }
2129 CallbackAction::CustomTemplate { template, .. } => {
2130 let expr = template_to_elixir_concat(template);
2134 let _ = writeln!(out, " {{:custom, {expr}}}");
2135 }
2136 }
2137 let _ = writeln!(out, " end,");
2138}
2139
2140fn template_to_elixir_concat(template: &str) -> String {
2145 let mut parts: Vec<String> = Vec::new();
2146 let mut static_buf = String::new();
2147 let mut chars = template.chars().peekable();
2148
2149 while let Some(ch) = chars.next() {
2150 if ch == '{' {
2151 let mut key = String::new();
2152 let mut closed = false;
2153 for kc in chars.by_ref() {
2154 if kc == '}' {
2155 closed = true;
2156 break;
2157 }
2158 key.push(kc);
2159 }
2160 if closed && !key.is_empty() {
2161 if !static_buf.is_empty() {
2162 let escaped = escape_elixir(&static_buf);
2163 parts.push(format!("\"{escaped}\""));
2164 static_buf.clear();
2165 }
2166 let escaped_key = escape_elixir(&key);
2167 parts.push(format!("Map.get(args, \"{escaped_key}\", \"\")"));
2168 } else {
2169 static_buf.push('{');
2170 static_buf.push_str(&key);
2171 if !closed {
2172 }
2174 }
2175 } else {
2176 static_buf.push(ch);
2177 }
2178 }
2179
2180 if !static_buf.is_empty() {
2181 let escaped = escape_elixir(&static_buf);
2182 parts.push(format!("\"{escaped}\""));
2183 }
2184
2185 if parts.is_empty() {
2186 return "\"\"".to_string();
2187 }
2188 parts.join(" <> ")
2189}
2190
2191fn fixture_has_elixir_callable(fixture: &Fixture, e2e_config: &E2eConfig) -> bool {
2192 if fixture.is_http_test() {
2194 return false;
2195 }
2196 let call_config = e2e_config.resolve_call_for_fixture(
2197 fixture.call.as_deref(),
2198 &fixture.id,
2199 &fixture.resolved_category(),
2200 &fixture.tags,
2201 &fixture.input,
2202 );
2203 let elixir_override = call_config
2204 .overrides
2205 .get("elixir")
2206 .or_else(|| e2e_config.call.overrides.get("elixir"));
2207 if elixir_override.and_then(|o| o.client_factory.as_deref()).is_some() {
2209 return true;
2210 }
2211 let function_from_override = elixir_override.and_then(|o| o.function.as_deref());
2216
2217 function_from_override.is_some() || !call_config.function.is_empty()
2219}