use crate::core::config::ResolvedCrateConfig;
use crate::core::ir::{EnumDef, EnumVariant, FieldDef, FunctionDef, TypeDef, TypeRef};
use crate::e2e::config::{CallConfig, E2eConfig};
use crate::e2e::fixture::{Assertion, Fixture};
use super::test_case::render_test_case;
const ENUM_MARKER: &str = "MyLib.DataNodeKind.wire_value(result.kind)";
const RAW_MARKER: &str = "assert result.kind == \"KeyValue\"";
fn data_node_kind_enum() -> EnumDef {
EnumDef {
name: "DataNodeKind".to_string(),
variants: vec![
EnumVariant {
name: "KeyValue".to_string(),
..EnumVariant::default()
},
EnumVariant {
name: "Sequence".to_string(),
..EnumVariant::default()
},
],
..EnumDef::default()
}
}
fn kind_field(ty: TypeRef, optional: bool) -> FieldDef {
FieldDef {
name: "kind".to_string(),
ty,
optional,
..FieldDef::default()
}
}
fn table_ir() -> (Vec<TypeDef>, Vec<EnumDef>, Vec<FunctionDef>) {
let type_defs = vec![
TypeDef {
name: "ProcessResult".to_string(),
fields: vec![kind_field(TypeRef::Named("DataNodeKind".to_string()), false)],
..TypeDef::default()
},
TypeDef {
name: "OtherResult".to_string(),
fields: vec![kind_field(TypeRef::String, false)],
..TypeDef::default()
},
TypeDef {
name: "OptionalResult".to_string(),
fields: vec![kind_field(
TypeRef::Optional(Box::new(TypeRef::Named("DataNodeKind".to_string()))),
true,
)],
..TypeDef::default()
},
];
let enums = vec![data_node_kind_enum()];
let functions = vec![
FunctionDef {
name: "process".to_string(),
return_type: TypeRef::Named("ProcessResult".to_string()),
..FunctionDef::default()
},
FunctionDef {
name: "other".to_string(),
return_type: TypeRef::Named("OtherResult".to_string()),
..FunctionDef::default()
},
FunctionDef {
name: "process_optional".to_string(),
return_type: TypeRef::Named("OptionalResult".to_string()),
..FunctionDef::default()
},
];
(type_defs, enums, functions)
}
fn fixture_calling(call: &str) -> Fixture {
Fixture {
id: "kind_smoke".to_string(),
description: "Kind field smoke".to_string(),
call: Some(call.to_string()),
assertions: vec![Assertion {
assertion_type: "equals".to_string(),
field: Some("kind".to_string()),
value: Some(serde_json::Value::String("KeyValue".to_string())),
..Assertion::default()
}],
..Fixture::default()
}
}
fn e2e_config_for(call: &str) -> E2eConfig {
let call_config = CallConfig {
function: call.to_string(),
module: "MyLib".to_string(),
result_var: "result".to_string(),
returns_result: true,
..CallConfig::default()
};
let mut e2e_config = E2eConfig::default();
e2e_config.calls.insert(call.to_string(), call_config);
e2e_config
}
fn render(
fixture: &Fixture,
e2e_config: &E2eConfig,
type_defs: &[TypeDef],
enums: &[EnumDef],
functions: &[FunctionDef],
) -> String {
let mut out = String::new();
render_test_case(
&mut out,
fixture,
e2e_config,
"",
"",
"",
&[],
None,
None,
&std::collections::HashMap::new(),
None,
&std::collections::HashSet::new(),
&[],
enums,
&ResolvedCrateConfig::default(),
type_defs,
&[],
functions,
);
out
}
struct Case {
name: &'static str,
call: &'static str,
expect_enum_branch: bool,
}
const CASES: &[Case] = &[
Case {
name: "an enum-typed field with no fields_enum config is classified as enum via the IR",
call: "process",
expect_enum_branch: true,
},
Case {
name: "a same-named non-enum field on an unrelated type is not misclassified as enum",
call: "other",
expect_enum_branch: false,
},
Case {
name: "an Option<Enum> field is classified as enum via the IR",
call: "process_optional",
expect_enum_branch: true,
},
];
#[test]
fn enum_field_classification_table() {
let (type_defs, enums, functions) = table_ir();
for case in CASES {
let e2e_config = e2e_config_for(case.call);
let fixture = fixture_calling(case.call);
let out = render(&fixture, &e2e_config, &type_defs, &enums, &functions);
let took_enum_branch = out.contains(ENUM_MARKER);
assert_eq!(
took_enum_branch, case.expect_enum_branch,
"{}: expected enum branch = {}, got:\n{out}",
case.name, case.expect_enum_branch
);
assert_eq!(
out.contains(RAW_MARKER),
!case.expect_enum_branch,
"{}: a raw comparison must be emitted only for the non-enum field, got:\n{out}",
case.name
);
}
}
#[test]
fn an_explicit_fields_enum_config_entry_still_classifies_as_enum() {
let (type_defs, enums, functions) = table_ir();
let mut e2e_config = e2e_config_for("other");
e2e_config.fields_enum.insert("kind".to_string());
let fixture = fixture_calling("other");
let out = render(&fixture, &e2e_config, &type_defs, &enums, &functions);
assert!(
out.contains("to_string(result.kind)"),
"explicit fields_enum config must still classify the field as enum (via the legacy \
to_string/1 fallback, since the concrete enum module is unknown here), got:\n{out}"
);
assert!(
!out.contains(ENUM_MARKER),
"the config-only path has no resolved enum type name to qualify wire_value/1 with, so \
it must not emit the IR-derived call, got:\n{out}"
);
}
#[test]
fn enum_equals_assertion_calls_wire_value_and_compares_the_literal_verbatim() {
let (type_defs, enums, functions) = table_ir();
let e2e_config = e2e_config_for("process");
let fixture = fixture_calling("process");
let out = render(&fixture, &e2e_config, &type_defs, &enums, &functions);
assert!(
out.contains("assert MyLib.DataNodeKind.wire_value(result.kind) == \"KeyValue\""),
"expected the fixture's wire literal verbatim, compared through wire_value/1, got:\n{out}"
);
assert!(
!out.contains("to_string(result.kind)"),
"an IR-resolved enum field must not fall back to to_string/1, got:\n{out}"
);
assert!(
!out.contains("\"key_value\""),
"must not lower/re-case the expected literal to match the atom's own spelling, got:\n{out}"
);
}
#[test]
fn a_data_carrying_enum_field_resolves_through_wire_value_not_to_string() {
let format_metadata_enum = EnumDef {
name: "FormatMetadata".to_string(),
variants: vec![
EnumVariant {
name: "Pdf".to_string(),
fields: vec![FieldDef {
name: "_0".to_string(),
ty: TypeRef::Named("PdfMetadata".to_string()),
..FieldDef::default()
}],
is_tuple: true,
..EnumVariant::default()
},
EnumVariant {
name: "Docx".to_string(),
fields: vec![FieldDef {
name: "_0".to_string(),
ty: TypeRef::Named("DocxMetadata".to_string()),
..FieldDef::default()
}],
is_tuple: true,
..EnumVariant::default()
},
],
..EnumDef::default()
};
let type_defs = vec![TypeDef {
name: "MetadataResult".to_string(),
fields: vec![FieldDef {
name: "metadata".to_string(),
ty: TypeRef::Named("FormatMetadata".to_string()),
..FieldDef::default()
}],
..TypeDef::default()
}];
let functions = vec![FunctionDef {
name: "process_metadata".to_string(),
return_type: TypeRef::Named("MetadataResult".to_string()),
..FunctionDef::default()
}];
let enums = vec![format_metadata_enum];
let e2e_config = e2e_config_for("process_metadata");
let fixture = Fixture {
id: "metadata_smoke".to_string(),
description: "Metadata field smoke".to_string(),
call: Some("process_metadata".to_string()),
assertions: vec![Assertion {
assertion_type: "equals".to_string(),
field: Some("metadata".to_string()),
value: Some(serde_json::Value::String("Pdf".to_string())),
..Assertion::default()
}],
..Fixture::default()
};
let out = render(&fixture, &e2e_config, &type_defs, &enums, &functions);
assert!(
out.contains("assert MyLib.FormatMetadata.wire_value(result.metadata) == \"Pdf\""),
"a data-carrying enum field must resolve through wire_value/1 with the fixture's wire \
literal verbatim, got:\n{out}"
);
assert!(
!out.contains("to_string(result.metadata)"),
"to_string/1 has no String.Chars impl for this enum's flat-struct/tuple runtime shape \
and raises Protocol.UndefinedError, got:\n{out}"
);
}