openethereum/ipc/tests/over_nano.rs

78 lines
2.4 KiB
Rust
Raw Normal View History

// Copyright 2015-2017 Parity Technologies (UK) Ltd.
// This file is part of Parity.
// Parity is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// Parity is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License
// along with Parity. If not, see <http://www.gnu.org/licenses/>.
#[cfg(test)]
mod tests {
use super::super::service::*;
2016-06-30 15:23:10 +02:00
use super::super::with_attrs::PrettyNamedClient;
use nanoipc;
use std::sync::Arc;
2016-04-23 17:15:50 +02:00
use std::io::Write;
2016-04-12 10:41:26 +02:00
use std::sync::atomic::{Ordering, AtomicBool};
2016-04-12 10:34:56 +02:00
fn dummy_write(addr: &str, buf: &[u8]) -> (::nanomsg::Socket, ::nanomsg::Endpoint) {
let mut socket = ::nanomsg::Socket::new(::nanomsg::Protocol::Pair).unwrap();
let endpoint = socket.connect(addr).unwrap();
socket.write(buf).unwrap();
(socket, endpoint)
}
fn init_worker(addr: &str) -> nanoipc::Worker<Service> {
2016-04-14 19:43:14 +02:00
let mut worker = nanoipc::Worker::<Service>::new(&Arc::new(Service::new()));
worker.add_duplex(addr).unwrap();
worker
}
#[test]
fn can_create_client() {
2016-04-14 17:49:25 +02:00
let client = nanoipc::init_duplex_client::<ServiceClient<_>>("ipc:///tmp/parity-nano-test10.ipc");
assert!(client.is_ok());
}
2016-06-30 15:23:10 +02:00
#[test]
fn can_create_renamed_client() {
let client = nanoipc::init_duplex_client::<PrettyNamedClient<_>>("ipc:///tmp/parity-nano-test10.ipc");
assert!(client.is_ok());
}
#[test]
fn can_call_handshake() {
2016-04-12 10:34:56 +02:00
let url = "ipc:///tmp/parity-test-nano-20.ipc";
2016-04-12 10:41:26 +02:00
let worker_should_exit = Arc::new(AtomicBool::new(false));
let worker_is_ready = Arc::new(AtomicBool::new(false));
2016-04-12 10:13:27 +02:00
let c_worker_should_exit = worker_should_exit.clone();
let c_worker_is_ready = worker_is_ready.clone();
::std::thread::spawn(move || {
let mut worker = init_worker(url);
2016-04-12 10:41:26 +02:00
while !c_worker_should_exit.load(Ordering::Relaxed) {
2016-04-12 10:13:27 +02:00
worker.poll();
2016-04-12 10:41:26 +02:00
c_worker_is_ready.store(true, Ordering::Relaxed);
2016-04-12 10:13:27 +02:00
}
});
2016-04-12 10:13:27 +02:00
2016-04-12 10:41:26 +02:00
while !worker_is_ready.load(Ordering::Relaxed) { }
2016-04-14 17:49:25 +02:00
let client = nanoipc::init_duplex_client::<ServiceClient<_>>(url).unwrap();
let hs = client.handshake();
2016-04-12 10:41:26 +02:00
worker_should_exit.store(true, Ordering::Relaxed);
assert!(hs.is_ok());
}
}