martin/tests/server_test.rs

extern crate log;

use actix_web::dev::Service;
use actix_web::{http, test, App};

use martin::dev::{mock_function_sources, mock_state, mock_table_sources};
use martin::function_source::FunctionSources;
use martin::server::router;
use martin::table_source::TableSources;

#[test]
fn test_get_table_sources_ok() {
  let state = test::run_on(|| mock_state(mock_table_sources(), None));
  let mut app = test::init_service(App::new().data(state).configure(router));

  let req = test::TestRequest::get().uri("/index.json").to_request();

  let response = test::block_on(app.call(req)).unwrap();
  assert!(response.status().is_success());

  let body = test::read_body(response);
  let table_sources: TableSources = serde_json::from_slice(&body).unwrap();
  assert!(table_sources.contains_key("public.table_source"));
}

#[test]
fn test_get_table_source_ok() {
  let state = test::run_on(|| mock_state(mock_table_sources(), None));
  let mut app = test::init_service(App::new().data(state).configure(router));

  let req = test::TestRequest::get()
    .uri("/public.non_existant.json")
    .to_request();

  let response = test::block_on(app.call(req)).unwrap();
  assert_eq!(response.status(), http::StatusCode::NOT_FOUND);

  let req = test::TestRequest::get()
    .uri("/public.table_source.json")
    .to_request();

  let response = test::block_on(app.call(req)).unwrap();
  assert!(response.status().is_success());
}

#[test]
fn test_get_table_source_tile_ok() {
  let state = test::run_on(|| mock_state(mock_table_sources(), None));
  let mut app = test::init_service(App::new().data(state).configure(router));

  let req = test::TestRequest::get()
    .uri("/public.table_source/0/0/0.pbf")
    .to_request();

  let future = test::run_on(|| app.call(req));
  let response = test::block_on(future).unwrap();
  assert!(response.status().is_success());
}

#[test]
fn test_get_function_sources_ok() {
  let state = test::run_on(|| mock_state(None, mock_function_sources()));
  let mut app = test::init_service(App::new().data(state).configure(router));

  let req = test::TestRequest::get().uri("/rpc/index.json").to_request();

  let response = test::block_on(app.call(req)).unwrap();
  assert!(response.status().is_success());

  let body = test::read_body(response);
  let function_sources: FunctionSources = serde_json::from_slice(&body).unwrap();
  assert!(function_sources.contains_key("public.function_source"));
}

#[test]
fn test_get_function_source_ok() {
  let state = test::run_on(|| mock_state(None, mock_function_sources()));
  let mut app = test::init_service(App::new().data(state).configure(router));

  let req = test::TestRequest::get()
    .uri("/rpc/public.non_existant.json")
    .to_request();

  let response = test::block_on(app.call(req)).unwrap();
  assert_eq!(response.status(), http::StatusCode::NOT_FOUND);

  let req = test::TestRequest::get()
    .uri("/rpc/public.function_source.json")
    .to_request();

  let response = test::block_on(app.call(req)).unwrap();
  assert!(response.status().is_success());
}

#[test]
fn test_get_function_source_tile_ok() {
  let state = test::run_on(|| mock_state(None, mock_function_sources()));
  let mut app = test::init_service(App::new().data(state).configure(router));

  let req = test::TestRequest::get()
    .uri("/rpc/public.function_source/0/0/0.pbf")
    .to_request();

  let future = test::run_on(|| app.call(req));
  let response = test::block_on(future).unwrap();
  assert!(response.status().is_success());
}