testing added

server/src/adresar/handlers/post_adresar.rs

@@ -8,6 +8,9 @@ pub async fn post_adresar(
     db_pool: &PgPool,
     request: PostAdresarRequest,
 ) -> Result<AdresarResponse, Status> {
+    if request.firma.is_empty() {
+        return Err(Status::invalid_argument("Firma is required"));
+    }
     let adresar = sqlx::query_as!(
         Adresar,
         r#"

server/src/adresar/tests/fixtures/migrations

@@ -1 +0,0 @@
-../../../../migrations

server/src/adresar/tests/post_adresar_test.rs

@@ -1,53 +1,33 @@
 // src/adresar/tests/post_adresar_test.rs
 use super::super::handlers::post_adresar;
 use common::proto::multieko2::adresar::PostAdresarRequest;
-use sqlx::{postgres::PgPoolOptions, PgPool};
+use sqlx::{postgres::PgPoolOptions, PgPool, Transaction, Postgres};
 use std::env;
 
-async fn setup_test_db() -> PgPool {
+async fn setup() -> (PgPool, Transaction<'static, Postgres>) {
-    dotenvy::dotenv().ok();  // Load .env file
+    dotenvy::from_filename(".env_test").ok();
-    let database_url = env::var("TEST_DATABASE_URL")
-        .expect("TEST_DATABASE_URL must be set in .env");
-
     let pool = PgPoolOptions::new()
         .max_connections(5)
-        .connect(&database_url)
+        .connect(&env::var("TEST_DATABASE_URL").unwrap())
         .await
-        .expect("Failed to connect to test database");
+        .unwrap();
-
+    
-    sqlx::migrate!()
+    let transaction = pool.begin().await.unwrap();
-        .run(&pool)
+    (pool, transaction)
-        .await
-        .expect("Migrations failed");
-
-    pool
 }
 
 #[tokio::test]
 async fn test_create_adresar_success() {
-    let pool = setup_test_db().await;
+    let (pool, mut transaction) = setup().await;
 
-    // Test data
     let request = PostAdresarRequest {
         firma: "Test Company".into(),
-        kz: "KZ123".into(),
+        ..Default::default()
-        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(),
     };
 
-    // Execute handler
+    let response = post_adresar(&mut transaction, request)
-    let response = post_adresar(&pool, request).await.unwrap();
+        .await
+        .unwrap();
 
     // Verify response
     assert!(response.id > 0);
@@ -113,27 +93,25 @@ async fn test_create_adresar_required_fields() {
 
 #[tokio::test]
 async fn test_create_adresar_empty_firma() {
-    let pool = setup_test_db().await;
+    let (_, mut transaction) = setup().await;
 
     let request = PostAdresarRequest {
-        firma: "".into(),  // Empty required field
+        firma: "".into(),
         ..Default::default()
     };
 
-    let result = post_adresar(&pool, request).await;
+    let result = post_adresar(&mut transaction, request).await;
 
     assert!(result.is_err());
-    assert_eq!(result.unwrap_err().code(), tonic::Code::Internal);
+    assert_eq!(result.unwrap_err().code(), tonic::Code::InvalidArgument);
 }
 
 #[tokio::test]
 async fn test_create_adresar_database_error() {
-    let pool = setup_test_db().await;
+    let (pool, _) = setup().await;
-    let broken_pool = PgPoolOptions::new()
+    // Simulate closed pool
-        .max_connections(0)  // Will fail to acquire connection
+    let broken_pool = pool.clone();
-        .connect("postgres://invalid:invalid@localhost/invalid")
+    broken_pool.close().await;
-        .await
-        .unwrap();
 
     let request = PostAdresarRequest {
         firma: "Test".into(),