use crate::core::config::ResolvedCrateConfig;
use crate::e2e::config::E2eConfig;
use crate::e2e::fixture::{Assertion, Fixture};
use std::collections::HashMap;
fn mock_assertion(field: &str, assertion_type: &str, value: serde_json::Value) -> Assertion {
Assertion {
assertion_type: assertion_type.to_string(),
field: Some(field.to_string()),
value: Some(value),
..Default::default()
}
}
fn render(assertions: Vec<Assertion>) -> String {
let fixture = Fixture {
id: "mock_capture_fixture".to_string(),
description: "asserts against the mock server request log".to_string(),
assertions,
..Fixture::default()
};
let mut e2e_config = E2eConfig::default();
e2e_config.call.function = "chat".to_string();
e2e_config.call.result_var = "result".to_string();
let enum_fields: HashMap<String, String> = HashMap::new();
super::spec_file::render_spec_file(
"mock_capture",
&[&fixture],
"Sample",
None,
"sample",
None,
&enum_fields,
false,
&e2e_config,
false,
false,
&[],
&ResolvedCrateConfig::default(),
&[],
&[],
&[],
&[],
)
}
#[test]
fn a_mock_assertion_gets_both_its_call_and_the_helper_definition_in_one_file() {
let out = render(vec![
mock_assertion("mock.requests.total", "equals", serde_json::json!(1)),
mock_assertion(
r#"mock.requests["POST /v1/chat"]"#,
"greater_than_or_equal",
serde_json::json!(1),
),
]);
assert!(
out.contains(
" expect(alef_mock_request_count(ENV.fetch('MOCK_SERVER_URL'), \"/__alef/requests/total?prefix=\")).to eq(1), 'expected mock.requests.total == 1'"
),
"expected the total-count call, got:\n{out}"
);
assert!(
out.contains(
" expect(alef_mock_request_count(ENV.fetch('MOCK_SERVER_URL'), \"/__alef/requests/one?key=POST%20%2Fv1%2Fchat\")).to be >= 1, 'expected mock.requests[\"POST /v1/chat\"] >= 1'"
),
"expected the bracketed-key call, got:\n{out}"
);
assert!(
out.contains("def alef_mock_request_count(base_url, path_and_query)"),
"expected the helper definition in the same file as the call, got:\n{out}"
);
}
#[test]
fn a_file_with_no_mock_assertion_defines_no_helper() {
let out = render(vec![Assertion {
assertion_type: "not_error".to_string(),
..Default::default()
}]);
assert!(
!out.contains("alef_mock_request_count"),
"a file with no mock.* assertion must not reference or define the helper, got:\n{out}"
);
}