1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
extern crate diesel;
extern crate fatcat;
extern crate fatcat_api_spec;
extern crate iron;
extern crate iron_test;
extern crate uuid;
use self::iron_test::response;
use fatcat_api_spec::client::Client;
use iron::headers::ContentType;
use iron::mime::Mime;
use iron::{status, Headers, Iron, Listening};
use std::{thread, time};
// A current problem with this method is that if the test fails (eg, panics, assert fails), the
// server never gets closed, and the server thread hangs forever.
// One workaround might be to invert the function, take a closure, capture the panic/failure, and
// cleanup.
pub fn setup_client() -> (Client, Listening) {
let server = fatcat::test_server().unwrap();
let router = fatcat_api_spec::router(server);
// this is an unfortunate hack for testings seeming to fail in CI
thread::sleep(time::Duration::from_millis(100));
let iron_server = Iron::new(router)
.http("localhost:9144")
.expect("Failed to start HTTP server");
let client = Client::try_new_http("http://localhost:9144").unwrap();
(client, iron_server)
}
pub fn setup_http() -> (
Headers,
fatcat_api_spec::router::Router,
diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::PgConnection>>,
) {
let server = fatcat::test_server().unwrap();
let conn = server.db_pool.get().expect("db_pool error");
let router = fatcat_api_spec::router(server);
let mut headers = Headers::new();
let mime: Mime = "application/json".parse().unwrap();
headers.set(ContentType(mime));
(headers, router, conn)
}
pub fn check_http_response(
resp: iron::IronResult<iron::response::Response>,
want_status: status::Status,
in_body: Option<&str>,
) {
let resp = resp.unwrap();
let status = resp.status;
let body = response::extract_body_to_string(resp);
println!("{}", body);
assert_eq!(status, Some(want_status));
if let Some(thing) = in_body {
assert!(body.contains(thing));
}
}
|
