proptest-arbitrary 0.2.0

The Arbitrary trait for proptest and implementations for the standard library. 
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

use super::*;
use std::net::*;
use std::net::Shutdown::*;

// TODO: Can we design a workable semantic for PBT wrt. actual networking
// connections?

impl_just!(AddrParseError, "".parse::<Ipv4Addr>().unwrap_err());

impl_arbitrary!(Ipv4Addr,
    TupleUnion<(
        W<Just<Self>>,
        W<Just<Self>>,
        W<FMapped<'a, u32, Self>>,
    )>,
    prop_oneof![
        1  => Just(Self::new(0, 0, 0, 0)),
        4  => Just(Self::new(127, 0, 0, 1)),
        10 => any_sinto::<u32, _>()
    ]
);
impl_arbitrary!(Ipv6Addr,
    TupleUnion<(W<SMapped<'a, Ipv4Addr, Self>>, W<FMapped<'a, [u16; 8], Self>>)>,
    prop_oneof![
        2 => any_with_smap((), |ip| ip.to_ipv6_mapped()),
        1 => any_sinto::<[u16; 8], _>()
    ]
);
impl_arbitrary!(SocketAddrV4, SMapped<'a, (Ipv4Addr, u16), Self>,
    static_map(any::<(Ipv4Addr, u16)>(), |(a, b)| Self::new(a, b))
);
impl_arbitrary!(SocketAddrV6, SMapped<'a, (Ipv6Addr, u16, u32, u32), Self>,
    static_map(any::<(Ipv6Addr, u16, u32, u32)>(),
        |(a, b, c, d)| Self::new(a, b, c, d))
);
impl_arbitrary!(IpAddr,
    TupleUnion<(W<FMapped<'a, Ipv4Addr, Self>>,
                W<FMapped<'a, Ipv6Addr, Self>>)>,
    prop_oneof![any_sinto::<Ipv4Addr, _>(), any_sinto::<Ipv6Addr, _>()]
);
impl_arbitrary!(Shutdown,
    TupleUnion<(W<Just<Self>>, W<Just<Self>>, W<Just<Self>>)>,
    prop_oneof![Just(Both), Just(Read), Just(Write)]
);
impl_arbitrary!(SocketAddr,
    TupleUnion<(W<FMapped<'a, SocketAddrV4, Self>>,
                W<FMapped<'a, SocketAddrV6, Self>>)>,
    prop_oneof![any_sinto::<SocketAddrV4, _>(), any_sinto::<SocketAddrV6, _>()]
);

#[cfg(feature = "nightly")]
impl_arbitrary!(Ipv6MulticastScope,
    TupleUnion<( W<Just<Self>>, W<Just<Self>>, W<Just<Self>>
               , W<Just<Self>>, W<Just<Self>>, W<Just<Self>>, W<Just<Self>>)>,
    {
        use std::net::Ipv6MulticastScope::*;
        prop_oneof![
            Just(InterfaceLocal),
            Just(LinkLocal),
            Just(RealmLocal),
            Just(AdminLocal),
            Just(SiteLocal),
            Just(OrganizationLocal),
            Just(Global),
        ]
    }
);