memsocket 0.1.3

An asynchronous in-memory socket-like interface.
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

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

extern crate memsocket;
extern crate tokio;

use std::str::from_utf8;
use tokio::prelude::*;

#[test]
fn write_then_read() {
    let (client, server) = memsocket::unbounded();

    tokio::runtime::current_thread::Runtime::new()
        .unwrap()
        .block_on({
            tokio::io::write_all(client, "hello world")
                .map(|(mut client, _)| client.shutdown())
                .and_then(|_| tokio::io::read_to_end(server, vec![]))
                .map(|(_, result)| assert_eq!(from_utf8(&result), Ok("hello world")))
                .map_err(|error| panic!("{:?}", error))
        })
        .unwrap();
}

#[test]
fn read_then_write() {
    let (client, server) = memsocket::unbounded();

    tokio::runtime::current_thread::Runtime::new()
        .unwrap()
        .block_on({
            tokio::io::read_to_end(server, vec![])
                .map(|(_, result)| assert_eq!(from_utf8(&result), Ok("hello world")))
                .join(
                    tokio::io::write_all(client, "hello world")
                        .map(|(mut client, _)| client.shutdown()),
                )
                .map_err(|error| panic!("{:?}", error))
        })
        .unwrap();
}

#[test]
fn write_then_read_partial() {
    let (client, mut server) = memsocket::unbounded();

    tokio::runtime::current_thread::Runtime::new()
        .unwrap()
        .block_on({
            tokio::io::write_all(client, "hello world")
                .and_then(|(client, _)| {
                    let mut buf = vec![0; 12];
                    server.read(&mut buf).map(|_| (client, buf))
                })
                .map(|(mut client, buf)| {
                    client.shutdown().unwrap();
                    buf
                })
                .map(|result| assert_eq!(from_utf8(&result), Ok("hello world\u{0}")))
                .map_err(|error| panic!("{:?}", error))
        })
        .unwrap();
}