From 043a0d0ff1b0fb7702bc60b5ef72f96dedf0e052 Mon Sep 17 00:00:00 2001
From: AlexanderManich <alexfredberg@gmail.com>
Date: Tue, 28 Nov 2023 11:37:22 +0100
Subject: [PATCH] more tests

---
 src/tests/api/model_logic.rs | 262 +++++++++++++++++++++++++++++++++++
 1 file changed, 262 insertions(+)

diff --git a/src/tests/api/model_logic.rs b/src/tests/api/model_logic.rs
index 424d1e8..8710955 100644
--- a/src/tests/api/model_logic.rs
+++ b/src/tests/api/model_logic.rs
@@ -122,3 +122,265 @@ async fn get_model_user_has_no_access_returns_err() {
 
     assert!(dbg!(res.code()) == Code::PermissionDenied);
 }
+
+#[tokio::test]
+async fn get_model_is_in_use_is_true() {
+    let mut mock_services = get_mock_services();
+
+    let model = model::Model {
+        id: Default::default(),
+        name: "model".to_string(),
+        components_info: Default::default(),
+        owner_id: 0,
+    };
+
+    let access = access::Model {
+        id: Default::default(),
+        role: "Editor".to_string(),
+        model_id: 1,
+        user_id: 1,
+    };
+
+    let in_use = in_use::Model {
+        model_id: Default::default(),
+        session_id: 0,
+        latest_activity: Default::default(),
+    };
+
+    let queries: Vec<query::Model> = vec![];
+
+    mock_services
+        .access_context_mock
+        .expect_get_access_by_uid_and_model_id()
+        .with(predicate::eq(0), predicate::eq(0))
+        .returning(move |_, _| Ok(Some(access.clone())));
+
+    mock_services
+        .model_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(Some(model.clone())));
+
+    mock_services
+        .in_use_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(Some(in_use.clone())));
+
+    mock_services
+        .query_context_mock
+        .expect_get_all_by_model_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(queries.clone()));
+
+    let mut request = Request::new(GetModelRequest { id: 0 });
+
+    request.metadata_mut().insert("uid", "0".parse().unwrap());
+
+    let api = get_mock_concrete_ecdar_api(mock_services);
+
+    let res = api.get_model(request).await;
+
+    assert!(res.unwrap().get_ref().in_use);
+}
+
+#[tokio::test]
+async fn get_model_is_in_use_is_false() {
+    let mut mock_services = get_mock_services();
+
+    let model = model::Model {
+        id: Default::default(),
+        name: "model".to_string(),
+        components_info: Default::default(),
+        owner_id: 0,
+    };
+
+    let access = access::Model {
+        id: Default::default(),
+        role: "Editor".to_string(),
+        model_id: 1,
+        user_id: 1,
+    };
+
+    let queries: Vec<query::Model> = vec![];
+
+    mock_services
+        .access_context_mock
+        .expect_get_access_by_uid_and_model_id()
+        .with(predicate::eq(0), predicate::eq(0))
+        .returning(move |_, _| Ok(Some(access.clone())));
+
+    mock_services
+        .model_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(Some(model.clone())));
+
+    mock_services
+        .in_use_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(None));
+
+    mock_services
+        .query_context_mock
+        .expect_get_all_by_model_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(queries.clone()));
+
+    let mut request = Request::new(GetModelRequest { id: 0 });
+
+    request.metadata_mut().insert("uid", "0".parse().unwrap());
+
+    let api = get_mock_concrete_ecdar_api(mock_services);
+
+    let res = api.get_model(request).await;
+
+    assert!(!res.unwrap().get_ref().in_use);
+}
+
+#[tokio::test]
+async fn get_model_model_has_no_queries_queries_are_empty() {
+    let mut mock_services = get_mock_services();
+
+    let model = model::Model {
+        id: Default::default(),
+        name: "model".to_string(),
+        components_info: Default::default(),
+        owner_id: 0,
+    };
+
+    let access = access::Model {
+        id: Default::default(),
+        role: "Editor".to_string(),
+        model_id: 1,
+        user_id: 1,
+    };
+
+    let in_use = in_use::Model {
+        model_id: Default::default(),
+        session_id: 0,
+        latest_activity: Default::default(),
+    };
+
+    let query1 = query::Model {
+        id: 0,
+        model_id: 1,
+        string: "query".to_owned(),
+        result: None,
+        outdated: false,
+    };
+
+    let query2 = query::Model {
+        id: 1,
+        model_id: 1,
+        string: "query".to_owned(),
+        result: Some(serde_json::to_value("result").unwrap()),
+        outdated: false,
+    };
+
+    let queries: Vec<query::Model> = vec![];
+
+    mock_services
+        .access_context_mock
+        .expect_get_access_by_uid_and_model_id()
+        .with(predicate::eq(0), predicate::eq(0))
+        .returning(move |_, _| Ok(Some(access.clone())));
+
+    mock_services
+        .model_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(Some(model.clone())));
+
+    mock_services
+        .in_use_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(Some(in_use.clone())));
+
+    mock_services
+        .query_context_mock
+        .expect_get_all_by_model_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(queries.clone()));
+
+    let mut request = Request::new(GetModelRequest { id: 0 });
+
+    request.metadata_mut().insert("uid", "0".parse().unwrap());
+
+    let api = get_mock_concrete_ecdar_api(mock_services);
+
+    let res = api.get_model(request).await;
+
+    assert!(res.unwrap().get_ref().queries.is_empty());
+}
+
+#[tokio::test]
+async fn get_model_query_has_no_result_query_is_empty() {
+    let mut mock_services = get_mock_services();
+
+    let model = model::Model {
+        id: Default::default(),
+        name: "model".to_string(),
+        components_info: Default::default(),
+        owner_id: 0,
+    };
+
+    let access = access::Model {
+        id: Default::default(),
+        role: "Editor".to_string(),
+        model_id: 1,
+        user_id: 1,
+    };
+
+    let in_use = in_use::Model {
+        model_id: Default::default(),
+        session_id: 0,
+        latest_activity: Default::default(),
+    };
+
+    let query = query::Model {
+        id: 0,
+        model_id: 1,
+        string: "query".to_owned(),
+        result: None,
+        outdated: false,
+    };
+
+    let queries: Vec<query::Model> = vec![query];
+
+    mock_services
+        .access_context_mock
+        .expect_get_access_by_uid_and_model_id()
+        .with(predicate::eq(0), predicate::eq(0))
+        .returning(move |_, _| Ok(Some(access.clone())));
+
+    mock_services
+        .model_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(Some(model.clone())));
+
+    mock_services
+        .in_use_context_mock
+        .expect_get_by_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(Some(in_use.clone())));
+
+    mock_services
+        .query_context_mock
+        .expect_get_all_by_model_id()
+        .with(predicate::eq(0))
+        .returning(move |_| Ok(queries.clone()));
+
+    let mut request = Request::new(GetModelRequest { id: 0 });
+
+    request.metadata_mut().insert("uid", "0".parse().unwrap());
+
+    let api = get_mock_concrete_ecdar_api(mock_services);
+
+    let res = api.get_model(request).await;
+
+    assert!(res.unwrap().get_ref().queries[0].result.is_empty());
+}