147 lines
3.9 KiB
Rust
147 lines
3.9 KiB
Rust
// Copyright 2015, 2016 Ethcore (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::*;
|
|
use super::super::nested::DBClient;
|
|
use ipc::*;
|
|
use devtools::*;
|
|
use semver::Version;
|
|
|
|
#[test]
|
|
fn call_service() {
|
|
// method_num = 0, f = 10 (method Service::commit)
|
|
let mut socket = TestSocket::new_ready(vec![0, 16, 0, 0, 0, 10]);
|
|
|
|
let service = Service::new();
|
|
assert_eq!(0, *service.commits.read().unwrap());
|
|
|
|
service.dispatch(&mut socket);
|
|
|
|
assert_eq!(10, *service.commits.read().unwrap());
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn call_service_handshake() {
|
|
let mut socket = TestSocket::new_ready(vec![0, 0,
|
|
// protocol version
|
|
0, 0, 0, 0, 0, 0, 0, 5, b'1', b'.', b'0', b'.', b'0',
|
|
// api version
|
|
0, 0, 0, 0, 0, 0, 0, 5, b'1', b'.', b'0', b'.', b'0',
|
|
// reserved
|
|
0, 0, 0, 0, 0, 0, 0, 64,
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
]);
|
|
|
|
let service = Service::new();
|
|
let result = service.dispatch(&mut socket);
|
|
|
|
// single `true`
|
|
assert_eq!(vec![1], result);
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn call_service_client() {
|
|
let mut socket = TestSocket::new();
|
|
socket.read_buffer = vec![0, 0, 0, 10];
|
|
let service_client = ServiceClient::init(socket);
|
|
|
|
let result = service_client.commit(5);
|
|
|
|
assert_eq!(vec![0, 16, 0, 0, 0, 5], service_client.socket().borrow().write_buffer.clone());
|
|
assert_eq!(10, result);
|
|
}
|
|
|
|
#[test]
|
|
fn call_service_client_optional() {
|
|
let mut socket = TestSocket::new();
|
|
socket.read_buffer = vec![0, 0, 0, 10];
|
|
let service_client = ServiceClient::init(socket);
|
|
|
|
let result = service_client.rollback(Some(5), 10);
|
|
|
|
assert_eq!(vec![0, 17, 1, 0, 0, 0, 5, 0, 0, 0, 10], service_client.socket().borrow().write_buffer.clone());
|
|
assert_eq!(10, result);
|
|
}
|
|
|
|
#[test]
|
|
fn query_default_version() {
|
|
let ver = Service::protocol_version();
|
|
assert_eq!(ver, Version::parse("1.0.0").unwrap());
|
|
let ver = Service::api_version();
|
|
assert_eq!(ver, Version::parse("1.0.0").unwrap());
|
|
}
|
|
|
|
#[test]
|
|
fn call_service_client_handshake() {
|
|
let mut socket = TestSocket::new();
|
|
socket.read_buffer = vec![1];
|
|
let service_client = ServiceClient::init(socket);
|
|
|
|
let result = service_client.handshake();
|
|
|
|
assert!(result.is_ok());
|
|
}
|
|
|
|
#[test]
|
|
fn can_use_custom_params() {
|
|
let mut socket = TestSocket::new();
|
|
socket.read_buffer = vec![1];
|
|
let service_client = ServiceClient::init(socket);
|
|
|
|
let result = service_client.push_custom(CustomData { a: 3, b: 11});
|
|
|
|
assert_eq!(vec![
|
|
// message num..
|
|
0, 18,
|
|
// payload length
|
|
0, 0, 0, 0, 0, 0, 0, 16,
|
|
// structure raw bytes (bigendians :( )
|
|
3, 0, 0, 0, 0, 0, 0, 0,
|
|
11, 0, 0, 0, 0, 0, 0, 0],
|
|
service_client.socket().borrow().write_buffer.clone());
|
|
assert_eq!(true, result);
|
|
}
|
|
|
|
#[test]
|
|
fn can_invoke_generic_service() {
|
|
let mut socket = TestSocket::new();
|
|
socket.read_buffer = vec![0, 0, 0, 0];
|
|
let db_client = DBClient::<u64, _>::init(socket);
|
|
|
|
let result = db_client.write(vec![0u8; 100]);
|
|
|
|
assert!(result.is_ok());
|
|
}
|
|
#[test]
|
|
fn can_handshake_generic_service() {
|
|
let mut socket = TestSocket::new();
|
|
socket.read_buffer = vec![1];
|
|
let db_client = DBClient::<u64, _>::init(socket);
|
|
|
|
let result = db_client.handshake();
|
|
|
|
assert!(result.is_ok());
|
|
}
|
|
}
|