openethereum/rpc/src/tests/rpc.rs

// Copyright 2015-2020 Parity Technologies (UK) Ltd.
// This file is part of OpenEthereum.

// OpenEthereum 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.

// OpenEthereum 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 OpenEthereum.  If not, see <http://www.gnu.org/licenses/>.

use http::{self, hyper};
use jsonrpc_core::MetaIoHandler;

use tests::{helpers::Server, http_client};
use v1::{extractors, Metadata};
use HttpServer;

fn serve(handler: Option<MetaIoHandler<Metadata>>) -> Server<HttpServer> {
    let address = "127.0.0.1:0".parse().unwrap();
    let handler = handler.unwrap_or_default();

    Server::new(|_remote| {
        ::start_http_with_middleware(
            &address,
            http::DomainsValidation::Disabled,
            http::DomainsValidation::Disabled,
            handler,
            extractors::RpcExtractor,
            |request: hyper::Request<hyper::Body>| http::RequestMiddlewareAction::Proceed {
                should_continue_on_invalid_cors: false,
                request,
            },
            1,
            5,
            false,
        )
        .unwrap()
    })
}

/// Test a single request to running server
fn request(server: Server<HttpServer>, request: &str) -> http_client::Response {
    http_client::request(server.server.address(), request)
}

#[cfg(test)]
mod tests {
    use super::{request, Server};
    use jsonrpc_core::{MetaIoHandler, Value};
    use v1::Metadata;

    fn serve() -> (Server<::HttpServer>, ::std::net::SocketAddr) {
        let mut io = MetaIoHandler::default();
        io.add_method_with_meta("hello", |_, meta: Metadata| {
            Ok(Value::String(format!("{}", meta.origin)))
        });
        let server = super::serve(Some(io));
        let address = server.server.address().to_owned();

        (server, address)
    }

    #[test]
    fn should_extract_rpc_origin() {
        // given
        let (server, address) = serve();

        // when
        let req = r#"{"method":"hello","params":[],"jsonrpc":"2.0","id":1}"#;
        let expected = "{\"jsonrpc\":\"2.0\",\"result\":\"unknown origin / unknown agent via RPC\",\"id\":1}\n";
        let res = request(
            server,
            &format!(
                "\
				POST / HTTP/1.1\r\n\
				Host: {}\r\n\
				Content-Type: application/json\r\n\
				Content-Length: {}\r\n\
				Connection: close\r\n\
				\r\n\
				{}
			",
                address,
                req.len(),
                req
            ),
        );

        // then
        res.assert_status("HTTP/1.1 200 OK");
        assert_eq!(res.body, expected);
    }

    #[test]
    fn should_extract_rpc_origin_with_service() {
        // given
        let (server, address) = serve();

        // when
        let req = r#"{"method":"hello","params":[],"jsonrpc":"2.0","id":1}"#;
        let expected =
            "{\"jsonrpc\":\"2.0\",\"result\":\"unknown origin / curl/7.16.3 via RPC\",\"id\":1}\n";
        let res = request(
            server,
            &format!(
                "\
				POST / HTTP/1.1\r\n\
				Host: {}\r\n\
				Content-Type: application/json\r\n\
				Content-Length: {}\r\n\
				Connection: close\r\n\
				User-Agent: curl/7.16.3\r\n\
				\r\n\
				{}
			",
                address,
                req.len(),
                req
            ),
        );

        // then
        res.assert_status("HTTP/1.1 200 OK");
        assert_eq!(res.body, expected);
    }

    #[test]
    fn should_respond_valid_to_any_requested_header() {
        // given
        let (server, address) = serve();
        let headers = "Something, Anything, Xyz, 123, _?";

        // when
        let res = request(
            server,
            &format!(
                "\
				OPTIONS / HTTP/1.1\r\n\
				Host: {}\r\n\
				Origin: http://openethereum.github.io\r\n\
				Content-Length: 0\r\n\
				Content-Type: application/json\r\n\
				Connection: close\r\n\
				Access-Control-Request-Headers: {}\r\n\
				\r\n\
			",
                address, headers
            ),
        );

        // then
        assert_eq!(res.status, "HTTP/1.1 200 OK".to_owned());
        let expected = format!("access-control-allow-headers: {}", headers);
        assert!(
            res.headers.contains(&expected),
            "Headers missing in {:?}",
            res.headers
        );
    }
}