2018-09-21 14:33:23 +03:00
|
|
|
extern crate grin_api as api;
|
|
|
|
|
extern crate grin_util as util;
|
|
|
|
|
extern crate hyper;
|
|
|
|
|
|
|
|
|
|
use api::*;
|
|
|
|
|
use hyper::{Body, Request};
|
|
|
|
|
use std::net::SocketAddr;
|
2018-09-21 20:57:59 +03:00
|
|
|
use std::sync::atomic::{AtomicUsize, Ordering, ATOMIC_USIZE_INIT};
|
|
|
|
|
use std::sync::Arc;
|
2018-09-21 14:33:23 +03:00
|
|
|
use std::{thread, time};
|
|
|
|
|
|
|
|
|
|
struct IndexHandler {
|
|
|
|
|
list: Vec<String>,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl IndexHandler {}
|
|
|
|
|
|
|
|
|
|
impl Handler for IndexHandler {
|
|
|
|
|
fn get(&self, _req: Request<Body>) -> ResponseFuture {
|
|
|
|
|
json_response_pretty(&self.list)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2018-09-21 20:57:59 +03:00
|
|
|
pub struct CounterMiddleware {
|
|
|
|
|
counter: AtomicUsize,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl CounterMiddleware {
|
|
|
|
|
fn new() -> CounterMiddleware {
|
|
|
|
|
CounterMiddleware {
|
|
|
|
|
counter: ATOMIC_USIZE_INIT,
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn value(&self) -> usize {
|
|
|
|
|
self.counter.load(Ordering::SeqCst)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl Handler for CounterMiddleware {
|
|
|
|
|
fn call(
|
|
|
|
|
&self,
|
|
|
|
|
req: Request<Body>,
|
|
|
|
|
mut handlers: Box<Iterator<Item = HandlerObj>>,
|
|
|
|
|
) -> ResponseFuture {
|
|
|
|
|
self.counter.fetch_add(1, Ordering::SeqCst);
|
|
|
|
|
handlers.next().unwrap().call(req, handlers)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2018-09-21 14:33:23 +03:00
|
|
|
fn build_router() -> Router {
|
|
|
|
|
let route_list = vec!["get blocks".to_string(), "get chain".to_string()];
|
|
|
|
|
let index_handler = IndexHandler { list: route_list };
|
|
|
|
|
let mut router = Router::new();
|
|
|
|
|
router
|
2018-09-21 20:57:59 +03:00
|
|
|
.add_route("/v1/*", Arc::new(index_handler))
|
|
|
|
|
.expect("add_route failed")
|
|
|
|
|
.add_middleware(Arc::new(LoggingMiddleware {}));
|
2018-09-21 14:33:23 +03:00
|
|
|
router
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn test_start_api() {
|
|
|
|
|
util::init_test_logger();
|
|
|
|
|
let mut server = ApiServer::new();
|
2018-09-21 20:57:59 +03:00
|
|
|
let mut router = build_router();
|
|
|
|
|
let counter = Arc::new(CounterMiddleware::new());
|
|
|
|
|
// add middleware to the root
|
|
|
|
|
router.add_middleware(counter.clone());
|
2018-09-21 14:33:23 +03:00
|
|
|
let server_addr = "127.0.0.1:14434";
|
|
|
|
|
let addr: SocketAddr = server_addr.parse().expect("unable to parse server address");
|
2018-10-02 10:49:36 +03:00
|
|
|
assert!(server.start(addr, router, None).is_ok());
|
2018-09-21 14:33:23 +03:00
|
|
|
let url = format!("http://{}/v1/", server_addr);
|
2018-09-26 23:38:44 +03:00
|
|
|
let index = api::client::get::<Vec<String>>(url.as_str(), None).unwrap();
|
2018-09-21 14:33:23 +03:00
|
|
|
assert_eq!(index.len(), 2);
|
2018-09-21 20:57:59 +03:00
|
|
|
assert_eq!(counter.value(), 1);
|
2018-09-21 14:33:23 +03:00
|
|
|
assert!(server.stop());
|
|
|
|
|
thread::sleep(time::Duration::from_millis(1_000));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// To enable this test you need a trusted PKCS12 (p12) certificate bundle
|
|
|
|
|
// Hyper-tls client doesn't accept self-signed certificates. The easiest way is to use mkcert
|
|
|
|
|
// https://github.com/FiloSottile/mkcert to install CA and generate a certificate on your local machine.
|
|
|
|
|
// You need to put the file to api/tests folder
|
|
|
|
|
#[ignore]
|
|
|
|
|
#[test]
|
|
|
|
|
fn test_start_api_tls() {
|
|
|
|
|
util::init_test_logger();
|
|
|
|
|
let tls_conf = TLSConfig {
|
|
|
|
|
pkcs_bytes: include_bytes!("localhost+1.p12").to_vec(),
|
|
|
|
|
pass: "changeit".to_string(),
|
|
|
|
|
};
|
|
|
|
|
let mut server = ApiServer::new();
|
|
|
|
|
let router = build_router();
|
|
|
|
|
let server_addr = "127.0.0.1:14444";
|
|
|
|
|
let addr: SocketAddr = server_addr.parse().expect("unable to parse server address");
|
2018-10-02 10:49:36 +03:00
|
|
|
assert!(server.start(addr, router, Some(tls_conf)).is_ok());
|
2018-09-21 14:33:23 +03:00
|
|
|
let url = format!("https://{}/v1/", server_addr);
|
2018-09-26 23:38:44 +03:00
|
|
|
let index = api::client::get::<Vec<String>>(url.as_str(), None).unwrap();
|
2018-09-21 14:33:23 +03:00
|
|
|
assert_eq!(index.len(), 2);
|
|
|
|
|
assert!(!server.stop());
|
|
|
|
|
}
|
