mirror of
https://github.com/LukeMathWalker/zero-to-production.git
synced 2024-11-15 21:21:00 +00:00
Split tests across files.
This commit is contained in:
LukeMathWalker
parent
e8ff7914c9
commit
0344317f85
3 changed files with 170 additions and 168 deletions
|
|
@ -1,84 +1,9 @@
|
||||||
use sqlx::{Connection, Executor, PgConnection, PgPool};
|
use crate::helpers;
|
||||||
use std::net::TcpListener;
|
|
||||||
use uuid::Uuid;
|
|
||||||
use zero2prod::configuration::{get_configuration, DatabaseSettings};
|
|
||||||
use zero2prod::email_client::EmailClient;
|
|
||||||
use zero2prod::startup::run;
|
|
||||||
use zero2prod::telemetry::{get_subscriber, init_subscriber};
|
|
||||||
|
|
||||||
// Ensure that the `tracing` stack is only initialised once using `lazy_static`
|
|
||||||
lazy_static::lazy_static! {
|
|
||||||
static ref TRACING: () = {
|
|
||||||
let filter = if std::env::var("TEST_LOG").is_ok() { "debug" } else { "" };
|
|
||||||
let subscriber = get_subscriber("test".into(), filter.into());
|
|
||||||
init_subscriber(subscriber);
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct TestApp {
|
|
||||||
pub address: String,
|
|
||||||
pub db_pool: PgPool,
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn spawn_app() -> TestApp {
|
|
||||||
// The first time `initialize` is invoked the code in `TRACING` is executed.
|
|
||||||
// All other invocations will instead skip execution.
|
|
||||||
lazy_static::initialize(&TRACING);
|
|
||||||
|
|
||||||
let listener = TcpListener::bind("127.0.0.1:0").expect("Failed to bind random port");
|
|
||||||
// We retrieve the port assigned to us by the OS
|
|
||||||
let port = listener.local_addr().unwrap().port();
|
|
||||||
let address = format!("http://127.0.0.1:{}", port);
|
|
||||||
|
|
||||||
let mut configuration = get_configuration().expect("Failed to read configuration.");
|
|
||||||
configuration.database.database_name = Uuid::new_v4().to_string();
|
|
||||||
let connection_pool = configure_database(&configuration.database).await;
|
|
||||||
|
|
||||||
let sender_email = configuration
|
|
||||||
.email_client
|
|
||||||
.sender()
|
|
||||||
.expect("Invalid sender email address.");
|
|
||||||
let email_client = EmailClient::new(
|
|
||||||
configuration.email_client.base_url,
|
|
||||||
sender_email,
|
|
||||||
configuration.email_client.authorization_token,
|
|
||||||
);
|
|
||||||
|
|
||||||
let server =
|
|
||||||
run(listener, connection_pool.clone(), email_client).expect("Failed to bind address");
|
|
||||||
let _ = tokio::spawn(server);
|
|
||||||
TestApp {
|
|
||||||
address,
|
|
||||||
db_pool: connection_pool,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn configure_database(config: &DatabaseSettings) -> PgPool {
|
|
||||||
// Create database
|
|
||||||
let mut connection = PgConnection::connect_with(&config.without_db())
|
|
||||||
.await
|
|
||||||
.expect("Failed to connect to Postgres");
|
|
||||||
connection
|
|
||||||
.execute(&*format!(r#"CREATE DATABASE "{}";"#, config.database_name))
|
|
||||||
.await
|
|
||||||
.expect("Failed to create database.");
|
|
||||||
|
|
||||||
// Migrate database
|
|
||||||
let connection_pool = PgPool::connect_with(config.with_db())
|
|
||||||
.await
|
|
||||||
.expect("Failed to connect to Postgres.");
|
|
||||||
sqlx::migrate!("./migrations")
|
|
||||||
.run(&connection_pool)
|
|
||||||
.await
|
|
||||||
.expect("Failed to migrate the database");
|
|
||||||
|
|
||||||
connection_pool
|
|
||||||
}
|
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn health_check_works() {
|
async fn health_check_works() {
|
||||||
// Arrange
|
// Arrange
|
||||||
let app = spawn_app().await;
|
let app = helpers::spawn_app().await;
|
||||||
let client = reqwest::Client::new();
|
let client = reqwest::Client::new();
|
||||||
|
|
||||||
// Act
|
// Act
|
||||||
|
|
@ -93,94 +18,3 @@ async fn health_check_works() {
|
||||||
assert!(response.status().is_success());
|
assert!(response.status().is_success());
|
||||||
assert_eq!(Some(0), response.content_length());
|
assert_eq!(Some(0), response.content_length());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[actix_rt::test]
|
|
||||||
async fn subscribe_returns_a_200_for_valid_form_data() {
|
|
||||||
// Arrange
|
|
||||||
let app = spawn_app().await;
|
|
||||||
let client = reqwest::Client::new();
|
|
||||||
let body = "name=le%20guin&email=ursula_le_guin%40gmail.com";
|
|
||||||
|
|
||||||
// Act
|
|
||||||
let response = client
|
|
||||||
.post(&format!("{}/subscriptions", &app.address))
|
|
||||||
.header("Content-Type", "application/x-www-form-urlencoded")
|
|
||||||
.body(body)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.expect("Failed to execute request.");
|
|
||||||
|
|
||||||
// Assert
|
|
||||||
assert_eq!(200, response.status().as_u16());
|
|
||||||
|
|
||||||
let saved = sqlx::query!("SELECT email, name FROM subscriptions",)
|
|
||||||
.fetch_one(&app.db_pool)
|
|
||||||
.await
|
|
||||||
.expect("Failed to fetch saved subscription.");
|
|
||||||
|
|
||||||
assert_eq!(saved.email, "ursula_le_guin@gmail.com");
|
|
||||||
assert_eq!(saved.name, "le guin");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[actix_rt::test]
|
|
||||||
async fn subscribe_returns_a_400_when_data_is_missing() {
|
|
||||||
// Arrange
|
|
||||||
let app = spawn_app().await;
|
|
||||||
let client = reqwest::Client::new();
|
|
||||||
let test_cases = vec![
|
|
||||||
("name=le%20guin", "missing the email"),
|
|
||||||
("email=ursula_le_guin%40gmail.com", "missing the name"),
|
|
||||||
("", "missing both name and email"),
|
|
||||||
];
|
|
||||||
|
|
||||||
for (invalid_body, error_message) in test_cases {
|
|
||||||
// Act
|
|
||||||
let response = client
|
|
||||||
.post(&format!("{}/subscriptions", &app.address))
|
|
||||||
.header("Content-Type", "application/x-www-form-urlencoded")
|
|
||||||
.body(invalid_body)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.expect("Failed to execute request.");
|
|
||||||
|
|
||||||
// Assert
|
|
||||||
assert_eq!(
|
|
||||||
400,
|
|
||||||
response.status().as_u16(),
|
|
||||||
// Additional customised error message on test failure
|
|
||||||
"The API did not fail with 400 Bad Request when the payload was {}.",
|
|
||||||
error_message
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[actix_rt::test]
|
|
||||||
async fn subscribe_returns_a_400_when_fields_are_present_but_invalid() {
|
|
||||||
// Arrange
|
|
||||||
let app = spawn_app().await;
|
|
||||||
let client = reqwest::Client::new();
|
|
||||||
let test_cases = vec![
|
|
||||||
("name=&email=ursula_le_guin%40gmail.com", "empty name"),
|
|
||||||
("name=Ursula&email=", "empty email"),
|
|
||||||
("name=Ursula&email=definitely-not-an-email", "invalid email"),
|
|
||||||
];
|
|
||||||
|
|
||||||
for (body, description) in test_cases {
|
|
||||||
// Act
|
|
||||||
let response = client
|
|
||||||
.post(&format!("{}/subscriptions", &app.address))
|
|
||||||
.header("Content-Type", "application/x-www-form-urlencoded")
|
|
||||||
.body(body)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.expect("Failed to execute request.");
|
|
||||||
|
|
||||||
// Assert
|
|
||||||
assert_eq!(
|
|
||||||
400,
|
|
||||||
response.status().as_u16(),
|
|
||||||
"The API did not return a 400 Bad Request when the payload was {}.",
|
|
||||||
description
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,76 @@
|
||||||
|
use sqlx::{Connection, Executor, PgConnection, PgPool};
|
||||||
|
use std::net::TcpListener;
|
||||||
|
use uuid::Uuid;
|
||||||
|
use zero2prod::configuration::{get_configuration, DatabaseSettings};
|
||||||
|
use zero2prod::email_client::EmailClient;
|
||||||
|
use zero2prod::startup::run;
|
||||||
|
use zero2prod::telemetry::{get_subscriber, init_subscriber};
|
||||||
|
|
||||||
|
// Ensure that the `tracing` stack is only initialised once using `lazy_static`
|
||||||
|
lazy_static::lazy_static! {
|
||||||
|
static ref TRACING: () = {
|
||||||
|
let filter = if std::env::var("TEST_LOG").is_ok() { "debug" } else { "" };
|
||||||
|
let subscriber = get_subscriber("test".into(), filter.into());
|
||||||
|
init_subscriber(subscriber);
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct TestApp {
|
||||||
|
pub address: String,
|
||||||
|
pub db_pool: PgPool,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn spawn_app() -> TestApp {
|
||||||
|
// The first time `initialize` is invoked the code in `TRACING` is executed.
|
||||||
|
// All other invocations will instead skip execution.
|
||||||
|
lazy_static::initialize(&TRACING);
|
||||||
|
|
||||||
|
let listener = TcpListener::bind("127.0.0.1:0").expect("Failed to bind random port");
|
||||||
|
// We retrieve the port assigned to us by the OS
|
||||||
|
let port = listener.local_addr().unwrap().port();
|
||||||
|
let address = format!("http://127.0.0.1:{}", port);
|
||||||
|
|
||||||
|
let mut configuration = get_configuration().expect("Failed to read configuration.");
|
||||||
|
configuration.database.database_name = Uuid::new_v4().to_string();
|
||||||
|
let connection_pool = configure_database(&configuration.database).await;
|
||||||
|
|
||||||
|
let sender_email = configuration
|
||||||
|
.email_client
|
||||||
|
.sender()
|
||||||
|
.expect("Invalid sender email address.");
|
||||||
|
let email_client = EmailClient::new(
|
||||||
|
configuration.email_client.base_url,
|
||||||
|
sender_email,
|
||||||
|
configuration.email_client.authorization_token,
|
||||||
|
);
|
||||||
|
|
||||||
|
let server =
|
||||||
|
run(listener, connection_pool.clone(), email_client).expect("Failed to bind address");
|
||||||
|
let _ = tokio::spawn(server);
|
||||||
|
TestApp {
|
||||||
|
address,
|
||||||
|
db_pool: connection_pool,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn configure_database(config: &DatabaseSettings) -> PgPool {
|
||||||
|
// Create database
|
||||||
|
let mut connection = PgConnection::connect_with(&config.without_db())
|
||||||
|
.await
|
||||||
|
.expect("Failed to connect to Postgres");
|
||||||
|
connection
|
||||||
|
.execute(&*format!(r#"CREATE DATABASE "{}";"#, config.database_name))
|
||||||
|
.await
|
||||||
|
.expect("Failed to create database.");
|
||||||
|
|
||||||
|
// Migrate database
|
||||||
|
let connection_pool = PgPool::connect_with(config.with_db())
|
||||||
|
.await
|
||||||
|
.expect("Failed to connect to Postgres.");
|
||||||
|
sqlx::migrate!("./migrations")
|
||||||
|
.run(&connection_pool)
|
||||||
|
.await
|
||||||
|
.expect("Failed to migrate the database");
|
||||||
|
|
||||||
|
connection_pool
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,92 @@
|
||||||
|
use crate::helpers;
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn subscribe_returns_a_200_for_valid_form_data() {
|
||||||
|
// Arrange
|
||||||
|
let app = helpers::spawn_app().await;
|
||||||
|
let client = reqwest::Client::new();
|
||||||
|
let body = "name=le%20guin&email=ursula_le_guin%40gmail.com";
|
||||||
|
|
||||||
|
// Act
|
||||||
|
let response = client
|
||||||
|
.post(&format!("{}/subscriptions", &app.address))
|
||||||
|
.header("Content-Type", "application/x-www-form-urlencoded")
|
||||||
|
.body(body)
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.expect("Failed to execute request.");
|
||||||
|
|
||||||
|
// Assert
|
||||||
|
assert_eq!(200, response.status().as_u16());
|
||||||
|
|
||||||
|
let saved = sqlx::query!("SELECT email, name FROM subscriptions",)
|
||||||
|
.fetch_one(&app.db_pool)
|
||||||
|
.await
|
||||||
|
.expect("Failed to fetch saved subscription.");
|
||||||
|
|
||||||
|
assert_eq!(saved.email, "ursula_le_guin@gmail.com");
|
||||||
|
assert_eq!(saved.name, "le guin");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn subscribe_returns_a_400_when_data_is_missing() {
|
||||||
|
// Arrange
|
||||||
|
let app = helpers::spawn_app().await;
|
||||||
|
let client = reqwest::Client::new();
|
||||||
|
let test_cases = vec![
|
||||||
|
("name=le%20guin", "missing the email"),
|
||||||
|
("email=ursula_le_guin%40gmail.com", "missing the name"),
|
||||||
|
("", "missing both name and email"),
|
||||||
|
];
|
||||||
|
|
||||||
|
for (invalid_body, error_message) in test_cases {
|
||||||
|
// Act
|
||||||
|
let response = client
|
||||||
|
.post(&format!("{}/subscriptions", &app.address))
|
||||||
|
.header("Content-Type", "application/x-www-form-urlencoded")
|
||||||
|
.body(invalid_body)
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.expect("Failed to execute request.");
|
||||||
|
|
||||||
|
// Assert
|
||||||
|
assert_eq!(
|
||||||
|
400,
|
||||||
|
response.status().as_u16(),
|
||||||
|
// Additional customised error message on test failure
|
||||||
|
"The API did not fail with 400 Bad Request when the payload was {}.",
|
||||||
|
error_message
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn subscribe_returns_a_400_when_fields_are_present_but_invalid() {
|
||||||
|
// Arrange
|
||||||
|
let app = helpers::spawn_app().await;
|
||||||
|
let client = reqwest::Client::new();
|
||||||
|
let test_cases = vec![
|
||||||
|
("name=&email=ursula_le_guin%40gmail.com", "empty name"),
|
||||||
|
("name=Ursula&email=", "empty email"),
|
||||||
|
("name=Ursula&email=definitely-not-an-email", "invalid email"),
|
||||||
|
];
|
||||||
|
|
||||||
|
for (body, description) in test_cases {
|
||||||
|
// Act
|
||||||
|
let response = client
|
||||||
|
.post(&format!("{}/subscriptions", &app.address))
|
||||||
|
.header("Content-Type", "application/x-www-form-urlencoded")
|
||||||
|
.body(body)
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.expect("Failed to execute request.");
|
||||||
|
|
||||||
|
// Assert
|
||||||
|
assert_eq!(
|
||||||
|
400,
|
||||||
|
response.status().as_u16(),
|
||||||
|
"The API did not return a 400 Bad Request when the payload was {}.",
|
||||||
|
description
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in a new issue