nextest-runner 0.114.0

Core runner logic for cargo nextest.
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

// Copyright (c) The nextest Contributors
// SPDX-License-Identifier: MIT OR Apache-2.0

//! Test helpers for proptest support in reporter types.

use crate::{
    config::{
        core::ConfigIdentifier,
        elements::{CustomTestGroup, TestGroup},
        scripts::ScriptId,
    },
    reporter::events::TestSlotAssignment,
};
use chrono::{DateTime, FixedOffset, TimeZone, Utc};
use proptest::prelude::*;
use smol_str::SmolStr;

/// Strategy for generating arbitrary `Duration` values.
pub fn arb_duration() -> impl Strategy<Value = std::time::Duration> {
    any::<u64>().prop_map(std::time::Duration::from_millis)
}

/// Strategy for generating arbitrary `DateTime<FixedOffset>` values.
pub fn arb_datetime_fixed_offset() -> impl Strategy<Value = DateTime<FixedOffset>> {
    // Generate a timestamp in a reasonable range (year 2000-2100).
    // Offset must be in minutes (not seconds) to survive JSON round-tripping,
    // since RFC 3339 serialization truncates sub-minute offset precision.
    (946684800i64..4102444800i64, -720i32..720i32).prop_map(|(secs, offset_minutes)| {
        let offset_secs = offset_minutes * 60;
        let offset =
            FixedOffset::east_opt(offset_secs).unwrap_or_else(|| FixedOffset::east_opt(0).unwrap());
        let utc = Utc
            .timestamp_opt(secs, 0)
            .single()
            .expect("valid timestamp");
        utc.with_timezone(&offset)
    })
}

/// Strategy for generating arbitrary `SmolStr` values.
pub fn arb_smol_str() -> impl Strategy<Value = SmolStr> {
    any::<String>().prop_map(SmolStr::new)
}

impl Arbitrary for ConfigIdentifier {
    type Parameters = ();
    type Strategy = BoxedStrategy<Self>;

    fn arbitrary_with(_: Self::Parameters) -> Self::Strategy {
        // Generate valid identifiers: start with XID_Start, followed by XID_Continue or hyphens.
        // For simplicity, use ASCII letters and hyphens.
        let regular_identifier = "[a-zA-Z][a-zA-Z0-9-]{0,20}";
        let tool_identifier = "@tool:[a-zA-Z][a-zA-Z0-9-]{0,10}:[a-zA-Z][a-zA-Z0-9-]{0,10}";

        prop_oneof![
            3 => regular_identifier
                .prop_map(|s| ConfigIdentifier::new(s.into()).expect("valid identifier")),
            1 => tool_identifier
                .prop_map(|s| ConfigIdentifier::new(s.into()).expect("valid tool identifier")),
        ]
        .boxed()
    }
}

impl Arbitrary for ScriptId {
    type Parameters = ();
    type Strategy = BoxedStrategy<Self>;

    fn arbitrary_with(_: Self::Parameters) -> Self::Strategy {
        any::<ConfigIdentifier>().prop_map(ScriptId).boxed()
    }
}

/// Creates a `TestSlotAssignment` for the global group with the given slot.
pub fn global_slot_assignment(global_slot: u64) -> TestSlotAssignment {
    TestSlotAssignment {
        global_slot,
        group_slot: None,
        test_group: TestGroup::Global,
    }
}

impl Arbitrary for TestGroup {
    type Parameters = ();
    type Strategy = BoxedStrategy<Self>;

    fn arbitrary_with(_: Self::Parameters) -> Self::Strategy {
        prop_oneof![
            1 => Just(TestGroup::Global),
            3 => any::<ConfigIdentifier>()
                .prop_map(|id| TestGroup::Custom(CustomTestGroup::from_identifier(id))),
        ]
        .boxed()
    }
}

impl Arbitrary for TestSlotAssignment {
    type Parameters = ();
    type Strategy = BoxedStrategy<Self>;

    fn arbitrary_with(_: Self::Parameters) -> Self::Strategy {
        (any::<u64>(), any::<Option<u64>>(), any::<TestGroup>())
            .prop_map(|(global_slot, group_slot, test_group)| TestSlotAssignment {
                global_slot,
                group_slot,
                test_group,
            })
            .boxed()
    }
}