alef 0.82.2

Opinionated polyglot binding generator for Rust libraries
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159

//! Streaming virtual-field assertion rendering for the Go e2e generator.
//!
//! A streaming fixture's assertions can target virtual fields (`finish_reason`,
//! `stream.has_page_event`, ...) that only resolve through the streaming accessor, never
//! through `FieldResolver`. This must be checked -- and rendered, or skip-marked -- before
//! any ordinary field resolution runs, so it must run ahead of the wildcard/availability
//! router and the `ResolvedAssertionTarget` derivation.
//!
//! Split out of `assertions.rs`, which is over the 1000-line cap and may not grow.

use crate::e2e::codegen::assertion_type_skip::{
    streaming_assertion_type_skip_line, streaming_assertion_value_skip_line,
};
use crate::e2e::codegen::field_skip::FieldSkip;
use crate::e2e::escape::go_string_literal;
use crate::e2e::fixture::Assertion;
use std::fmt::Write as FmtWrite;

use super::AssertionRenderContext;

/// Render an assertion whose field names a streaming virtual field.
///
/// Returns `true` when the assertion targets a streaming virtual field (and the
/// assertion, or its skip-marker fallback, has been written to `out`); `false` when the
/// caller must fall through to ordinary field resolution.
pub(super) fn render_streaming_field_assertion(
    out: &mut String,
    assertion: &Assertion,
    context: &AssertionRenderContext<'_>,
) -> bool {
    let Some(f) = assertion.field.as_deref() else {
        return false;
    };
    if context.result_is_simple
        || !context.is_streaming
        || f.is_empty()
        || !crate::e2e::codegen::streaming_assertions::is_streaming_virtual_field(f)
    {
        return false;
    }
    let streaming_item_type = context.streaming_item_type;

    if let Some(expr) =
        crate::e2e::codegen::streaming_assertions::StreamingFieldResolver::accessor_with_streaming_context(
            f,
            "go",
            "chunks",
            None,
            streaming_item_type,
        )
    {
        // ~keep The value-narrowing arms below used to fall through to nothing when the
        // fixture's value did not survive `as_u64()` / the string pattern, so the assertion
        // disappeared with no line for any funnel to count.
        let value_skip = || streaming_assertion_value_skip_line("\t", "//", f, &assertion.assertion_type);
        match assertion.assertion_type.as_str() {
            "count_min" => {
                if let Some(val) = &assertion.value
                    && let Some(n) = val.as_u64()
                {
                    let _ = writeln!(
                        out,
                        "\tassert.GreaterOrEqual(t, len({expr}), {n}, \"expected >= {n} chunks\")"
                    );
                } else {
                    let _ = writeln!(out, "{}", value_skip());
                }
            }
            "count_equals" => {
                if let Some(val) = &assertion.value
                    && let Some(n) = val.as_u64()
                {
                    let _ = writeln!(
                        out,
                        "\tassert.Equal(t, {n}, len({expr}), \"expected exactly {n} chunks\")"
                    );
                } else {
                    let _ = writeln!(out, "{}", value_skip());
                }
            }
            "equals" => {
                if let Some(serde_json::Value::String(s)) = &assertion.value {
                    let escaped = go_string_literal(s);
                    let is_deep_path = f.contains('.') || f.contains('[');
                    let safe_expr = if is_deep_path {
                        format!("func() string {{ v := {expr}; if v == nil {{ return \"\" }}; return *v }}()")
                    } else {
                        expr.clone()
                    };
                    let _ = writeln!(out, "\tassert.Equal(t, {escaped}, {safe_expr})");
                } else if let Some(val) = &assertion.value
                    && let Some(n) = val.as_u64()
                {
                    let _ = writeln!(out, "\tassert.Equal(t, {n}, {expr})");
                } else {
                    let _ = writeln!(out, "{}", value_skip());
                }
            }
            "not_empty" => {
                let _ = writeln!(out, "\tassert.NotEmpty(t, {expr}, \"expected non-empty\")");
            }
            "is_empty" => {
                let _ = writeln!(out, "\tassert.Empty(t, {expr}, \"expected empty\")");
            }
            "is_true" => {
                let _ = writeln!(out, "\tassert.True(t, {expr}, \"expected true\")");
            }
            "is_false" => {
                let _ = writeln!(out, "\tassert.False(t, {expr}, \"expected false\")");
            }
            "greater_than" => {
                if let Some(val) = &assertion.value
                    && let Some(n) = val.as_u64()
                {
                    let _ = writeln!(out, "\tassert.Greater(t, {expr}, {n}, \"expected > {n}\")");
                } else {
                    let _ = writeln!(out, "{}", value_skip());
                }
            }
            "greater_than_or_equal" => {
                if let Some(val) = &assertion.value
                    && let Some(n) = val.as_u64()
                {
                    let _ = writeln!(out, "\tassert.GreaterOrEqual(t, {expr}, {n}, \"expected >= {n}\")");
                } else {
                    let _ = writeln!(out, "{}", value_skip());
                }
            }
            "contains" => {
                if let Some(serde_json::Value::String(s)) = &assertion.value {
                    let escaped = crate::e2e::escape::go_string_literal(s);
                    let _ = writeln!(out, "\tassert.Contains(t, {expr}, {escaped}, \"expected to contain\")");
                } else {
                    let _ = writeln!(out, "{}", value_skip());
                }
            }
            _ => {
                let _ = writeln!(
                    out,
                    "{}",
                    streaming_assertion_type_skip_line("\t", "//", f, &assertion.assertion_type)
                );
            }
        }
    } else {
        // ~keep The accessor returns `None` for reachable inputs (a `stream.has_*_event`
        // predicate with no resolved item type, for one), and this branch used to be absent:
        // the assertion vanished with no line for `fail_on_unavailable_field_markers` to see,
        // so a clean strict-gate run was indistinguishable from one that dropped it. alef's
        // own streaming adapter owns the gap, so it is counted, never fatal.
        let _ = writeln!(
            out,
            "\t// skipped: {}",
            FieldSkip::StreamingAssertionOnUnsupportedField.message(f)
        );
    }

    true
}