Merge pull request #67 from ECDAR-AAU-SW-P5/delete-model-tests

Add delete_model tests

Cargo.toml

@@ -25,6 +25,7 @@ uuid = { version = "1.5.0", features = ["v4"] }
 regex = "1.10.2"
 mockall = "0.11.4"
 bcrypt = "0.15.0"
+serde_json = "1.0.108"
 
 [build-dependencies]
 tonic-build = "0.10.2"

src/api/ecdar_api.rs

@@ -571,3 +571,7 @@ mod tests;
 #[cfg(test)]
 #[path = "../tests/api/query_logic.rs"]
 mod query_logic;
+
+#[cfg(test)]
+#[path = "../tests/api/model_logic.rs"]
+mod model_logic;

src/tests/api/model_logic.rs

@@ -0,0 +1,108 @@
+use std::str::FromStr;
+
+use mockall::predicate;
+use tonic::{metadata, Code, Request};
+
+use crate::{
+    api::server::server::{
+        ecdar_api_server::EcdarApi, Component, ComponentsInfo, DeleteModelRequest,
+    },
+    entities::model,
+    tests::api::helpers::{get_mock_concrete_ecdar_api, get_mock_services},
+};
+
+#[tokio::test]
+async fn delete_not_owner_returns_err() {
+    let mut mock_services = get_mock_services();
+
+    mock_services
+        .model_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(1))
+        .returning(move |_| {
+            Ok(Some(model::Model {
+                id: 1,
+                name: Default::default(),
+                components_info: Default::default(),
+                owner_id: 2,
+            }))
+        });
+
+    let mut request = Request::new(DeleteModelRequest { id: 1 });
+
+    request
+        .metadata_mut()
+        .insert("uid", metadata::MetadataValue::from_str("1").unwrap());
+
+    let api = get_mock_concrete_ecdar_api(mock_services);
+
+    let res = api.delete_model(request).await.unwrap_err();
+
+    assert_eq!(res.code(), Code::PermissionDenied);
+}
+
+#[tokio::test]
+async fn delete_invalid_model_returns_err() {
+    let mut mock_services = get_mock_services();
+
+    mock_services
+        .model_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(2))
+        .returning(move |_| Ok(None));
+
+    let mut request = Request::new(DeleteModelRequest { id: 2 });
+
+    request
+        .metadata_mut()
+        .insert("uid", metadata::MetadataValue::from_str("1").unwrap());
+
+    let api = get_mock_concrete_ecdar_api(mock_services);
+
+    let res = api.delete_model(request).await.unwrap_err();
+
+    assert_eq!(res.code(), Code::NotFound);
+}
+
+#[tokio::test]
+async fn delete_model_returns_ok() {
+    let mut mock_services = get_mock_services();
+
+    mock_services
+        .model_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(1))
+        .returning(move |_| {
+            Ok(Some(model::Model {
+                id: 1,
+                name: Default::default(),
+                components_info: Default::default(),
+                owner_id: 1,
+            }))
+        });
+
+    mock_services
+        .model_context_mock
+        .expect_delete()
+        .with(predicate::eq(1))
+        .returning(move |_| {
+            Ok(model::Model {
+                id: 1,
+                name: Default::default(),
+                components_info: Default::default(),
+                owner_id: 1,
+            })
+        });
+
+    let mut request = Request::new(DeleteModelRequest { id: 1 });
+
+    request
+        .metadata_mut()
+        .insert("uid", metadata::MetadataValue::from_str("1").unwrap());
+
+    let api = get_mock_concrete_ecdar_api(mock_services);
+
+    let res = api.delete_model(request).await;
+
+    assert!(res.is_ok());
+}