use crate::config::E2eConfig;
use crate::escape::{escape_gleam, sanitize_filename, sanitize_ident};
use crate::field_access::FieldResolver;
use crate::fixture::{Assertion, Fixture, FixtureGroup, ValidationErrorExpectation};
use alef_core::backend::GeneratedFile;
use alef_core::config::ResolvedCrateConfig;
use alef_core::hash::{self, CommentStyle};
use anyhow::Result;
use heck::{ToPascalCase, ToSnakeCase};
use std::collections::HashSet;
use std::fmt::Write as FmtWrite;
use std::path::PathBuf;
use super::E2eCodegen;
use super::client;
pub struct GleamE2eCodegen;
impl E2eCodegen for GleamE2eCodegen {
fn generate(
&self,
groups: &[FixtureGroup],
e2e_config: &E2eConfig,
config: &ResolvedCrateConfig,
_type_defs: &[alef_core::ir::TypeDef],
) -> Result<Vec<GeneratedFile>> {
let lang = self.language_name();
let output_base = PathBuf::from(e2e_config.effective_output()).join(lang);
let mut files = Vec::new();
let call = &e2e_config.call;
let overrides = call.overrides.get(lang);
let module_path = overrides
.and_then(|o| o.module.as_ref())
.cloned()
.unwrap_or_else(|| call.module.clone());
let function_name = overrides
.and_then(|o| o.function.as_ref())
.cloned()
.unwrap_or_else(|| call.function.clone());
let result_var = &call.result_var;
let gleam_pkg = e2e_config.resolve_package("gleam");
let pkg_path = gleam_pkg
.as_ref()
.and_then(|p| p.path.as_ref())
.cloned()
.unwrap_or_else(|| "../../packages/gleam".to_string());
let pkg_name = gleam_pkg
.as_ref()
.and_then(|p| p.name.as_ref())
.cloned()
.unwrap_or_else(|| config.name.to_snake_case());
files.push(GeneratedFile {
path: output_base.join("gleam.toml"),
content: render_gleam_toml(&pkg_path, &pkg_name, e2e_config.dep_mode),
generated_header: false,
});
let app_name = pkg_name.clone();
let e2e_helpers = format!(
"// Generated by alef. Do not edit by hand.\n\
// E2e helper module — provides file-reading utilities for Gleam tests.\n\
import gleam/dynamic\n\
\n\
/// Read a file into a BitArray via the Erlang :file module.\n\
/// The path is relative to the e2e working directory when `gleam test` runs.\n\
@external(erlang, \"file\", \"read_file\")\n\
pub fn read_file_bytes(path: String) -> Result(BitArray, dynamic.Dynamic)\n\
\n\
/// Ensure the {app_name} OTP application and all its dependencies are started.\n\
/// This is required when running `gleam test` outside of `mix test`, since the\n\
/// Rustler NIF init hook needs the :{app_name} application to be started before\n\
/// any binding-native functions can be called.\n\
/// Calls the Erlang shim e2e_startup:start_app/0.\n\
@external(erlang, \"e2e_startup\", \"start_app\")\n\
pub fn start_app() -> Nil\n",
);
let erlang_startup = format!(
"%% Generated by alef. Do not edit by hand.\n\
%% Starts the {app_name} OTP application and all its dependencies.\n\
%% Called by e2e_gleam_test.main/0 before gleeunit.main/0.\n\
-module(e2e_startup).\n\
-export([start_app/0]).\n\
\n\
start_app() ->\n\
\x20\x20\x20\x20%% Elixir runtime must be started before {app_name} NIF init\n\
\x20\x20\x20\x20%% because Rustler uses Elixir.Application.app_dir/2 to locate the .so.\n\
\x20\x20\x20\x20{{ok, _}} = application:ensure_all_started(elixir),\n\
\x20\x20\x20\x20{{ok, _}} = application:ensure_all_started({app_name}),\n\
\x20\x20\x20\x20nil.\n",
);
files.push(GeneratedFile {
path: output_base.join("src").join("e2e_gleam.gleam"),
content: e2e_helpers,
generated_header: false,
});
files.push(GeneratedFile {
path: output_base.join("src").join("e2e_startup.erl"),
content: erlang_startup,
generated_header: false,
});
let mut any_tests = false;
for group in groups {
let active: Vec<&Fixture> = group
.fixtures
.iter()
.filter(|f| super::should_include_fixture(f, lang, e2e_config))
.filter(|f| {
if let Some(http) = &f.http {
let has_upgrade = http
.request
.headers
.iter()
.any(|(k, v)| k.eq_ignore_ascii_case("upgrade") && v.eq_ignore_ascii_case("websocket"));
!has_upgrade
} else {
true
}
})
.collect();
if active.is_empty() {
continue;
}
let filename = format!("{}_test.gleam", sanitize_filename(&group.category));
let field_resolver = FieldResolver::new(
&e2e_config.fields,
&e2e_config.fields_optional,
&e2e_config.result_fields,
&e2e_config.fields_array,
&e2e_config.fields_method_calls,
);
let element_constructors: &[alef_core::config::GleamElementConstructor] = config
.gleam
.as_ref()
.map(|g| g.element_constructors.as_slice())
.unwrap_or(&[]);
let json_object_wrapper: Option<&str> = config
.gleam
.as_ref()
.and_then(|g| g.json_object_wrapper.as_deref());
let content = render_test_file(
&group.category,
&active,
e2e_config,
&module_path,
&function_name,
result_var,
&e2e_config.call.args,
&field_resolver,
&e2e_config.fields_enum,
element_constructors,
json_object_wrapper,
);
files.push(GeneratedFile {
path: output_base.join("test").join(filename),
content,
generated_header: true,
});
any_tests = true;
}
let entry = if any_tests {
concat!(
"// Generated by alef. Do not edit by hand.\n",
"import gleeunit\n",
"import e2e_gleam\n",
"\n",
"pub fn main() {\n",
" let _ = e2e_gleam.start_app()\n",
" gleeunit.main()\n",
"}\n",
)
.to_string()
} else {
concat!(
"// Generated by alef. Do not edit by hand.\n",
"// No fixture-driven tests for Gleam — e2e tests require HTTP fixtures\n",
"// or non-HTTP fixtures with gleam-specific call overrides.\n",
"import gleeunit\n",
"import gleeunit/should\n",
"\n",
"pub fn main() {\n",
" gleeunit.main()\n",
"}\n",
"\n",
"pub fn compilation_smoke_test() {\n",
" True |> should.equal(True)\n",
"}\n",
)
.to_string()
};
files.push(GeneratedFile {
path: output_base.join("test").join("e2e_gleam_test.gleam"),
content: entry,
generated_header: false,
});
Ok(files)
}
fn language_name(&self) -> &'static str {
"gleam"
}
}
fn render_gleam_toml(pkg_path: &str, pkg_name: &str, dep_mode: crate::config::DependencyMode) -> String {
use alef_core::template_versions::hex;
let stdlib = hex::GLEAM_STDLIB_VERSION_RANGE;
let gleeunit = hex::GLEEUNIT_VERSION_RANGE;
let gleam_httpc = hex::GLEAM_HTTPC_VERSION_RANGE;
let envoy = hex::ENVOY_VERSION_RANGE;
let deps = match dep_mode {
crate::config::DependencyMode::Registry => {
format!(
r#"{pkg_name} = ">= 0.1.0"
gleam_stdlib = "{stdlib}"
gleeunit = "{gleeunit}"
gleam_httpc = "{gleam_httpc}"
gleam_http = ">= 4.0.0 and < 5.0.0"
envoy = "{envoy}""#
)
}
crate::config::DependencyMode::Local => {
format!(
r#"{pkg_name} = {{ path = "{pkg_path}" }}
gleam_stdlib = "{stdlib}"
gleeunit = "{gleeunit}"
gleam_httpc = "{gleam_httpc}"
gleam_http = ">= 4.0.0 and < 5.0.0"
envoy = "{envoy}""#
)
}
};
format!(
r#"name = "e2e_gleam"
version = "0.1.0"
target = "erlang"
[dependencies]
{deps}
"#
)
}
#[allow(clippy::too_many_arguments)]
fn render_test_file(
_category: &str,
fixtures: &[&Fixture],
e2e_config: &E2eConfig,
module_path: &str,
function_name: &str,
result_var: &str,
args: &[crate::config::ArgMapping],
field_resolver: &FieldResolver,
enum_fields: &HashSet<String>,
element_constructors: &[alef_core::config::GleamElementConstructor],
json_object_wrapper: Option<&str>,
) -> String {
let mut out = String::new();
out.push_str(&hash::header(CommentStyle::DoubleSlash));
let _ = writeln!(out, "import gleeunit");
let _ = writeln!(out, "import gleeunit/should");
let has_http_fixtures = fixtures.iter().any(|f| f.is_http_test());
if has_http_fixtures {
let _ = writeln!(out, "import gleam/httpc");
let _ = writeln!(out, "import gleam/http");
let _ = writeln!(out, "import gleam/http/request");
let _ = writeln!(out, "import gleam/list");
let _ = writeln!(out, "import gleam/result");
let _ = writeln!(out, "import gleam/string");
let _ = writeln!(out, "import envoy");
}
let has_non_http_with_override = fixtures.iter().any(|f| !f.is_http_test());
if has_non_http_with_override {
let _ = writeln!(out, "import {module_path}");
let _ = writeln!(out, "import e2e_gleam");
}
let _ = writeln!(out);
let mut needed_modules: std::collections::BTreeSet<&'static str> = std::collections::BTreeSet::new();
for fixture in fixtures {
if fixture.is_http_test() {
continue; }
let call_config = e2e_config.resolve_call_for_fixture(fixture.call.as_deref(), &fixture.input);
let has_bytes_arg = call_config.args.iter().any(|a| a.arg_type == "bytes");
let has_optional_string_arg = call_config.args.iter().any(|a| a.arg_type == "string" && a.optional);
let has_json_object_arg = call_config.args.iter().any(|a| a.arg_type == "json_object");
if has_bytes_arg || has_optional_string_arg || has_json_object_arg {
needed_modules.insert("option");
}
for assertion in &fixture.assertions {
let needs_case_expr = assertion
.field
.as_deref()
.is_some_and(|f| field_resolver.tagged_union_split(f).is_some());
if needs_case_expr {
needed_modules.insert("option");
}
if let Some(f) = &assertion.field {
if field_resolver.is_optional(f) {
needed_modules.insert("option");
}
}
match assertion.assertion_type.as_str() {
"contains_any" => {
needed_modules.insert("string");
needed_modules.insert("list");
}
"contains" | "contains_all" | "not_contains" | "starts_with" | "ends_with" => {
needed_modules.insert("string");
if let Some(f) = &assertion.field {
let resolved = field_resolver.resolve(f);
if field_resolver.is_array(f) || field_resolver.is_array(resolved) {
needed_modules.insert("list");
}
} else {
if call_config.result_is_array
|| call_config.result_is_vec
|| field_resolver.is_array("")
|| field_resolver.is_array(field_resolver.resolve(""))
{
needed_modules.insert("list");
}
}
}
"not_empty" | "is_empty" | "count_min" | "count_equals" => {
needed_modules.insert("list");
}
"min_length" | "max_length" => {
needed_modules.insert("string");
}
"greater_than" | "less_than" | "greater_than_or_equal" | "less_than_or_equal" => {
}
_ => {}
}
if needs_case_expr {
if let Some(f) = &assertion.field {
let resolved = field_resolver.resolve(f);
if field_resolver.is_array(resolved) {
needed_modules.insert("list");
}
}
}
if let Some(f) = &assertion.field {
if !f.is_empty() {
let parts: Vec<&str> = f.split('.').collect();
let has_opt_prefix = (1..parts.len()).any(|i| {
let prefix_path = parts[..i].join(".");
field_resolver.is_optional(&prefix_path)
});
if has_opt_prefix {
needed_modules.insert("option");
}
}
}
}
}
for module in &needed_modules {
let _ = writeln!(out, "import gleam/{module}");
}
if !needed_modules.is_empty() {
let _ = writeln!(out);
}
for fixture in fixtures {
if fixture.is_http_test() {
render_http_test_case(&mut out, fixture);
} else {
render_test_case(
&mut out,
fixture,
e2e_config,
module_path,
function_name,
result_var,
args,
field_resolver,
enum_fields,
element_constructors,
json_object_wrapper,
);
}
let _ = writeln!(out);
}
out
}
struct GleamTestClientRenderer;
impl client::TestClientRenderer for GleamTestClientRenderer {
fn language_name(&self) -> &'static str {
"gleam"
}
fn sanitize_test_name(&self, id: &str) -> String {
let raw = sanitize_ident(id);
let stripped = raw.trim_start_matches(|c: char| c == '_' || c.is_ascii_digit());
if stripped.is_empty() { raw } else { stripped.to_string() }
}
fn render_test_open(&self, out: &mut String, fn_name: &str, description: &str, skip_reason: Option<&str>) {
let _ = writeln!(out, "// {description}");
let _ = writeln!(out, "pub fn {fn_name}_test() {{");
if let Some(reason) = skip_reason {
let escaped = escape_gleam(reason);
let _ = writeln!(out, " // skipped: {escaped}");
let _ = writeln!(out, " Nil");
}
}
fn render_test_close(&self, out: &mut String) {
let _ = writeln!(out, "}}");
}
fn render_call(&self, out: &mut String, ctx: &client::CallCtx<'_>) {
let path = ctx.path;
let _ = writeln!(out, " let base_url = case envoy.get(\"MOCK_SERVER_URL\") {{");
let _ = writeln!(out, " Ok(u) -> u");
let _ = writeln!(out, " Error(_) -> \"http://localhost:8080\"");
let _ = writeln!(out, " }}");
let _ = writeln!(out, " let assert Ok(req) = request.to(base_url <> \"{path}\")");
let method_const = match ctx.method.to_uppercase().as_str() {
"GET" => "Get",
"POST" => "Post",
"PUT" => "Put",
"DELETE" => "Delete",
"PATCH" => "Patch",
"HEAD" => "Head",
"OPTIONS" => "Options",
_ => "Post",
};
let _ = writeln!(out, " let req = request.set_method(req, http.{method_const})");
if ctx.body.is_some() {
let content_type = ctx.content_type.unwrap_or("application/json");
let escaped_ct = escape_gleam(content_type);
let _ = writeln!(
out,
" let req = request.set_header(req, \"content-type\", \"{escaped_ct}\")"
);
}
for (name, value) in ctx.headers {
let lower = name.to_lowercase();
if matches!(lower.as_str(), "content-length" | "host" | "transfer-encoding") {
continue;
}
let escaped_name = escape_gleam(name);
let escaped_value = escape_gleam(value);
let _ = writeln!(
out,
" let req = request.set_header(req, \"{escaped_name}\", \"{escaped_value}\")"
);
}
if !ctx.cookies.is_empty() {
let cookie_str: Vec<String> = ctx.cookies.iter().map(|(k, v)| format!("{k}={v}")).collect();
let escaped_cookie = escape_gleam(&cookie_str.join("; "));
let _ = writeln!(
out,
" let req = request.set_header(req, \"cookie\", \"{escaped_cookie}\")"
);
}
if let Some(body) = ctx.body {
let json_str = serde_json::to_string(body).unwrap_or_default();
let escaped = escape_gleam(&json_str);
let _ = writeln!(out, " let req = request.set_body(req, \"{escaped}\")");
}
let resp = ctx.response_var;
let _ = writeln!(out, " let assert Ok({resp}) = httpc.send(req)");
}
fn render_assert_status(&self, out: &mut String, response_var: &str, status: u16) {
let _ = writeln!(out, " {response_var}.status |> should.equal({status})");
}
fn render_assert_header(&self, out: &mut String, response_var: &str, name: &str, expected: &str) {
let escaped_name = escape_gleam(&name.to_lowercase());
match expected {
"<<absent>>" => {
let _ = writeln!(
out,
" {response_var}.headers\n |> list.find(fn(h: #(String, String)) {{ h.0 == \"{escaped_name}\" }})\n |> result.is_ok()\n |> should.be_false()"
);
}
"<<present>>" | "<<uuid>>" => {
let _ = writeln!(
out,
" {response_var}.headers\n |> list.find(fn(h: #(String, String)) {{ h.0 == \"{escaped_name}\" }})\n |> result.is_ok()\n |> should.be_true()"
);
}
literal => {
let _escaped_value = escape_gleam(literal);
let _ = writeln!(
out,
" {response_var}.headers\n |> list.find(fn(h: #(String, String)) {{ h.0 == \"{escaped_name}\" }})\n |> result.is_ok()\n |> should.be_true()"
);
}
}
}
fn render_assert_json_body(&self, out: &mut String, response_var: &str, expected: &serde_json::Value) {
let escaped = match expected {
serde_json::Value::String(s) => escape_gleam(s),
other => escape_gleam(&serde_json::to_string(other).unwrap_or_default()),
};
let _ = writeln!(
out,
" {response_var}.body |> string.trim |> should.equal(\"{escaped}\")"
);
}
fn render_assert_partial_body(&self, out: &mut String, response_var: &str, expected: &serde_json::Value) {
if let Some(obj) = expected.as_object() {
for (key, val) in obj {
let fragment = escape_gleam(&format!("\"{}\":", key));
let _ = writeln!(
out,
" {response_var}.body |> string.contains(\"{fragment}\") |> should.equal(True)"
);
let _ = val; }
}
}
fn render_assert_validation_errors(
&self,
out: &mut String,
response_var: &str,
errors: &[ValidationErrorExpectation],
) {
for err in errors {
let escaped_msg = escape_gleam(&err.msg);
let _ = writeln!(
out,
" {response_var}.body |> string.contains(\"{escaped_msg}\") |> should.equal(True)"
);
}
}
}
fn render_http_test_case(out: &mut String, fixture: &Fixture) {
client::http_call::render_http_test(out, &GleamTestClientRenderer, fixture);
}
#[allow(clippy::too_many_arguments)]
fn render_test_case(
out: &mut String,
fixture: &Fixture,
e2e_config: &E2eConfig,
module_path: &str,
_function_name: &str,
_result_var: &str,
_args: &[crate::config::ArgMapping],
field_resolver: &FieldResolver,
enum_fields: &HashSet<String>,
element_constructors: &[alef_core::config::GleamElementConstructor],
json_object_wrapper: Option<&str>,
) {
let call_config = e2e_config.resolve_call_for_fixture(fixture.call.as_deref(), &fixture.input);
let lang = "gleam";
let call_overrides = call_config.overrides.get(lang);
let function_name = call_overrides
.and_then(|o| o.function.as_ref())
.cloned()
.unwrap_or_else(|| call_config.function.clone());
let result_var = &call_config.result_var;
let args = &call_config.args;
let raw_name = sanitize_ident(&fixture.id);
let stripped = raw_name.trim_start_matches(|c: char| c == '_' || c.is_ascii_digit());
let test_name = if stripped.is_empty() {
raw_name.as_str()
} else {
stripped
};
let description = &fixture.description;
let expects_error = fixture.assertions.iter().any(|a| a.assertion_type == "error");
let test_documents_path = e2e_config.test_documents_relative_from(0);
let (setup_lines, args_str) = build_args_and_setup(
&fixture.input,
args,
&fixture.id,
&test_documents_path,
element_constructors,
json_object_wrapper,
);
let _ = writeln!(out, "// {description}");
let _ = writeln!(out, "pub fn {test_name}_test() {{");
for line in &setup_lines {
let _ = writeln!(out, " {line}");
}
if expects_error {
let _ = writeln!(out, " {module_path}.{function_name}({args_str}) |> should.be_error()");
let _ = writeln!(out, "}}");
return;
}
let _ = writeln!(out, " let {result_var} = {module_path}.{function_name}({args_str})");
let _ = writeln!(out, " {result_var} |> should.be_ok()");
let _ = writeln!(out, " let assert Ok(r) = {result_var}");
let result_is_array = call_config.result_is_array || call_config.result_is_vec;
let pkg_module = e2e_config
.resolve_package("gleam")
.as_ref()
.and_then(|p| p.name.as_ref())
.cloned()
.unwrap_or_else(|| {
module_path
.split('.')
.next()
.unwrap_or(module_path)
.to_string()
});
for assertion in &fixture.assertions {
render_assertion(
out,
assertion,
"r",
field_resolver,
enum_fields,
result_is_array,
&pkg_module,
);
}
let _ = writeln!(out, "}}");
}
fn build_args_and_setup(
input: &serde_json::Value,
args: &[crate::config::ArgMapping],
_fixture_id: &str,
test_documents_path: &str,
element_constructors: &[alef_core::config::GleamElementConstructor],
json_object_wrapper: Option<&str>,
) -> (Vec<String>, String) {
if args.is_empty() {
return (Vec::new(), String::new());
}
let mut setup_lines: Vec<String> = Vec::new();
let mut parts: Vec<String> = Vec::new();
let mut bytes_var_counter = 0usize;
for arg in args {
let field = arg.field.strip_prefix("input.").unwrap_or(&arg.field);
let val = input.get(field);
match arg.arg_type.as_str() {
"file_path" => {
let path = val.and_then(|v| v.as_str()).unwrap_or("");
let full_path = format!("{test_documents_path}/{path}");
parts.push(format!("\"{}\"", escape_gleam(&full_path)));
}
"bytes" => {
let path = val.and_then(|v| v.as_str()).unwrap_or("");
let var_name = if bytes_var_counter == 0 {
"data_bytes__".to_string()
} else {
format!("data_bytes_{bytes_var_counter}__")
};
bytes_var_counter += 1;
let full_path = format!("{test_documents_path}/{path}");
setup_lines.push(format!(
"let assert Ok({var_name}) = e2e_gleam.read_file_bytes(\"{}\")",
escape_gleam(&full_path)
));
parts.push(var_name);
}
"string" if arg.optional => {
match val {
None | Some(serde_json::Value::Null) => {
parts.push("option.None".to_string());
}
Some(serde_json::Value::String(s)) if s.is_empty() => {
parts.push("option.None".to_string());
}
Some(serde_json::Value::String(s)) => {
parts.push(format!("option.Some(\"{}\")", escape_gleam(s)));
}
Some(v) => {
parts.push(format!("option.Some({})", json_to_gleam(v)));
}
}
}
"string" => {
match val {
None | Some(serde_json::Value::Null) => {
parts.push("\"\"".to_string());
}
Some(serde_json::Value::String(s)) => {
parts.push(format!("\"{}\"", escape_gleam(s)));
}
Some(v) => {
parts.push(json_to_gleam(v));
}
}
}
"json_object" => {
let element_type = arg.element_type.as_deref().unwrap_or("");
let recipe = if element_type.is_empty() {
None
} else {
element_constructors
.iter()
.find(|r| r.element_type == element_type)
};
if let Some(recipe) = recipe {
let items_expr = match val {
Some(serde_json::Value::Array(arr)) => {
let items: Vec<String> = arr
.iter()
.map(|item| {
render_gleam_element_constructor(item, recipe, test_documents_path)
})
.collect();
format!("[{}]", items.join(", "))
}
_ => "[]".to_string(),
};
if arg.optional && (val.is_none() || val == Some(&serde_json::Value::Null)) {
parts.push("[]".to_string());
} else {
parts.push(items_expr);
}
} else if arg.optional && (val.is_none() || val == Some(&serde_json::Value::Null)) {
parts.push("option.None".to_string());
} else {
let empty_obj = serde_json::Value::Object(Default::default());
let config_val = val.unwrap_or(&empty_obj);
let json_literal = json_to_gleam(config_val);
let emitted = match json_object_wrapper {
Some(template) => template.replace("{json}", &json_literal),
None => json_literal,
};
parts.push(emitted);
}
}
"int" | "integer" => match val {
None | Some(serde_json::Value::Null) if arg.optional => {}
None | Some(serde_json::Value::Null) => parts.push("0".to_string()),
Some(v) => parts.push(json_to_gleam(v)),
},
"bool" | "boolean" => match val {
Some(serde_json::Value::Bool(true)) => parts.push("True".to_string()),
Some(serde_json::Value::Bool(false)) | None | Some(serde_json::Value::Null) => {
if !arg.optional {
parts.push("False".to_string());
}
}
Some(v) => parts.push(json_to_gleam(v)),
},
_ => {
match val {
None | Some(serde_json::Value::Null) if arg.optional => {}
None | Some(serde_json::Value::Null) => parts.push("Nil".to_string()),
Some(v) => parts.push(json_to_gleam(v)),
}
}
}
}
(setup_lines, parts.join(", "))
}
fn render_gleam_element_constructor(
item: &serde_json::Value,
recipe: &alef_core::config::GleamElementConstructor,
test_documents_path: &str,
) -> String {
let mut field_exprs: Vec<String> = Vec::with_capacity(recipe.fields.len());
for field in &recipe.fields {
let expr = match field.kind.as_str() {
"file_path" => {
let json_field = field.json_field.as_deref().unwrap_or("");
let path = item.get(json_field).and_then(|v| v.as_str()).unwrap_or("");
let full = if path.starts_with('/') {
path.to_string()
} else {
format!("{test_documents_path}/{path}")
};
format!("\"{}\"", escape_gleam(&full))
}
"byte_array" => {
let json_field = field.json_field.as_deref().unwrap_or("");
let bytes: Vec<String> = item
.get(json_field)
.and_then(|v| v.as_array())
.map(|arr| arr.iter().map(|b| b.as_u64().unwrap_or(0).to_string()).collect())
.unwrap_or_default();
if bytes.is_empty() {
"<<>>".to_string()
} else {
format!("<<{}>>", bytes.join(", "))
}
}
"string" => {
let json_field = field.json_field.as_deref().unwrap_or("");
let value = item
.get(json_field)
.and_then(|v| v.as_str())
.map(str::to_string)
.or_else(|| field.default.clone())
.unwrap_or_default();
format!("\"{}\"", escape_gleam(&value))
}
"literal" => field.value.clone().unwrap_or_default(),
other => {
field.value.clone().unwrap_or_else(|| format!("\"<unsupported kind: {other}>\""))
}
};
field_exprs.push(format!("{}: {}", field.gleam_field, expr));
}
format!("{}({})", recipe.constructor, field_exprs.join(", "))
}
fn render_tagged_union_assertion(
out: &mut String,
assertion: &Assertion,
result_var: &str,
prefix: &str,
variant: &str,
suffix: &str,
field_resolver: &FieldResolver,
pkg_module: &str,
) {
let prefix_expr = if prefix.is_empty() {
result_var.to_string()
} else {
format!("{result_var}.{prefix}")
};
let constructor = variant.to_pascal_case();
let module_qualifier = pkg_module;
let inner_var = "fmt_inner__";
let full_suffix_path = if prefix.is_empty() {
format!("{variant}.{suffix}")
} else {
format!("{prefix}.{variant}.{suffix}")
};
let suffix_is_optional = field_resolver.is_optional(&full_suffix_path);
let suffix_is_array = field_resolver.is_array(&full_suffix_path);
let _ = writeln!(out, " case {prefix_expr} {{");
let _ = writeln!(
out,
" option.Some({module_qualifier}.{constructor}({inner_var})) -> {{"
);
let inner_field_expr = if suffix.is_empty() {
inner_var.to_string()
} else {
format!("{inner_var}.{suffix}")
};
match assertion.assertion_type.as_str() {
"equals" => {
if let Some(expected) = &assertion.value {
let gleam_val = json_to_gleam(expected);
if suffix_is_optional {
let default = default_gleam_value_for_optional(&gleam_val);
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap({default}) |> should.equal({gleam_val})"
);
} else {
let _ = writeln!(out, " {inner_field_expr} |> should.equal({gleam_val})");
}
}
}
"contains" => {
if let Some(expected) = &assertion.value {
let gleam_val = json_to_gleam(expected);
if suffix_is_array {
let _ = writeln!(out, " let items__ = {inner_field_expr} |> option.unwrap([])");
let _ = writeln!(
out,
" items__ |> list.any(fn(item__) {{ string.contains(item__, {gleam_val}) }}) |> should.equal(True)"
);
} else if suffix_is_optional {
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap(\"\") |> string.contains({gleam_val}) |> should.equal(True)"
);
} else {
let _ = writeln!(
out,
" {inner_field_expr} |> string.contains({gleam_val}) |> should.equal(True)"
);
}
}
}
"contains_all" => {
if let Some(values) = &assertion.values {
if suffix_is_array {
let _ = writeln!(out, " let items__ = {inner_field_expr} |> option.unwrap([])");
for val in values {
let gleam_val = json_to_gleam(val);
let _ = writeln!(
out,
" items__ |> list.any(fn(item__) {{ string.contains(item__, {gleam_val}) }}) |> should.equal(True)"
);
}
} else if suffix_is_optional {
for val in values {
let gleam_val = json_to_gleam(val);
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap(\"\") |> string.contains({gleam_val}) |> should.equal(True)"
);
}
} else {
for val in values {
let gleam_val = json_to_gleam(val);
let _ = writeln!(
out,
" {inner_field_expr} |> string.contains({gleam_val}) |> should.equal(True)"
);
}
}
}
}
"greater_than_or_equal" => {
if let Some(val) = &assertion.value {
let gleam_val = json_to_gleam(val);
if suffix_is_optional {
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap(0) |> fn(n__) {{ n__ >= {gleam_val} }} |> should.equal(True)"
);
} else {
let _ = writeln!(
out,
" {inner_field_expr} |> fn(n__) {{ n__ >= {gleam_val} }} |> should.equal(True)"
);
}
}
}
"greater_than" => {
if let Some(val) = &assertion.value {
let gleam_val = json_to_gleam(val);
if suffix_is_optional {
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap(0) |> fn(n__) {{ n__ > {gleam_val} }} |> should.equal(True)"
);
} else {
let _ = writeln!(
out,
" {inner_field_expr} |> fn(n__) {{ n__ > {gleam_val} }} |> should.equal(True)"
);
}
}
}
"less_than" => {
if let Some(val) = &assertion.value {
let gleam_val = json_to_gleam(val);
if suffix_is_optional {
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap(0) |> fn(n__) {{ n__ < {gleam_val} }} |> should.equal(True)"
);
} else {
let _ = writeln!(
out,
" {inner_field_expr} |> fn(n__) {{ n__ < {gleam_val} }} |> should.equal(True)"
);
}
}
}
"less_than_or_equal" => {
if let Some(val) = &assertion.value {
let gleam_val = json_to_gleam(val);
if suffix_is_optional {
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap(0) |> fn(n__) {{ n__ <= {gleam_val} }} |> should.equal(True)"
);
} else {
let _ = writeln!(
out,
" {inner_field_expr} |> fn(n__) {{ n__ <= {gleam_val} }} |> should.equal(True)"
);
}
}
}
"count_min" => {
if let Some(val) = &assertion.value {
if let Some(n) = val.as_u64() {
if suffix_is_optional {
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap([]) |> list.length |> fn(n__) {{ n__ >= {n} }} |> should.equal(True)"
);
} else {
let _ = writeln!(
out,
" {inner_field_expr} |> list.length |> fn(n__) {{ n__ >= {n} }} |> should.equal(True)"
);
}
}
}
}
"count_equals" => {
if let Some(val) = &assertion.value {
if let Some(n) = val.as_u64() {
if suffix_is_optional {
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap([]) |> list.length |> should.equal({n})"
);
} else {
let _ = writeln!(out, " {inner_field_expr} |> list.length |> should.equal({n})");
}
}
}
}
"not_empty" => {
if suffix_is_optional {
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap([]) |> list.is_empty |> should.equal(False)"
);
} else {
let _ = writeln!(out, " {inner_field_expr} |> list.is_empty |> should.equal(False)");
}
}
"is_empty" => {
if suffix_is_optional {
let _ = writeln!(
out,
" {inner_field_expr} |> option.unwrap([]) |> list.is_empty |> should.equal(True)"
);
} else {
let _ = writeln!(out, " {inner_field_expr} |> list.is_empty |> should.equal(True)");
}
}
"is_true" => {
let _ = writeln!(out, " {inner_field_expr} |> should.equal(True)");
}
"is_false" => {
let _ = writeln!(out, " {inner_field_expr} |> should.equal(False)");
}
other => {
let _ = writeln!(
out,
" // tagged-union assertion '{other}' not yet implemented for Gleam"
);
}
}
let _ = writeln!(out, " }}");
let _ = writeln!(
out,
" _ -> panic as \"expected {module_qualifier}.{constructor} format metadata\""
);
let _ = writeln!(out, " }}");
}
fn default_gleam_value_for_optional(gleam_val: &str) -> &'static str {
if gleam_val.starts_with('"') {
"\"\""
} else if gleam_val == "True" || gleam_val == "False" {
"False"
} else if gleam_val.contains('.') {
"0.0"
} else {
"0"
}
}
fn render_assertion(
out: &mut String,
assertion: &Assertion,
result_var: &str,
field_resolver: &FieldResolver,
enum_fields: &HashSet<String>,
result_is_array: bool,
pkg_module: &str,
) {
if let Some(f) = &assertion.field {
if !f.is_empty() && !field_resolver.is_valid_for_result(f) {
let _ = writeln!(out, " // skipped: field '{f}' not available on result type");
return;
}
}
if let Some(f) = &assertion.field {
if !f.is_empty() {
if let Some((prefix, variant, suffix)) = field_resolver.tagged_union_split(f) {
render_tagged_union_assertion(
out,
assertion,
result_var,
&prefix,
&variant,
&suffix,
field_resolver,
pkg_module,
);
return;
}
}
}
if let Some(f) = &assertion.field {
if !f.is_empty() {
let parts: Vec<&str> = f.split('.').collect();
let mut opt_prefix: Option<(String, usize)> = None;
for i in 1..parts.len() {
let prefix_path = parts[..i].join(".");
if field_resolver.is_optional(&prefix_path) {
opt_prefix = Some((prefix_path, i));
break;
}
}
if let Some((optional_prefix, suffix_start)) = opt_prefix {
let prefix_expr = format!("{result_var}.{optional_prefix}");
let suffix_parts = &parts[suffix_start..];
let suffix_str = suffix_parts.join(".");
let inner_var = "opt_inner__";
let inner_expr = if suffix_str.is_empty() {
inner_var.to_string()
} else {
format!("{inner_var}.{suffix_str}")
};
let _ = writeln!(out, " case {prefix_expr} {{");
let _ = writeln!(out, " option.Some({inner_var}) -> {{");
match assertion.assertion_type.as_str() {
"count_min" => {
if let Some(val) = &assertion.value {
if let Some(n) = val.as_u64() {
let _ = writeln!(
out,
" {inner_expr} |> list.length |> fn(n__) {{ n__ >= {n} }} |> should.equal(True)"
);
}
}
}
"count_equals" => {
if let Some(val) = &assertion.value {
if let Some(n) = val.as_u64() {
let _ = writeln!(out, " {inner_expr} |> list.length |> should.equal({n})");
}
}
}
"not_empty" => {
let _ = writeln!(out, " {inner_expr} |> list.is_empty |> should.equal(False)");
}
"min_length" => {
if let Some(val) = &assertion.value {
if let Some(n) = val.as_u64() {
let _ = writeln!(
out,
" {inner_expr} |> string.length |> fn(n__) {{ n__ >= {n} }} |> should.equal(True)"
);
}
}
}
other => {
let _ = writeln!(
out,
" // optional-prefix assertion '{other}' not yet implemented for Gleam"
);
}
}
let _ = writeln!(out, " }}");
let _ = writeln!(out, " option.None -> should.fail()");
let _ = writeln!(out, " }}");
return;
}
}
}
let field_is_optional = assertion
.field
.as_deref()
.is_some_and(|f| !f.is_empty() && field_resolver.is_optional(f));
let _field_is_enum = assertion
.field
.as_deref()
.is_some_and(|f| enum_fields.contains(f) || enum_fields.contains(field_resolver.resolve(f)));
let field_expr = match &assertion.field {
Some(f) if !f.is_empty() => field_resolver.accessor(f, "gleam", result_var),
_ => result_var.to_string(),
};
let field_is_array = {
let f = assertion.field.as_deref().unwrap_or("");
let is_root = f.is_empty();
(is_root && result_is_array) || field_resolver.is_array(f) || field_resolver.is_array(field_resolver.resolve(f))
};
match assertion.assertion_type.as_str() {
"equals" => {
if let Some(expected) = &assertion.value {
let gleam_val = json_to_gleam(expected);
if field_is_optional {
let _ = writeln!(out, " {field_expr} |> should.equal(option.Some({gleam_val}))");
} else {
let _ = writeln!(out, " {field_expr} |> should.equal({gleam_val})");
}
}
}
"contains" => {
if let Some(expected) = &assertion.value {
let gleam_val = json_to_gleam(expected);
if field_is_array {
let _ = writeln!(
out,
" {field_expr} |> list.any(fn(item__) {{ string.contains(item__, {gleam_val}) }}) |> should.equal(True)"
);
} else if field_is_optional {
let _ = writeln!(
out,
" {field_expr} |> option.unwrap(\"\") |> string.contains({gleam_val}) |> should.equal(True)"
);
} else {
let _ = writeln!(
out,
" {field_expr} |> string.contains({gleam_val}) |> should.equal(True)"
);
}
}
}
"contains_all" => {
if let Some(values) = &assertion.values {
for val in values {
let gleam_val = json_to_gleam(val);
if field_is_optional {
let _ = writeln!(
out,
" {field_expr} |> option.unwrap(\"\") |> string.contains({gleam_val}) |> should.equal(True)"
);
} else {
let _ = writeln!(
out,
" {field_expr} |> string.contains({gleam_val}) |> should.equal(True)"
);
}
}
}
}
"not_contains" => {
if let Some(expected) = &assertion.value {
let gleam_val = json_to_gleam(expected);
let _ = writeln!(
out,
" {field_expr} |> string.contains({gleam_val}) |> should.equal(False)"
);
}
}
"not_empty" => {
if field_is_optional {
let _ = writeln!(out, " {field_expr} |> option.is_some |> should.equal(True)");
} else {
let _ = writeln!(out, " {field_expr} |> list.is_empty |> should.equal(False)");
}
}
"is_empty" => {
if field_is_optional {
let _ = writeln!(out, " {field_expr} |> option.is_none |> should.equal(True)");
} else {
let _ = writeln!(out, " {field_expr} |> list.is_empty |> should.equal(True)");
}
}
"starts_with" => {
if let Some(expected) = &assertion.value {
let gleam_val = json_to_gleam(expected);
let _ = writeln!(
out,
" {field_expr} |> string.starts_with({gleam_val}) |> should.equal(True)"
);
}
}
"ends_with" => {
if let Some(expected) = &assertion.value {
let gleam_val = json_to_gleam(expected);
let _ = writeln!(
out,
" {field_expr} |> string.ends_with({gleam_val}) |> should.equal(True)"
);
}
}
"min_length" => {
if let Some(val) = &assertion.value {
if let Some(n) = val.as_u64() {
let _ = writeln!(
out,
" {field_expr} |> string.length |> fn(n__) {{ n__ >= {n} }} |> should.equal(True)"
);
}
}
}
"max_length" => {
if let Some(val) = &assertion.value {
if let Some(n) = val.as_u64() {
let _ = writeln!(
out,
" {field_expr} |> string.length |> fn(n__) {{ n__ <= {n} }} |> should.equal(True)"
);
}
}
}
"count_min" => {
if let Some(val) = &assertion.value {
if let Some(n) = val.as_u64() {
let _ = writeln!(
out,
" {field_expr} |> list.length |> fn(n__) {{ n__ >= {n} }} |> should.equal(True)"
);
}
}
}
"count_equals" => {
if let Some(val) = &assertion.value {
if let Some(n) = val.as_u64() {
let _ = writeln!(out, " {field_expr} |> list.length |> should.equal({n})");
}
}
}
"is_true" => {
let _ = writeln!(out, " {field_expr} |> should.equal(True)");
}
"is_false" => {
let _ = writeln!(out, " {field_expr} |> should.equal(False)");
}
"not_error" => {
}
"error" => {
}
"greater_than" => {
if let Some(val) = &assertion.value {
let gleam_val = json_to_gleam(val);
let _ = writeln!(
out,
" {field_expr} |> fn(n__) {{ n__ > {gleam_val} }} |> should.equal(True)"
);
}
}
"less_than" => {
if let Some(val) = &assertion.value {
let gleam_val = json_to_gleam(val);
let _ = writeln!(
out,
" {field_expr} |> fn(n__) {{ n__ < {gleam_val} }} |> should.equal(True)"
);
}
}
"greater_than_or_equal" => {
if let Some(val) = &assertion.value {
let gleam_val = json_to_gleam(val);
let _ = writeln!(
out,
" {field_expr} |> fn(n__) {{ n__ >= {gleam_val} }} |> should.equal(True)"
);
}
}
"less_than_or_equal" => {
if let Some(val) = &assertion.value {
let gleam_val = json_to_gleam(val);
let _ = writeln!(
out,
" {field_expr} |> fn(n__) {{ n__ <= {gleam_val} }} |> should.equal(True)"
);
}
}
"contains_any" => {
if let Some(values) = &assertion.values {
let vals_list = values.iter().map(json_to_gleam).collect::<Vec<_>>().join(", ");
let _ = writeln!(
out,
" [{vals_list}] |> list.any(fn(v__) {{ string.contains({field_expr}, v__) }}) |> should.equal(True)"
);
}
}
"matches_regex" => {
let _ = writeln!(out, " // regex match not yet implemented for Gleam");
}
"method_result" => {
let _ = writeln!(out, " // method_result assertions not yet implemented for Gleam");
}
other => {
panic!("Gleam e2e generator: unsupported assertion type: {other}");
}
}
}
fn json_to_gleam(value: &serde_json::Value) -> String {
match value {
serde_json::Value::String(s) => format!("\"{}\"", escape_gleam(s)),
serde_json::Value::Bool(b) => {
if *b {
"True".to_string()
} else {
"False".to_string()
}
}
serde_json::Value::Number(n) => n.to_string(),
serde_json::Value::Null => "Nil".to_string(),
serde_json::Value::Array(arr) => {
let items: Vec<String> = arr.iter().map(json_to_gleam).collect();
format!("[{}]", items.join(", "))
}
serde_json::Value::Object(_) => {
let json_str = serde_json::to_string(value).unwrap_or_default();
format!("\"{}\"", escape_gleam(&json_str))
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use alef_core::config::{GleamElementConstructor, GleamElementField};
fn batch_file_item_recipe() -> GleamElementConstructor {
GleamElementConstructor {
element_type: "BatchFileItem".to_string(),
constructor: "kreuzberg.BatchFileItem".to_string(),
fields: vec![
GleamElementField {
gleam_field: "path".to_string(),
kind: "file_path".to_string(),
json_field: Some("path".to_string()),
default: None,
value: None,
},
GleamElementField {
gleam_field: "config".to_string(),
kind: "literal".to_string(),
json_field: None,
default: None,
value: Some("option.None".to_string()),
},
],
}
}
#[test]
fn render_element_constructor_file_path_relative_path_gets_test_documents_prefix() {
let item = serde_json::json!({ "path": "docx/fake.docx" });
let out = render_gleam_element_constructor(&item, &batch_file_item_recipe(), "../../test_documents");
assert_eq!(
out,
"kreuzberg.BatchFileItem(path: \"../../test_documents/docx/fake.docx\", config: option.None)"
);
}
#[test]
fn render_element_constructor_file_path_absolute_path_passes_through() {
let item = serde_json::json!({ "path": "/etc/some/absolute" });
let out = render_gleam_element_constructor(&item, &batch_file_item_recipe(), "../../test_documents");
assert!(
out.contains("\"/etc/some/absolute\""),
"absolute paths must NOT receive the test_documents prefix; got:\n{out}"
);
}
#[test]
fn render_element_constructor_byte_array_emits_bitarray() {
let recipe = GleamElementConstructor {
element_type: "BatchBytesItem".to_string(),
constructor: "kreuzberg.BatchBytesItem".to_string(),
fields: vec![
GleamElementField {
gleam_field: "content".to_string(),
kind: "byte_array".to_string(),
json_field: Some("content".to_string()),
default: None,
value: None,
},
GleamElementField {
gleam_field: "mime_type".to_string(),
kind: "string".to_string(),
json_field: Some("mime_type".to_string()),
default: Some("text/plain".to_string()),
value: None,
},
GleamElementField {
gleam_field: "config".to_string(),
kind: "literal".to_string(),
json_field: None,
default: None,
value: Some("option.None".to_string()),
},
],
};
let item = serde_json::json!({ "content": [72, 105], "mime_type": "text/html" });
let out = render_gleam_element_constructor(&item, &recipe, "../../test_documents");
assert_eq!(
out,
"kreuzberg.BatchBytesItem(content: <<72, 105>>, mime_type: \"text/html\", config: option.None)"
);
}
#[test]
fn build_args_with_json_object_wrapper_substitutes_placeholder() {
use crate::config::ArgMapping;
let arg = ArgMapping {
name: "config".to_string(),
field: "config".to_string(),
arg_type: "json_object".to_string(),
optional: false,
owned: false,
element_type: None,
go_type: None,
};
let input = serde_json::json!({
"config": { "use_cache": true, "force_ocr": false }
});
let (_setup, args_str) = build_args_and_setup(
&input,
&[arg],
"test_fixture",
"../../test_documents",
&[],
Some("k.config_from_json_string({json})"),
);
assert!(
args_str.starts_with("k.config_from_json_string("),
"wrapper must envelop the JSON literal; got:\n{args_str}"
);
assert!(
args_str.contains("use_cache"),
"JSON payload must reach the wrapper; got:\n{args_str}"
);
}
#[test]
fn build_args_without_json_object_wrapper_emits_bare_json_string() {
use crate::config::ArgMapping;
let arg = ArgMapping {
name: "config".to_string(),
field: "config".to_string(),
arg_type: "json_object".to_string(),
optional: false,
owned: false,
element_type: None,
go_type: None,
};
let input = serde_json::json!({ "config": { "x": 1 } });
let (_setup, args_str) = build_args_and_setup(
&input,
&[arg],
"test_fixture",
"../../test_documents",
&[],
None,
);
assert!(
!args_str.contains("from_json_string"),
"no wrapper configured must not synthesise one; got:\n{args_str}"
);
assert!(
args_str.starts_with('"'),
"bare emission is a Gleam string literal starting with a quote; got:\n{args_str}"
);
}
#[test]
fn render_element_constructor_string_falls_back_to_default() {
let recipe = GleamElementConstructor {
element_type: "BatchBytesItem".to_string(),
constructor: "k.BatchBytesItem".to_string(),
fields: vec![GleamElementField {
gleam_field: "mime_type".to_string(),
kind: "string".to_string(),
json_field: Some("mime_type".to_string()),
default: Some("text/plain".to_string()),
value: None,
}],
};
let item = serde_json::json!({});
let out = render_gleam_element_constructor(&item, &recipe, "../../test_documents");
assert!(
out.contains("mime_type: \"text/plain\""),
"missing string field must fall back to default; got:\n{out}"
);
}
}