summaryrefslogtreecommitdiffstats
path: root/tokio-test/tests/macros.rs
blob: 2183fc8373dca67ab8ee4bea4984bc637dae8973 (plain)
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
#![warn(rust_2018_idioms)]

use std::task::Poll;
use tokio_test::{
    assert_pending, assert_ready, assert_ready_eq, assert_ready_err, assert_ready_ok,
};

fn ready() -> Poll<()> {
    Poll::Ready(())
}

fn ready_ok() -> Poll<Result<(), ()>> {
    Poll::Ready(Ok(()))
}

fn ready_err() -> Poll<Result<(), ()>> {
    Poll::Ready(Err(()))
}

fn pending() -> Poll<()> {
    Poll::Pending
}

#[derive(Debug)]
enum Test {
    Data,
}

#[test]
fn assert_ready() {
    let poll = ready();
    assert_ready!(poll);
    assert_ready!(poll, "some message");
    assert_ready!(poll, "{:?}", ());
    assert_ready!(poll, "{:?}", Test::Data);
}

#[test]
#[should_panic]
fn assert_ready_on_pending() {
    let poll = pending();
    assert_ready!(poll);
}

#[test]
fn assert_pending() {
    let poll = pending();
    assert_pending!(poll);
    assert_pending!(poll, "some message");
    assert_pending!(poll, "{:?}", ());
    assert_pending!(poll, "{:?}", Test::Data);
}

#[test]
#[should_panic]
fn assert_pending_on_ready() {
    let poll = ready();
    assert_pending!(poll);
}

#[test]
fn assert_ready_ok() {
    let poll = ready_ok();
    assert_ready_ok!(poll);
    assert_ready_ok!(poll, "some message");
    assert_ready_ok!(poll, "{:?}", ());
    assert_ready_ok!(poll, "{:?}", Test::Data);
}

#[test]
#[should_panic]
fn assert_ok_on_err() {
    let poll = ready_err();
    assert_ready_ok!(poll);
}

#[test]
fn assert_ready_err() {
    let poll = ready_err();
    assert_ready_err!(poll);
    assert_ready_err!(poll, "some message");
    assert_ready_err!(poll, "{:?}", ());
    assert_ready_err!(poll, "{:?}", Test::Data);
}

#[test]
#[should_panic]
fn assert_err_on_ok() {
    let poll = ready_ok();
    assert_ready_err!(poll);
}

#[test]
fn assert_ready_eq() {
    let poll = ready();
    assert_ready_eq!(poll, ());
    assert_ready_eq!(poll, (), "some message");
    assert_ready_eq!(poll, (), "{:?}", ());
    assert_ready_eq!(poll, (), "{:?}", Test::Data);
}

#[test]
#[should_panic]
fn assert_eq_on_not_eq() {
    let poll = ready_err();
    assert_ready_eq!(poll, Ok(()));
}