feat(warehouse): Add Disabled + CreatedAt Param to Warehouse

Author: flo-ride

entity/src/models/warehouse.rs

@@ -24,6 +24,12 @@ pub struct Model {
     pub id: Uuid,
     /// The name of the warehouse
     pub name: String,
+    /// Indicates whether the warehouse is disabled.
+    #[sea_orm(filter_single)]
+    pub disabled: bool,
+    /// Timestamp for when the warehouse was created.
+    #[sea_orm(filter_override = "chrono::DateTime<chrono::Utc>", filter_plus_order)]
+    pub created_at: DateTimeWithTimeZone,
 }
 
 /// Enum representing the relationships of the `warehouse` entity in SeaORM.

entity/src/request/warehouse.rs

@@ -61,6 +61,8 @@ impl TryFrom<NewWarehouseRequest> for ActiveModel {
                 }
                 Set(name)
             },
+            disabled: Set(false),
+            created_at: Set(chrono::offset::Local::now().into()),
         })
     }
 }
@@ -70,6 +72,9 @@ impl TryFrom<NewWarehouseRequest> for ActiveModel {
 pub struct EditWarehouseRequest {
     /// New name for the Warehouse.
     pub name: Option<String>,
+
+    /// Optional new disabled status for the warehouse.
+    pub disabled: Option<bool>,
 }
 
 /// Converts `EditWarehouseRequest` into `ActiveModel` with validation.
@@ -95,6 +100,11 @@ impl TryFrom<EditWarehouseRequest> for ActiveModel {
                 }
                 None => NotSet,
             },
+            disabled: match value.disabled {
+                Some(disabled) => Set(disabled),
+                None => NotSet,
+            },
+            ..Default::default()
         })
     }
 }

entity/src/response/warehouse.rs

@@ -16,6 +16,12 @@ pub struct WarehouseResponse {
 
     /// Name of the warehouse.
     name: String,
+
+    /// The timestamp indicating when the warehouse was created.
+    pub created_at: chrono::DateTime<chrono::Utc>,
+
+    /// Indicates whether the warehouse is currently disabled.
+    pub disabled: bool,
 }
 
 impl From<crate::models::warehouse::Model> for WarehouseResponse {
@@ -24,6 +30,8 @@ impl From<crate::models::warehouse::Model> for WarehouseResponse {
         Self {
             id: value.id,
             name: value.name,
+            created_at: value.created_at.into(),
+            disabled: value.disabled,
         }
     }
 }
@@ -38,6 +46,7 @@ impl From<crate::models::warehouse::Model> for WarehouseResponse {
             {
                 "id": "1a731f58-18f1-4c95-8de5-611bde07f4f1",
                 "name": "Warehouse Central", 
+                "disabled": false,
                 "creation_time": "2024-10-09T17:55:30.795279Z"
             }
         ]

migration/src/m20220101_000007_create_warehouse_table.rs

@@ -13,6 +13,11 @@ impl MigrationTrait for Migration {
                     .if_not_exists()
                     .col(uuid(Warehouse::Id).primary_key())
                     .col(string(Warehouse::Name))
+                    .col(boolean(Warehouse::Disabled).default(false))
+                    .col(
+                        timestamp_with_time_zone(Warehouse::CreatedAt)
+                            .default(Expr::current_timestamp()),
+                    )
                     .to_owned(),
             )
             .await
@@ -30,4 +35,6 @@ pub enum Warehouse {
     Table,
     Id,
     Name,
+    Disabled,
+    CreatedAt,
 }

tests/warehouse_tests.rs

@@ -338,3 +338,91 @@ async fn warehouse_edit_too_long_name() {
         ]
     }));
 }
+
+#[tokio::test]
+async fn warehouse_create_disabled() {
+    let realm = Realm::default();
+    let (mut server, _ids, _nodes) = create_basic_session(realm.clone()).await;
+    let cookies = create_realm_session(&mut server, realm.users).await;
+
+    let response = server
+        .post("/warehouse")
+        .json(&json!({
+            "name": "Warehouse 1",
+        }))
+        .add_cookie(cookies[0].clone())
+        .await;
+    response.assert_status(StatusCode::CREATED);
+    let id = response.text();
+
+    let response = server
+        .get(&format!("/warehouse/{id}"))
+        .add_cookie(cookies[0].clone())
+        .await;
+    response.assert_status_ok();
+    response.assert_json_contains(&json!({
+        "disabled": false,
+    }));
+
+    let response = server
+        .get("/warehouse")
+        .add_cookie(cookies[0].clone())
+        .await;
+    response.assert_status_ok();
+    response.assert_json_contains(&json!({
+        "warehouses": [
+            {
+                "disabled": false,
+            },
+        ]
+    }));
+}
+
+#[tokio::test]
+async fn warehouse_edit_disabled() {
+    let realm = Realm::default();
+    let (mut server, _ids, _nodes) = create_basic_session(realm.clone()).await;
+    let cookies = create_realm_session(&mut server, realm.users).await;
+
+    // New warehouse with Ok name
+    let response = server
+        .post("/warehouse")
+        .json(&json!({
+            "name": "Warehouse 1",
+        }))
+        .add_cookie(cookies[0].clone())
+        .await;
+    response.assert_status(StatusCode::CREATED);
+    let id = response.text();
+
+    let response = server
+        .put(&format!("/warehouse/{id}"))
+        .json(&json!({
+            "disabled": true,
+        }))
+        .add_cookie(cookies[0].clone())
+        .await;
+    response.assert_status_ok();
+
+    let response = server
+        .get(&format!("/warehouse/{id}"))
+        .add_cookie(cookies[0].clone())
+        .await;
+    response.assert_status_ok();
+    response.assert_json_contains(&json!({
+        "disabled": true,
+    }));
+
+    let response = server
+        .get("/warehouse")
+        .add_cookie(cookies[0].clone())
+        .await;
+    response.assert_status_ok();
+    response.assert_json_contains(&json!({
+        "warehouses": [
+            {
+                "disabled": true,
+            },
+        ]
+    }));
+}