scaproust 0.3.2

Nanomsg scalability protocols implementation in rust. Various messaging patterns over pluggable transports
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

// Copyright (c) 2015-2017 Contributors as noted in the AUTHORS file.
//
// 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.

use transport::async::stub::*;
use transport::async::state::*;
use transport::async::handshake::HandshakeTx; 
use transport::pipe::Context;

pub struct Initial<S : AsyncPipeStub> {
    stub: S,
    proto_ids: (u16, u16)
}

impl<S : AsyncPipeStub> Initial<S> {
    pub fn new(s: S, pids: (u16, u16)) -> Initial<S> {
        Initial {
            stub: s,
            proto_ids: pids
        }
    }
}

impl<S : AsyncPipeStub> Into<HandshakeTx<S>> for Initial<S> {
    fn into(self) -> HandshakeTx<S> {
        HandshakeTx::new(self.stub, self.proto_ids)
    }
}

impl<S : AsyncPipeStub + 'static> PipeState<S> for Initial<S> {

    fn name(&self) -> &'static str {"Initial"}
    
    fn open(self: Box<Self>, ctx: &mut Context) -> Box<PipeState<S>> {
        transition::<Initial<S>, HandshakeTx<S>, S>(self, ctx)
    }

}

#[cfg(test)]
mod tests {
    use transport::tests::*;
    use transport::async::state::*;
    use transport::async::tests::*;
    use transport::async::initial::*;

    #[test]
    fn open_should_cause_transition_to_handshake() {
        let stub = TestStepStream::new();
        let state = Box::new(Initial::new(stub, (1, 1)));
        let mut ctx = TestPipeContext::new();
        let new_state = state.open(&mut ctx);

        assert_eq!(1, ctx.get_registrations().len()); // this is caused by HandshakeTx::enter
        assert_eq!(0, ctx.get_reregistrations().len());
        assert_eq!(0, ctx.get_deregistrations());

        assert_eq!("HandshakeTx", new_state.name());
    }

    #[test]
    fn close_should_cause_a_transition_to_dead() {
        let stub = TestStepStream::new();
        let state = Box::new(Initial::new(stub, (1, 1)));
        let mut ctx = TestPipeContext::new();
        let new_state = state.close(&mut ctx);

        assert_eq!(0, ctx.get_registrations().len());
        assert_eq!(0, ctx.get_reregistrations().len());
        assert_eq!(0, ctx.get_deregistrations());

        assert_eq!("Dead", new_state.name());
    }
}