Integration Tests With Rocket

[iot-resister]

Hello,
Thank you for the great library. I'm posting this in case someone else finds it useful. I was trying to create a simple integration test with rocket. I was bashing my head until I learned that cfg(test) is false when doing an integration test.

See: rust-lang/rust#45599

I have an e2e flag on my config and used that instead. Anyway here is my lib.rs and integration_test.rs, respectively. (Please excuse the quality, I'm new to rust. Any feedback is appreciated)

mod config;

#[macro_use]
extern crate rocket;

use config::load_config;
use rocket::{Build, Rocket};
use serde::{Deserialize, Serialize};
use std::io;
use std::net::Ipv4Addr;

#[derive(Debug, Serialize, Deserialize)]
pub struct Resp {
    pub origin: Ipv4Addr,
}

#[get("/")]
async fn index() -> io::Result<String> {
    let config = load_config().unwrap();
    let url = match config.e2e {
        true => "https://httpbin.org/ip".to_string(),
        false => mockito::server_url(),
    };
    let res = reqwest::get(url)
        .await
        .unwrap()
        .json::<Resp>()
        .await
        .unwrap();
    Ok(res.origin.to_string())
}

pub fn app() -> Rocket<Build> {
    let config = load_config().unwrap();
    let figment = rocket::Config::figment()
        .merge(("address", &config.host))
        .merge(("port", &config.port))
        .merge(("log_level", "off"));

    rocket::custom(figment).mount("/", routes![index])
}

use rs_101::app;

use mockito::mock;
use rocket::local::asynchronous::Client;

#[tokio::test]
async fn test_with_reqwest() {
    let _m = mock("GET", "/")
        .with_body(r#"{"origin": "0.0.0.0"}"#)
        .create();

    let client = Client::tracked(app()).await.unwrap();
    let resp = client.get("/").dispatch().await;
    assert_eq!("0.0.0.0", resp.into_string().await.unwrap());
}