working perfectly well
This commit is contained in:
@@ -1,28 +1,12 @@
|
||||
// src/adresar/tests/post_adresar_test.rs
|
||||
// tests/adresar/post_adresar_test.rs
|
||||
use server::adresar::handlers::post_adresar;
|
||||
use common::proto::multieko2::adresar::PostAdresarRequest;
|
||||
use sqlx::{postgres::PgPoolOptions, PgPool};
|
||||
use std::env;
|
||||
use crate::common::setup_test_db;
|
||||
use sqlx::PgPool;
|
||||
|
||||
// Helper functions
|
||||
mod test_helpers {
|
||||
use super::*;
|
||||
|
||||
pub async fn setup_test_db() -> PgPool {
|
||||
dotenvy::from_filename(".env_test").ok();
|
||||
let pool = PgPoolOptions::new()
|
||||
.max_connections(5)
|
||||
.connect(&env::var("TEST_DATABASE_URL").unwrap())
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
sqlx::migrate!()
|
||||
.run(&pool)
|
||||
.await
|
||||
.expect("Migrations failed");
|
||||
|
||||
pool
|
||||
}
|
||||
|
||||
pub fn valid_request() -> PostAdresarRequest {
|
||||
PostAdresarRequest {
|
||||
@@ -61,7 +45,7 @@ mod test_helpers {
|
||||
assert_eq!(db_record.firma, response.firma);
|
||||
assert_eq!(db_record.telefon, Some(response.telefon.clone()));
|
||||
// Add assertions for other fields...
|
||||
|
||||
|
||||
// Verify default values
|
||||
assert!(!db_record.deleted);
|
||||
assert!(db_record.created_at.is_some());
|
||||
@@ -70,7 +54,7 @@ mod test_helpers {
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_success() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
let pool = setup_test_db().await;
|
||||
let request = test_helpers::valid_request();
|
||||
|
||||
let response = post_adresar(&pool, request).await.unwrap();
|
||||
@@ -78,20 +62,20 @@ async fn test_create_adresar_success() {
|
||||
// Basic response validation
|
||||
assert!(response.id > 0, "Should return positive ID");
|
||||
assert_eq!(response.firma, "Test Company", "Firma should match");
|
||||
|
||||
|
||||
// Verify all fields against request
|
||||
let request = test_helpers::valid_request();
|
||||
assert_eq!(response.kz, request.kz);
|
||||
assert_eq!(response.drc, request.drc);
|
||||
// Continue for all fields...
|
||||
|
||||
|
||||
// Comprehensive database check
|
||||
test_helpers::assert_response_matches(&pool, &response).await;
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_whitespace_trimming() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
let pool = setup_test_db().await;
|
||||
|
||||
let mut request = test_helpers::valid_request();
|
||||
request.firma = " Test Company ".into();
|
||||
@@ -107,7 +91,7 @@ async fn test_create_adresar_whitespace_trimming() {
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_empty_optional_fields() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
let pool = setup_test_db().await;
|
||||
|
||||
let mut request = test_helpers::valid_request();
|
||||
request.telefon = " ".into(); // Only whitespace
|
||||
@@ -126,7 +110,7 @@ async fn test_create_adresar_empty_optional_fields() {
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_invalid_firma() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
let pool = setup_test_db().await;
|
||||
|
||||
let mut request = test_helpers::valid_request();
|
||||
request.firma = " ".into(); // Only whitespace
|
||||
@@ -138,14 +122,14 @@ async fn test_create_adresar_invalid_firma() {
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_minimal_valid_request() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
let pool = setup_test_db().await;
|
||||
let request = test_helpers::minimal_request();
|
||||
|
||||
let response = post_adresar(&pool, request).await.unwrap();
|
||||
|
||||
assert!(response.id > 0);
|
||||
assert_eq!(response.firma, "Required Only");
|
||||
|
||||
|
||||
// Verify optional fields are empty
|
||||
assert!(response.kz.is_empty());
|
||||
assert!(response.drc.is_empty());
|
||||
@@ -154,7 +138,7 @@ async fn test_create_adresar_minimal_valid_request() {
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_empty_firma() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
let pool = setup_test_db().await;
|
||||
let mut request = test_helpers::minimal_request();
|
||||
request.firma = "".into();
|
||||
|
||||
@@ -166,7 +150,7 @@ async fn test_create_adresar_empty_firma() {
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_database_error() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
let pool = setup_test_db().await;
|
||||
pool.close().await;
|
||||
|
||||
let result = post_adresar(&pool, test_helpers::minimal_request()).await;
|
||||
@@ -177,47 +161,47 @@ async fn test_create_adresar_database_error() {
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_field_length_limits() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
|
||||
let pool = setup_test_db().await;
|
||||
|
||||
let mut request = test_helpers::valid_request();
|
||||
request.firma = "a".repeat(255);
|
||||
request.telefon = "1".repeat(20); // Example length based on DB schema
|
||||
|
||||
|
||||
let response = post_adresar(&pool, request).await.unwrap();
|
||||
|
||||
|
||||
assert_eq!(response.firma.len(), 255);
|
||||
assert_eq!(response.telefon.len(), 20);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_special_characters() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
|
||||
let pool = setup_test_db().await;
|
||||
|
||||
let mut request = test_helpers::valid_request();
|
||||
request.telefon = "+420 123-456.789".into();
|
||||
request.ulica = "Náměstí 28. října".into();
|
||||
|
||||
|
||||
let response = post_adresar(&pool, request.clone()).await.unwrap();
|
||||
|
||||
|
||||
assert_eq!(response.telefon, request.telefon);
|
||||
assert_eq!(response.ulica, request.ulica);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_create_adresar_optional_fields_null_vs_empty() {
|
||||
let pool = test_helpers::setup_test_db().await;
|
||||
|
||||
let pool = setup_test_db().await;
|
||||
|
||||
// Test explicit null (using default values)
|
||||
let mut request = test_helpers::valid_request();
|
||||
request.telefon = String::new();
|
||||
|
||||
|
||||
let response = post_adresar(&pool, request).await.unwrap();
|
||||
|
||||
|
||||
// Verify empty string is stored as NULL in database
|
||||
let record = sqlx::query!("SELECT telefon FROM adresar WHERE id = $1", response.id)
|
||||
.fetch_one(&pool)
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
|
||||
assert!(record.telefon.is_none());
|
||||
}
|
||||
|
||||
@@ -1,77 +1,30 @@
|
||||
// tests/common/mod.rs
|
||||
use dotenvy;
|
||||
use sqlx::{postgres::PgPoolOptions, PgPool};
|
||||
use std::env;
|
||||
use tokio::sync::OnceCell;
|
||||
use std::path::Path;
|
||||
|
||||
static DB_POOL: OnceCell<PgPool> = OnceCell::const_new();
|
||||
pub async fn setup_test_db() -> PgPool {
|
||||
// Get path to server directory
|
||||
let manifest_dir = env::var("CARGO_MANIFEST_DIR").expect("CARGO_MANIFEST_DIR must be set");
|
||||
let env_path = Path::new(&manifest_dir).join(".env_test");
|
||||
|
||||
pub async fn get_db_pool() -> PgPool {
|
||||
DB_POOL.get_or_init(|| async {
|
||||
dotenvy::from_filename(".env_test").ok();
|
||||
let pool = PgPoolOptions::new()
|
||||
.max_connections(5)
|
||||
.connect(&env::var("TEST_DATABASE_URL").unwrap())
|
||||
.await
|
||||
.unwrap();
|
||||
// Load environment variables
|
||||
dotenvy::from_path(env_path).ok();
|
||||
|
||||
// Run migrations
|
||||
sqlx::migrate!()
|
||||
.run(&pool)
|
||||
.await
|
||||
.expect("Migrations failed");
|
||||
|
||||
pool
|
||||
}).await.clone()
|
||||
}
|
||||
|
||||
pub async fn cleanup_db(pool: &PgPool) {
|
||||
sqlx::query!("TRUNCATE TABLE adresar, uctovnictvo RESTART IDENTITY CASCADE")
|
||||
.execute(pool)
|
||||
// Create connection pool
|
||||
let database_url = env::var("TEST_DATABASE_URL").expect("TEST_DATABASE_URL must be set");
|
||||
let pool = PgPoolOptions::new()
|
||||
.max_connections(5)
|
||||
.connect(&database_url)
|
||||
.await
|
||||
.expect("Failed to cleanup test database");
|
||||
}
|
||||
|
||||
// Test helpers
|
||||
pub mod adresar_helpers {
|
||||
use common::proto::multieko2::adresar::{PostAdresarRequest, AdresarResponse};
|
||||
use sqlx::PgPool;
|
||||
|
||||
pub fn valid_request() -> PostAdresarRequest {
|
||||
PostAdresarRequest {
|
||||
firma: "Test Company".into(),
|
||||
kz: "KZ123".into(),
|
||||
drc: "DRC456".into(),
|
||||
ulica: "Test Street".into(),
|
||||
psc: "12345".into(),
|
||||
mesto: "Test City".into(),
|
||||
stat: "Test Country".into(),
|
||||
banka: "Test Bank".into(),
|
||||
ucet: "123456789".into(),
|
||||
skladm: "Warehouse M".into(),
|
||||
ico: "12345678".into(),
|
||||
kontakt: "John Doe".into(),
|
||||
telefon: "+421123456789".into(),
|
||||
skladu: "Warehouse U".into(),
|
||||
fax: "+421123456700".into(),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn minimal_request() -> PostAdresarRequest {
|
||||
PostAdresarRequest {
|
||||
firma: "Required Only".into(),
|
||||
..Default::default()
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn assert_response_matches(pool: &PgPool, response: &AdresarResponse) {
|
||||
let db_record = sqlx::query!("SELECT * FROM adresar WHERE id = $1", response.id)
|
||||
.fetch_one(pool)
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
assert_eq!(db_record.firma, response.firma);
|
||||
assert_eq!(db_record.telefon, Some(response.telefon.clone()));
|
||||
assert!(!db_record.deleted);
|
||||
assert!(db_record.created_at.is_some());
|
||||
}
|
||||
.expect("Failed to create pool");
|
||||
|
||||
// Run migrations
|
||||
sqlx::migrate!()
|
||||
.run(&pool)
|
||||
.await
|
||||
.expect("Migrations failed");
|
||||
|
||||
pool
|
||||
}
|
||||
|
||||
@@ -1,2 +1,3 @@
|
||||
// tests/mod.rs
|
||||
pub mod adresar;
|
||||
pub mod common;
|
||||
|
||||
Reference in New Issue
Block a user