105 lines
8.0 KiB
Rust
105 lines
8.0 KiB
Rust
|
// 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/>.
|
||
|
|
||
|
use std::sync::Arc;
|
||
|
|
||
|
use jsonrpc_core::MetaIoHandler;
|
||
|
use jsonrpc_core::futures::{self, Stream, Future};
|
||
|
use jsonrpc_pubsub::Session;
|
||
|
|
||
|
use v1::{EthPubSub, EthPubSubClient, Metadata};
|
||
|
|
||
|
use ethcore::client::{TestBlockChainClient, EachBlockWith, ChainNotify};
|
||
|
use parity_reactor::EventLoop;
|
||
|
|
||
|
#[test]
|
||
|
fn should_subscribe_to_new_heads() {
|
||
|
// given
|
||
|
let el = EventLoop::spawn();
|
||
|
let mut client = TestBlockChainClient::new();
|
||
|
// Insert some blocks
|
||
|
client.add_blocks(3, EachBlockWith::Nothing);
|
||
|
let h3 = client.block_hash_delta_minus(1);
|
||
|
let h2 = client.block_hash_delta_minus(2);
|
||
|
let h1 = client.block_hash_delta_minus(3);
|
||
|
|
||
|
let pubsub = EthPubSubClient::new(Arc::new(client), el.remote());
|
||
|
let handler = pubsub.handler();
|
||
|
let pubsub = pubsub.to_delegate();
|
||
|
|
||
|
let mut io = MetaIoHandler::default();
|
||
|
io.extend_with(pubsub);
|
||
|
|
||
|
let mut metadata = Metadata::default();
|
||
|
let (sender, receiver) = futures::sync::mpsc::channel(8);
|
||
|
metadata.session = Some(Arc::new(Session::new(sender)));
|
||
|
|
||
|
// Subscribe
|
||
|
let request = r#"{"jsonrpc": "2.0", "method": "eth_subscribe", "params": ["newHeads"], "id": 1}"#;
|
||
|
let response = r#"{"jsonrpc":"2.0","result":1,"id":1}"#;
|
||
|
assert_eq!(io.handle_request_sync(request, metadata.clone()), Some(response.to_owned()));
|
||
|
|
||
|
// Check notifications
|
||
|
handler.new_blocks(vec![], vec![], vec![h1], vec![], vec![], vec![], 0);
|
||
|
let (res, receiver) = receiver.into_future().wait().unwrap();
|
||
|
let response = r#"{"jsonrpc":"2.0","method":"eth_subscription","params":{"result":{"author":"0x0000000000000000000000000000000000000000","difficulty":"0x1","extraData":"0x","gasLimit":"0xf4240","gasUsed":"0x0","hash":"0x3457d2fa2e3dd33c78ac681cf542e429becf718859053448748383af67e23218","logsBloom":"0x00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000","miner":"0x0000000000000000000000000000000000000000","number":"0x1","parentHash":"0x0cd786a2425d16f152c658316c423e6ce1181e15c3295826d7c9904cba9ce303","receiptsRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421","sealFields":[],"sha3Uncles":"0x1dcc4de8dec75d7aab85b567b6ccd41ad312451b948a7413f0a142fd40d49347","size":"0x1c9","stateRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421","timestamp":"0x0","transactionsRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421"},"subscription":1}}"#;
|
||
|
assert_eq!(res, Some(response.into()));
|
||
|
|
||
|
// Notify about two blocks
|
||
|
handler.new_blocks(vec![], vec![], vec![h2, h3], vec![], vec![], vec![], 0);
|
||
|
|
||
|
// Receive both
|
||
|
let (res, receiver) = receiver.into_future().wait().unwrap();
|
||
|
let response = r#"{"jsonrpc":"2.0","method":"eth_subscription","params":{"result":{"author":"0x0000000000000000000000000000000000000000","difficulty":"0x2","extraData":"0x","gasLimit":"0xf4240","gasUsed":"0x0","hash":"0x44e5ecf454ea99af9d8a8f2ca0daba96964c90de05db7a78f59b84ae9e749706","logsBloom":"0x00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000","miner":"0x0000000000000000000000000000000000000000","number":"0x2","parentHash":"0x3457d2fa2e3dd33c78ac681cf542e429becf718859053448748383af67e23218","receiptsRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421","sealFields":[],"sha3Uncles":"0x1dcc4de8dec75d7aab85b567b6ccd41ad312451b948a7413f0a142fd40d49347","size":"0x1c9","stateRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421","timestamp":"0x0","transactionsRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421"},"subscription":1}}"#;
|
||
|
assert_eq!(res, Some(response.into()));
|
||
|
let (res, receiver) = receiver.into_future().wait().unwrap();
|
||
|
let response = r#"{"jsonrpc":"2.0","method":"eth_subscription","params":{"result":{"author":"0x0000000000000000000000000000000000000000","difficulty":"0x3","extraData":"0x","gasLimit":"0xf4240","gasUsed":"0x0","hash":"0xdf04a98bb0c6fa8441bd429822f65a46d0cb553f6bcef602b973e65c81497f8e","logsBloom":"0x00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000","miner":"0x0000000000000000000000000000000000000000","number":"0x3","parentHash":"0x44e5ecf454ea99af9d8a8f2ca0daba96964c90de05db7a78f59b84ae9e749706","receiptsRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421","sealFields":[],"sha3Uncles":"0x1dcc4de8dec75d7aab85b567b6ccd41ad312451b948a7413f0a142fd40d49347","size":"0x1c9","stateRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421","timestamp":"0x0","transactionsRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421"},"subscription":1}}"#;
|
||
|
assert_eq!(res, Some(response.into()));
|
||
|
|
||
|
// And unsubscribe
|
||
|
let request = r#"{"jsonrpc": "2.0", "method": "eth_unsubscribe", "params": [1], "id": 1}"#;
|
||
|
let response = r#"{"jsonrpc":"2.0","result":true,"id":1}"#;
|
||
|
assert_eq!(io.handle_request_sync(request, metadata), Some(response.to_owned()));
|
||
|
|
||
|
let (res, _receiver) = receiver.into_future().wait().unwrap();
|
||
|
assert_eq!(res, None);
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn should_return_unimplemented() {
|
||
|
// given
|
||
|
let el = EventLoop::spawn();
|
||
|
let client = TestBlockChainClient::new();
|
||
|
let pubsub = EthPubSubClient::new(Arc::new(client), el.remote());
|
||
|
let pubsub = pubsub.to_delegate();
|
||
|
|
||
|
let mut io = MetaIoHandler::default();
|
||
|
io.extend_with(pubsub);
|
||
|
|
||
|
let mut metadata = Metadata::default();
|
||
|
let (sender, _receiver) = futures::sync::mpsc::channel(8);
|
||
|
metadata.session = Some(Arc::new(Session::new(sender)));
|
||
|
|
||
|
// Subscribe
|
||
|
let response = r#"{"jsonrpc":"2.0","error":{"code":-32000,"message":"This request is not implemented yet. Please create an issue on Github repo."},"id":1}"#;
|
||
|
let request = r#"{"jsonrpc": "2.0", "method": "eth_subscribe", "params": ["newPendingTransactions"], "id": 1}"#;
|
||
|
assert_eq!(io.handle_request_sync(request, metadata.clone()), Some(response.to_owned()));
|
||
|
let request = r#"{"jsonrpc": "2.0", "method": "eth_subscribe", "params": ["logs"], "id": 1}"#;
|
||
|
assert_eq!(io.handle_request_sync(request, metadata.clone()), Some(response.to_owned()));
|
||
|
let request = r#"{"jsonrpc": "2.0", "method": "eth_subscribe", "params": ["syncing"], "id": 1}"#;
|
||
|
assert_eq!(io.handle_request_sync(request, metadata.clone()), Some(response.to_owned()));
|
||
|
}
|