alef 0.79.4

//! Python accessor rendering, narrowing an `Optional` member before anything further reads it,
//! and switching between attribute access (`result.field`) and subscript access
//! (`result["field"]`) at exactly the links the pyo3 backend emits as a `TypedDict`.
//!
//! Python has no null-safe navigation operator (`?.`), so unlike the TypeScript/Kotlin/C#
//! `_with_optionals` renderers in `optional_renderers.rs`, a crossing cannot be spelled as a
//! single extra token on the link itself. Instead each crossing wraps the expression built so
//! far in a conditional expression (`tail if prefix else None`) that repeats the same raw prefix
//! text as both the condition and (as part of `tail`) the consequent — type checkers narrow a
//! member-access expression when it recurs unchanged across a ternary's condition and its
//! consequent, so this deliberately re-renders the prefix rather than binding it to a local name.
//!
//! `render_accessor`'s plain `render_dot_access` fallback answered nothing about optionality, so
//! a `list[T] | None` field reached a subscript with no guard at all
//! (`reportOptionalSubscript`/`reportOptionalMemberAccess`) — this is the Python counterpart to
//! the fix `csharp_optional_index_tests.rs` already covers for C#'s null-forgiving operator.
//!
//! Separately, `render_dot_access` answered nothing about `TypedDict` return types either: a
//! `[workspace.dto] python_output = "typed-dict"` crate's return type is a plain `dict` at
//! runtime (`TypedDict` is a type-checking fiction only), so `.field` on it is
//! `AttributeError: 'dict' object has no attribute 'field'`. `render_python_accessor` walks
//! `segments` with a "current owner type" cursor (mirroring `render_swift_with_first_class_map`)
//! and consults [`PythonTypedDictMap::is_typeddict`] at each link to pick `["field"]` vs.
//! `.field` — asking the pyo3 backend's own predicate for the answer rather than re-deriving it.
//!
//! A THIRD, later defect: that owner cursor always started at `typeddict_map.root_type` — the
//! call's declared RESULT type — even when rendering an element-anchored path (the closure body
//! a wildcard `container[].field` fixture path expands to). A result envelope can be a
//! `TypedDict` while its collection ELEMENTS stay a native `#[pyclass]` (attribute access), or
//! vice versa — the two are independent per-type classifications, not one style inherited down
//! the whole path. Starting the element cursor at the result root produced
//! `any("Function" in str(_e["kind"]) for _e in (result["structure"] or []))` against a consumer
//! whose `SampleItem` elements are plain attribute-access `#[pyclass]` instances:
//! `TypeError: 'SampleItem' object is not subscriptable`. [`python_element_owner_type`] resolves
//! the actual element owner type by walking the array field's own path through
//! `typeddict_map.field_types`, so [`FieldResolver::python_element_accessor`] can start the
//! cursor there instead.
//!
//! A FOURTH instance of that same "the wrong type answered the question" shape: a `map[key]` hop
//! left the cursor on the map's OWNER, so `extras[key].title` classified `title` against the
//! struct that declares `extras` rather than against the map's VALUE type. Retaining the owner
//! and defaulting to `None` are both guesses — each happens to be right for a different config —
//! so the fix is a DERIVED edge (recorded by `python_typeddict` from
//! `call_ir::map_value_named_type`) rather than a different default:
//! `named_type` names nothing for a map, so before that edge existed the IR had no answer to
//! give at all. ~keep

use super::optional_renderers::{push_key_field_name, push_key_index_suffix};
use super::renderers::quoted_key_literal;
use super::types::{PathSegment, PythonMapValueEdges, PythonTypedDictMap};
use std::collections::HashSet;

/// Render a Python accessor expression, wrapping every point where the chain crosses an
/// `Optional` field (per `optional_fields`) in a narrowing conditional expression.
///
/// A crossing at the last segment needs no guard: nothing further reads the value, so leaving it
/// `Optional` in the rendered expression is correct as-is. A crossing anywhere earlier gets its
/// own nested ternary, innermost (deepest into the path) first, so each guard's own condition is
/// always evaluated either at the very start of the path (safe by construction) or already inside
/// an enclosing guard's narrowed branch (safe because that guard ran first).
pub(super) fn render_python_with_optionals(
    segments: &[PathSegment],
    result_var: &str,
    optional_fields: &HashSet<String>,
    typeddict_map: &PythonTypedDictMap,
    map_value_edges: &PythonMapValueEdges,
) -> String {
    render_python_with_optionals_from_owner(
        segments,
        result_var,
        optional_fields,
        typeddict_map,
        map_value_edges,
        typeddict_map.root_type.clone(),
    )
}

/// [`render_python_with_optionals`], but for a path that is already relative to a bound
/// collection element (the closure/loop variable a wildcard fixture path expands to) rather than
/// to the call's result variable — `owner_type` is the IR type of THAT element, resolved by
/// [`python_element_owner_type`], not `typeddict_map.root_type`. See the module doc for the
/// runtime failure this fixes.
pub(super) fn render_python_element_with_optionals(
    segments: &[PathSegment],
    element_var: &str,
    optional_fields: &HashSet<String>,
    typeddict_map: &PythonTypedDictMap,
    map_value_edges: &PythonMapValueEdges,
    owner_type: Option<String>,
) -> String {
    render_python_with_optionals_from_owner(
        segments,
        element_var,
        optional_fields,
        typeddict_map,
        map_value_edges,
        owner_type,
    )
}

fn render_python_with_optionals_from_owner(
    segments: &[PathSegment],
    result_var: &str,
    optional_fields: &HashSet<String>,
    typeddict_map: &PythonTypedDictMap,
    map_value_edges: &PythonMapValueEdges,
    owner_type: Option<String>,
) -> String {
    let last_index = segments.len().saturating_sub(1);
    let mut crossings: Vec<usize> = Vec::new();
    let mut path_so_far = String::new();
    for (index, segment) in segments.iter().enumerate() {
        push_key_field_name(&mut path_so_far, segment);
        if index != last_index && optional_fields.contains(&path_so_far) {
            crossings.push(index);
        }
        push_key_index_suffix(&mut path_so_far, segment);
    }

    let mut expression =
        render_python_accessor_from_owner(segments, result_var, typeddict_map, map_value_edges, owner_type.clone());
    for &index in crossings.iter().rev() {
        let condition = render_python_accessor_from_owner(
            &field_only_prefix(segments, index),
            result_var,
            typeddict_map,
            map_value_edges,
            owner_type.clone(),
        );
        expression = format!("({expression} if {condition} else None)");
    }
    expression
}

/// The IR type that owns the ELEMENTS of `array_segments` — e.g. `"SampleItem"` for a
/// `structure: Vec<SampleItem>` field — walking `typeddict_map.field_types` from
/// `typeddict_map.root_type` through every segment of the array field's own path, exactly the
/// way `render_python_accessor`'s cursor advances. `None` under the same "IR cannot judge"
/// conditions [`PythonTypedDictMap::advance`] answers `None` for: an unresolved root, or a
/// segment the map never recorded a traversal edge for. [`PythonTypedDictMap::is_typeddict`]
/// treats `None` as "attribute access", the correct default for an opaque/native `#[pyclass]`
/// element type. A `map[key]` hop advances through the resolver's private map-value edges instead,
/// per [`advance_through_map_access`].
///
/// ~keep `array_segments` must be parsed from the path the CONTAINER was actually rendered from
/// (`FieldResolver::result_relative_path`, envelope projection applied), not from the raw fixture
/// spelling: this walk starts at `root_type`, so a path that skips the projection's hops finds no
/// edge, returns `None`, and silently answers "attribute access" for every projected container.
pub(super) fn python_element_owner_type(
    array_segments: &[PathSegment],
    typeddict_map: &PythonTypedDictMap,
    map_value_edges: &PythonMapValueEdges,
) -> Option<String> {
    let mut current_type = typeddict_map.root_type.clone();
    for segment in array_segments {
        match segment {
            PathSegment::Field(name) | PathSegment::ArrayField { name, .. } => {
                current_type = typeddict_map.advance(current_type.as_deref(), name);
            }
            PathSegment::MapAccess { field, .. } => {
                current_type = advance_through_map_access(current_type, field, map_value_edges);
            }
            PathSegment::Length => {}
        }
    }
    current_type
}

/// Advance the owner cursor across one `map[key]` access: to the map's VALUE type when the IR
/// recorded one, otherwise unchanged.
///
/// ~keep Retaining the previous owner when there is no map-value edge is not a second guess at the
/// value's shape — it is declining to make one. `is_typeddict(None)` means "attribute access", a
/// positive claim. When the value shape is unnamed (a scalar, JSON string carrier, nested map,
/// or foreign type), retaining the existing owner classification preserves the renderer's prior
/// behaviour without pretending the IR derived a named target. The only owner this changes is
/// the one the IR can actually derive.
fn advance_through_map_access(
    current_type: Option<String>,
    field: &str,
    map_value_edges: &PythonMapValueEdges,
) -> Option<String> {
    let advanced = current_type
        .as_deref()
        .and_then(|owner| map_value_edges.get(owner))
        .and_then(|fields| fields.get(field))
        .cloned();
    advanced.or(current_type)
}

/// Render a Python accessor expression for `segments`, tracking the IR type that "owns" each
/// segment so a field whose owner the pyo3 backend emits as a `TypedDict` gets subscript access
/// (`result["field"]`) while every other owner (dataclass / pydantic / msgspec / native
/// `#[pyclass]` — all attribute-access shapes in Python) keeps `.field`.
///
/// `TypedDict`-ness is checked at the OWNER of each segment, not at the segment's target type —
/// mirroring `render_swift_with_first_class_map`'s per-segment dispatch — so a path that starts
/// on a `TypedDict` result and descends into a field whose own type is not itself emitted as a
/// `TypedDict` correctly switches back to attribute access at that link, and does not need a
/// special case: the cursor just stops finding `is_typeddict(current_type) == true` for it.
#[cfg(test)]
pub(super) fn render_python_accessor(
    segments: &[PathSegment],
    result_var: &str,
    map: &PythonTypedDictMap,
    map_value_edges: &PythonMapValueEdges,
) -> String {
    render_python_accessor_from_owner(segments, result_var, map, map_value_edges, map.root_type.clone())
}

/// `render_python_accessor`, but starting the owner-type cursor at `owner_type` instead of
/// `map.root_type` — see the module doc and [`python_element_owner_type`] for why an
/// element-anchored path needs a different starting owner.
fn render_python_accessor_from_owner(
    segments: &[PathSegment],
    result_var: &str,
    map: &PythonTypedDictMap,
    map_value_edges: &PythonMapValueEdges,
    owner_type: Option<String>,
) -> String {
    let mut out = result_var.to_string();
    let mut current_type = owner_type;
    for seg in segments {
        match seg {
            PathSegment::Field(f) => {
                push_field_access(&mut out, f, current_type.as_deref(), map);
                current_type = map.advance(current_type.as_deref(), f);
            }
            PathSegment::ArrayField { name, index } => {
                push_field_access(&mut out, name, current_type.as_deref(), map);
                out.push_str(&format!("[{index}]"));
                current_type = map.advance(current_type.as_deref(), name);
            }
            PathSegment::MapAccess { field, key } => {
                push_field_access(&mut out, field, current_type.as_deref(), map);
                // ~keep The key/index suffix itself is unaffected by `TypedDict`-ness (the map is
                // a plain `dict` whatever the owning struct's DTO style), mirroring
                // `render_dot_access`. The owner cursor, however, MUST advance to the map's VALUE
                // type here: everything after `extras[key]` belongs to the value, and classifying
                // it against the type that merely OWNS `extras` answers a question about the
                // wrong type. See [`advance_through_map_access`] for the no-edge case.
                current_type = advance_through_map_access(current_type, field, map_value_edges);
                if key.chars().all(|c| c.is_ascii_digit()) {
                    let idx: usize = key.parse().unwrap_or(0);
                    out.push_str(&format!("[{idx}]"));
                } else {
                    out.push_str(&format!(".get({})", quoted_key_literal(key)));
                }
            }
            PathSegment::Length => {
                let current = std::mem::take(&mut out);
                out = format!("len({current})");
            }
        }
    }
    out
}

/// Append one field access (`.field` or `["field"]`) to `out`, per whether `owner_type` is
/// classified as a `TypedDict` in `map`.
fn push_field_access(out: &mut String, field: &str, owner_type: Option<&str>, map: &PythonTypedDictMap) {
    if map.is_typeddict(owner_type) {
        out.push_str(&format!("[{}]", quoted_key_literal(field)));
    } else {
        out.push('.');
        out.push_str(field);
    }
}

/// `segments[..=index]` with segment `index`'s own `[..]`/key suffix stripped — the "is this
/// collection present at all" question, asked before the index that would subscript a `None`.
fn field_only_prefix(segments: &[PathSegment], index: usize) -> Vec<PathSegment> {
    let mut prefix: Vec<PathSegment> = segments[..index].to_vec();
    prefix.push(match &segments[index] {
        PathSegment::ArrayField { name, .. } => PathSegment::Field(name.clone()),
        PathSegment::MapAccess { field, .. } => PathSegment::Field(field.clone()),
        other => other.clone(),
    });
    prefix
}

#[cfg(test)]
mod tests {
    use super::super::parse::parse_path;
    use super::*;

    fn typeddict_map(
        typeddict_types: &[&str],
        field_types: &[(&str, &str, &str)],
        root_type: &str,
    ) -> PythonTypedDictMap {
        let mut map = PythonTypedDictMap {
            typeddict_types: typeddict_types.iter().map(|s| s.to_string()).collect(),
            root_type: Some(root_type.to_string()),
            ..Default::default()
        };
        for (owner, field, target) in field_types {
            map.field_types
                .entry(owner.to_string())
                .or_default()
                .insert(field.to_string(), target.to_string());
        }
        map
    }

    /// A scalar field on a `TypedDict` result renders as a subscript, not an attribute — the
    /// exact defect reported against a consumer: `result.status_code` raised `AttributeError:
    /// 'dict' object has no attribute 'status_code'` because the result is a plain `dict` at
    /// runtime, not the (type-checking-only) `TypedDict` its annotation claims.
    #[test]
    fn a_scalar_field_on_a_typeddict_result_is_subscripted() {
        let map = typeddict_map(&["ApiResult"], &[], "ApiResult");
        let segments = parse_path("status_code");
        assert_eq!(
            render_python_accessor(&segments, "result", &map, &PythonMapValueEdges::new()),
            r#"result["status_code"]"#
        );
    }

    /// CONTROL: the identical field on a result type NOT classified as `TypedDict` keeps plain
    /// attribute access — proving the new behaviour is conditional on the map, not blanket.
    #[test]
    fn a_scalar_field_on_a_non_typeddict_result_stays_attribute_access() {
        let map = PythonTypedDictMap::default();
        let segments = parse_path("status_code");
        assert_eq!(
            render_python_accessor(&segments, "result", &map, &PythonMapValueEdges::new()),
            "result.status_code"
        );
    }

    /// A `TypedDict` result with an `Optional` field: the narrowing ternary's condition AND
    /// consequent both use subscript access on the `TypedDict` owner, matching the shape
    /// `render_python_with_optionals` already produces for attribute access.
    #[test]
    fn a_typeddict_result_with_an_optional_field_narrows_via_subscript() {
        let map = typeddict_map(&["ApiResult"], &[], "ApiResult");
        let optional: HashSet<String> = ["markdown".to_string()].into_iter().collect();
        let segments = parse_path("markdown");
        assert_eq!(
            render_python_with_optionals(&segments, "result", &optional, &map, &PythonMapValueEdges::new()),
            r#"result["markdown"]"#,
            "a crossing at the LAST segment needs no ternary guard"
        );
    }

    /// A `TypedDict` result descending through an `Optional` `TypedDict` field into a further
    /// scalar: the crossing's ternary condition subscripts the intermediate field, and the full
    /// expression continues subscripting past it.
    #[test]
    fn a_typeddict_result_with_an_optional_nested_typeddict_field_narrows_before_descending() {
        let map = typeddict_map(
            &["ApiResult", "Markdown"],
            &[("ApiResult", "markdown", "Markdown")],
            "ApiResult",
        );
        let optional: HashSet<String> = ["markdown".to_string()].into_iter().collect();
        let segments = parse_path("markdown.content");
        assert_eq!(
            render_python_with_optionals(&segments, "result", &optional, &map, &PythonMapValueEdges::new()),
            r#"(result["markdown"]["content"] if result["markdown"] else None)"#
        );
    }

    /// A `TypedDict` result descending into a field whose OWN type is not itself classified as
    /// `TypedDict` (e.g. it stays a native `#[pyclass]`) must switch back to attribute access at
    /// that link — the classification is checked per-segment-owner, not inherited from the root.
    #[test]
    fn descending_from_a_typeddict_into_a_non_typeddict_nested_type_switches_to_attribute_access() {
        let map = typeddict_map(&["ApiResult"], &[("ApiResult", "metadata", "Metadata")], "ApiResult");
        let segments = parse_path("metadata.title");
        assert_eq!(
            render_python_accessor(&segments, "result", &map, &PythonMapValueEdges::new()),
            r#"result["metadata"].title"#
        );
    }

    /// An indexed array field on a `TypedDict` owner subscripts the field name, then indexes the
    /// resulting list with plain `[N]` — list indexing is unaffected by the owning struct's DTO
    /// style.
    #[test]
    fn an_array_field_on_a_typeddict_result_subscripts_the_field_then_indexes_the_list() {
        let map = typeddict_map(&["ApiResult"], &[], "ApiResult");
        let segments = parse_path("pages[0]");
        assert_eq!(
            render_python_accessor(&segments, "result", &map, &PythonMapValueEdges::new()),
            r#"result["pages"][0]"#
        );
    }

    /// CONTROL (pre-existing behaviour, unaffected by the empty default map): indexing past an
    /// `Optional` collection field still narrows before subscripting.
    #[test]
    fn indexing_past_an_optional_field_narrows_it_first() {
        let map = PythonTypedDictMap::default();
        let optional: HashSet<String> = ["choices[0].message.tool_calls".to_string()].into_iter().collect();
        let segments = parse_path("choices[0].message.tool_calls[0].function.name");
        assert_eq!(
            render_python_with_optionals(&segments, "result", &optional, &map, &PythonMapValueEdges::new()),
            "(result.choices[0].message.tool_calls[0].function.name if result.choices[0].message.tool_calls else None)"
        );
    }

    /// `python_element_owner_type` must advance the cursor once per container segment, all the way
    /// to the collection's element type.
    ///
    /// ~keep The multi-segment case is the one that can silently half-work: consulting only the
    /// LAST segment answers `advance("Envelope", "records")` — no such edge, so `None`, which
    /// `is_typeddict` reads as attribute access — and consulting only the FIRST answers `Report`,
    /// the container's owner rather than the element's. Only a full walk reaches `Entry`, so this
    /// expectation separates all three implementations.
    #[test]
    fn element_owner_type_advances_through_every_container_segment() {
        let map = typeddict_map(
            &["Envelope", "Report", "Entry"],
            &[("Envelope", "results", "Report"), ("Report", "records", "Entry")],
            "Envelope",
        );
        let segments = parse_path("results[0].records");
        assert_eq!(
            python_element_owner_type(&segments, &map, &PythonMapValueEdges::new()),
            Some("Entry".to_string())
        );
    }

    /// An `ArrayField` segment advances by its field NAME; the index is list subscripting and
    /// carries no type information. Pinned separately because the envelope projection
    /// `FieldResolver::python_element_accessor` walks always produces an indexed first hop, so a
    /// walk that skipped `ArrayField` would answer `None` for every projected container.
    #[test]
    fn element_owner_type_advances_through_an_indexed_segment_by_field_name() {
        let map = typeddict_map(&["Report"], &[("Report", "records", "Entry")], "Report");
        assert_eq!(
            python_element_owner_type(&parse_path("records[0]"), &map, &PythonMapValueEdges::new()),
            Some("Entry".to_string())
        );
    }

    /// A segment the map recorded no traversal edge for yields `None` — the documented "the IR
    /// cannot judge this owner" answer, which `is_typeddict` resolves to attribute access, the
    /// correct default for an opaque/native `#[pyclass]` element.
    #[test]
    fn element_owner_type_is_none_for_a_segment_with_no_recorded_edge() {
        let map = typeddict_map(&["Envelope"], &[("Envelope", "results", "Report")], "Envelope");
        assert_eq!(
            python_element_owner_type(&parse_path("records"), &map, &PythonMapValueEdges::new()),
            None
        );
    }

    /// An unresolved root type yields `None` rather than guessing: `advance` short-circuits on a
    /// `None` owner, so every later segment stays `None` too.
    #[test]
    fn element_owner_type_is_none_when_the_root_type_is_unresolved() {
        let map = PythonTypedDictMap::default();
        assert_eq!(
            python_element_owner_type(&parse_path("results.records"), &map, &PythonMapValueEdges::new()),
            None
        );
    }

    fn with_map_values(
        map: PythonTypedDictMap,
        values: &[(&str, &str, &str)],
    ) -> (PythonTypedDictMap, PythonMapValueEdges) {
        let mut edges = PythonMapValueEdges::new();
        for (owner, field, target) in values {
            edges
                .entry(owner.to_string())
                .or_default()
                .insert(field.to_string(), target.to_string());
        }
        (map, edges)
    }

    /// DIRECTION ONE — the map's VALUE is a `TypedDict` while the struct that OWNS the map is not.
    /// Everything after the key access belongs to the value, so `title` must be SUBSCRIPTED even
    /// though `entries` itself was reached by attribute access.
    ///
    /// ~keep Revert the fix and the cursor stays on `Report` (not a `TypedDict`), rendering
    /// `.title` on what is a plain `dict` at runtime: `AttributeError: 'dict' object has no
    /// attribute 'title'`. The owner and the value are deliberately classified OPPOSITELY — a
    /// fixture where both agree cannot tell the fixed renderer from the broken one.
    #[test]
    fn a_typeddict_map_value_under_a_non_typeddict_owner_is_subscripted() {
        let (map, edges) = with_map_values(
            typeddict_map(&["Meta"], &[], "Report"),
            &[("Report", "entries", "Meta")],
        );
        assert_eq!(
            render_python_accessor(&parse_path("entries[alpha].title"), "result", &map, &edges),
            r#"result.entries.get("alpha")["title"]"#
        );
    }

    /// DIRECTION TWO — the mirror image: the map's OWNER is a `TypedDict` while its VALUE stays a
    /// reexported/native `#[pyclass]`. `entries` is subscripted, and `title` on the value must
    /// switch back to ATTRIBUTE access.
    ///
    /// ~keep Revert the fix and the cursor stays on `ApiResult` (a `TypedDict`), rendering
    /// `["title"]` against a compiled pyclass instance: `TypeError: 'Meta' object is not
    /// subscriptable`. This is the direction the previously-proposed `current_type = None` fix
    /// also gets right — and direction one is the one it gets wrong, which is why both tests
    /// exist.
    #[test]
    fn a_native_map_value_under_a_typeddict_owner_uses_attribute_access() {
        let (map, edges) = with_map_values(
            typeddict_map(&["ApiResult"], &[], "ApiResult"),
            &[("ApiResult", "entries", "Meta")],
        );
        assert_eq!(
            render_python_accessor(&parse_path("entries[alpha].title"), "result", &map, &edges),
            r#"result["entries"].get("alpha").title"#
        );
    }

    /// PRESERVATION PIN (passes before and after the fix, by design): under the dominant
    /// `python_output = "typed-dict"` config the owner AND the value are both `TypedDict`s, and
    /// subscript access all the way through stays correct. Recorded so a future change that
    /// starts answering `None` for a derived map value is caught here rather than in a consumer.
    #[test]
    fn a_typeddict_map_value_under_a_typeddict_owner_stays_subscripted() {
        let (map, edges) = with_map_values(
            typeddict_map(&["ApiResult", "Meta"], &[], "ApiResult"),
            &[("ApiResult", "entries", "Meta")],
        );
        assert_eq!(
            render_python_accessor(&parse_path("entries[alpha].title"), "result", &map, &edges),
            r#"result["entries"].get("alpha")["title"]"#
        );
    }

    /// A map the IR recorded NO value edge for (values are scalars, JSON string carriers, nested
    /// maps, or foreign types) keeps the previous owner rather than asserting a named target.
    ///
    /// ~keep No edge means no derived named owner, so the renderer declines to change its answer.
    /// This is a compatibility control, not a claim that every unnamed value supports the
    /// trailing field path; PyO3 exposes `TypeRef::Json` as a JSON string carrier, for example.
    #[test]
    fn a_map_with_no_recorded_value_edge_retains_the_owner_classification() {
        let map = typeddict_map(&["ApiResult"], &[], "ApiResult");
        assert_eq!(
            render_python_accessor(
                &parse_path("extras[alpha].title"),
                "result",
                &map,
                &PythonMapValueEdges::new(),
            ),
            r#"result["extras"].get("alpha")["title"]"#
        );
    }

    /// The owner walk advances through a `map[key]` hop too, so an element-anchored path whose
    /// container sits inside a map resolves the element type instead of stalling.
    ///
    /// ~keep Revert the fix and the `MapAccess` arm `continue`s: the cursor stays on `Envelope`,
    /// `advance("Envelope", "records")` finds no edge, and the walk answers `None` — silently
    /// "attribute access" for every element of every map-nested container.
    #[test]
    fn element_owner_type_advances_through_a_map_access_segment() {
        let (map, edges) = with_map_values(
            typeddict_map(
                &["Envelope", "Report", "Entry"],
                &[("Report", "records", "Entry")],
                "Envelope",
            ),
            &[("Envelope", "reports", "Report")],
        );
        assert_eq!(
            python_element_owner_type(&parse_path("reports[alpha].records"), &map, &edges),
            Some("Entry".to_string())
        );
    }

    /// CONTROL for the walk above: `Field` and `ArrayField` hops still consult `field_types` and
    /// are untouched by the map-value edge — a `Vec`/`Option` traversal resolves exactly as it did
    /// before, and a map-value edge on the same owner name does not answer for them.
    #[test]
    fn element_owner_type_ignores_map_value_edges_for_plain_field_hops() {
        let (map, edges) = with_map_values(
            typeddict_map(&["Envelope"], &[("Envelope", "results", "Report")], "Envelope"),
            &[("Envelope", "results", "Decoy")],
        );
        assert_eq!(
            python_element_owner_type(&parse_path("results[0]"), &map, &edges),
            Some("Report".to_string()),
            "a field hop never reads the internal map-value edge namespace"
        );
    }
}