hooch 0.1.6

A small async runtime that's been bootlegged for my projects.
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

use std::{
    io::{Read, Write},
    sync::{Arc, Mutex},
};

pub use hooch::net::HoochTcpListener;
use hooch::{
    net::HoochTcpStream,
    runtime::{Handle, RuntimeBuilder},
};

fn build_runtime() -> Handle {
    RuntimeBuilder::default().build()
}

fn get_free_port() -> std::io::Result<u16> {
    let listener = std::net::TcpListener::bind("127.0.0.1:0")?; // Bind to any available port
    listener.local_addr().map(|addr| addr.port())
}

#[test]
fn test_accept_tcp_listener() {
    let port = get_free_port().unwrap();
    let addr = format!("localhost:{}", port);
    let addr_clone = addr.clone();

    let result = Arc::new(Mutex::new(false));

    let result_clone = Arc::clone(&result);

    let (tx, rx) = std::sync::mpsc::sync_channel(1);
    let (tx_bind, rx_bind) = std::sync::mpsc::sync_channel(1);
    std::thread::spawn(|| {
        let runtime_handle = build_runtime();

        runtime_handle.run_blocking(async move {
            let listener = HoochTcpListener::bind(addr_clone).await.unwrap();
            let _ = tx_bind.send(());
            if listener.accept().await.is_ok() {
                *result_clone.lock().unwrap() = true;
            }
            let _ = tx.send(());
        })
    });

    rx_bind.recv().unwrap();

    std::thread::spawn(|| {
        let _ = std::net::TcpStream::connect(addr);
    });

    rx.recv().unwrap();

    assert!(*result.lock().unwrap());
}

#[test]
fn test_tcp_stream_connect() {
    let port = get_free_port().unwrap();
    let addr = format!("localhost:{}", port);
    let addr_clone = addr.clone();

    let (tx_listen, rx_listen) = std::sync::mpsc::sync_channel(1);
    std::thread::spawn(move || {
        let listener = std::net::TcpListener::bind(addr_clone).unwrap();
        let _ = tx_listen.send(());
        let _ = listener.accept();
    });

    rx_listen.recv().unwrap();

    let runtime_handler = build_runtime();
    let result = runtime_handler.run_blocking(async move { HoochTcpStream::connect(addr).await });

    assert!(result.is_ok())
}

#[test]
fn test_tcp_stream_read() {
    let port = get_free_port().unwrap();
    let addr = format!("localhost:{}", port);
    let addr_clone = addr.clone();

    let (tx_listen, rx_listen) = std::sync::mpsc::sync_channel(1);
    std::thread::spawn(move || {
        let listener = std::net::TcpListener::bind(addr_clone).unwrap();
        let _ = tx_listen.send(());
        let (mut stream, _) = listener.accept().unwrap();
        stream.write_all(&[1]).unwrap();
    });

    rx_listen.recv().unwrap();

    let runtime_handler = build_runtime();
    let result = runtime_handler.run_blocking(async move {
        let mut buffer = [0; 1];
        let mut hooch_stream = HoochTcpStream::connect(addr).await.unwrap();
        hooch_stream.read(&mut buffer).await.unwrap();
        buffer
    });

    assert!(result == [1])
}

#[test]
fn test_tcp_stream_write() {
    let port = get_free_port().unwrap();
    let addr = format!("localhost:{}", port);
    let addr_clone = addr.clone();

    let (tx_listen, rx_listen) = std::sync::mpsc::sync_channel(1);
    let handle = std::thread::spawn(move || {
        let listener = std::net::TcpListener::bind(addr_clone).unwrap();
        let _ = tx_listen.send(());
        let (mut stream, _) = listener.accept().unwrap();
        let mut buf = [0; 1];
        stream.read_exact(&mut buf).unwrap();
        buf
    });

    rx_listen.recv().unwrap();

    let (tx, rx) = std::sync::mpsc::sync_channel(1);
    let runtime_handler = build_runtime();
    runtime_handler.run_blocking(async move {
        let buffer = [3; 1];
        let mut hooch_stream = HoochTcpStream::connect(addr).await.unwrap();
        hooch_stream.write(&buffer).await.unwrap();
        let _ = tx.send(());
    });

    rx.recv().unwrap();
    let result = handle.join().unwrap();
    assert!(result == [3])
}