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

//-
// Copyright 2017, 2018 The proptest developers
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

//! Arbitrary implementations for `std::time`.

use core::ops::Range;
use std::time::*;

use crate::arbitrary::*;
use crate::num;
use crate::strategy::statics::{self, static_map};

arbitrary!(Duration, SMapped<(u64, u32), Self>;
    static_map(any::<(u64, u32)>(), |(a, b)| Duration::new(a, b))
);

// Instant::now() "never" returns the same Instant, so no shrinking may occur!
arbitrary!(Instant; Self::now());

arbitrary!(
    // We can't use `any::<Duration>()` because the addition to `SystemTime`
    // can overflow and panic. To be conservative, we only allow seconds to go
    // to i32::MAX since a certain popular OS still uses `i32` to represent the
    // seconds counter.
    SystemTime, statics::Map<(num::i32::Any, Range<u32>),
                             fn ((i32, u32)) -> SystemTime>;
    static_map((num::i32::ANY, 0..1_000_000_000u32),
                |(sec, ns)| {
                    if sec >= 0 {
                        UNIX_EPOCH + Duration::new(sec as u64, ns)
                    } else {
                        UNIX_EPOCH - Duration::new((-(sec as i64)) as u64, ns)
                    }
                })
);

#[cfg(test)]
mod test {
    no_panic_test!(
        duration => Duration,
        instant  => Instant,
        system_time => SystemTime
    );
}