set_up_test_env

Author: stneng

tests/common/mod.rs

@@ -0,0 +1,33 @@
+#![allow(dead_code)]
+use colink::{
+    extensions::instant_server::{InstantRegistry, InstantServer},
+    CoLink,
+};
+
+pub async fn set_up_test_env(
+    num: usize,
+) -> Result<
+    (InstantRegistry, Vec<InstantServer>, Vec<CoLink>),
+    Box<dyn std::error::Error + Send + Sync + 'static>,
+> {
+    let ir = InstantRegistry::new();
+    let mut iss = vec![];
+    let mut cls = vec![];
+    for _ in 0..num {
+        let is = InstantServer::new();
+        let cl = is.get_colink().switch_to_generated_user().await?;
+        iss.push(is);
+        cls.push(cl);
+    }
+    Ok((ir, iss, cls))
+}
+
+pub async fn set_up_test_env_single_user() -> Result<
+    (InstantRegistry, InstantServer, CoLink),
+    Box<dyn std::error::Error + Send + Sync + 'static>,
+> {
+    let ir = InstantRegistry::new();
+    let is = InstantServer::new();
+    let cl = is.get_colink().switch_to_generated_user().await?;
+    Ok((ir, is, cl))
+}

tests/test_counter_with_lock.rs

@@ -1,4 +1,5 @@
-use colink::extensions::instant_server::{InstantRegistry, InstantServer};
+mod common;
+use common::*;
 use std::{
     sync::{Arc, Mutex},
     thread,
@@ -7,9 +8,7 @@ use std::{
 #[tokio::test]
 async fn test_counter_with_lock() -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>>
 {
-    let _ir = InstantRegistry::new();
-    let is = InstantServer::new();
-    let cl = is.get_colink().switch_to_generated_user().await?;
+    let (_ir, _is, cl) = set_up_test_env_single_user().await?;
 
     cl.update_entry("example_lock_counter", &0_i32.to_le_bytes())
         .await

tests/test_policy_module.rs

@@ -1,11 +1,10 @@
-use colink::extensions::instant_server::{InstantRegistry, InstantServer};
+mod common;
 use colink::extensions::policy_module::{Action, Rule, TaskFilter};
+use common::*;
 
 #[tokio::test]
 async fn test_policy_module() -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
-    let _ir = InstantRegistry::new();
-    let is = InstantServer::new();
-    let cl = is.get_colink().switch_to_generated_user().await?;
+    let (_ir, _is, cl) = set_up_test_env_single_user().await?;
     //default policy
     let res = cl.policy_module_get_rules().await?;
     assert!(res.len() == 1);

tests/test_protocol_variable_transfer.rs

@@ -1,8 +1,7 @@
 #![allow(unused_variables)]
-use colink::{
-    extensions::instant_server::{InstantRegistry, InstantServer},
-    CoLink, Participant, ProtocolEntry,
-};
+mod common;
+use colink::{CoLink, Participant, ProtocolEntry};
+use common::*;
 
 struct Initiator;
 #[colink::async_trait]
@@ -55,19 +54,13 @@ impl ProtocolEntry for Receiver {
 
 #[tokio::test]
 async fn test_vt() -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
-    let ir = InstantRegistry::new();
-    let mut iss = vec![];
-    let mut cls = vec![];
-    for i in 0..8 {
-        let is = InstantServer::new();
-        let cl = is.get_colink().switch_to_generated_user().await?;
+    let (ir, iss, cls) = set_up_test_env(8).await?;
+    for cl in &cls {
         colink::protocol_attach!(
             cl,
             ("variable_transfer_test:initiator", Initiator),
             ("variable_transfer_test:receiver", Receiver)
         );
-        iss.push(is);
-        cls.push(cl);
     }
     let mut participants = vec![Participant {
         user_id: cls[0].get_user_id()?,

tests/test_remote_storage.rs

@@ -1,15 +1,11 @@
-use colink::{
-    extensions::instant_server::{InstantRegistry, InstantServer},
-    SubscriptionMessage,
-};
+mod common;
+use colink::SubscriptionMessage;
+use common::*;
 
 #[tokio::test]
 async fn test_remote_storage() -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
-    let _ir = InstantRegistry::new();
-    let is_a = InstantServer::new();
-    let is_b = InstantServer::new();
-    let cl_a = is_a.get_colink().switch_to_generated_user().await?;
-    let cl_b = is_b.get_colink().switch_to_generated_user().await?;
+    let (_ir, _iss, mut cls) = set_up_test_env(2).await?;
+    let (cl_a, cl_b) = (cls.pop().unwrap(), cls.pop().unwrap());
     let msg = "hello";
     let user_id_a = cl_a.get_user_id().unwrap();
     let user_id_b = cl_b.get_user_id().unwrap();

tests/test_storage_macro.rs

@@ -1,13 +1,12 @@
-use colink::{
-    extensions::instant_server::{InstantRegistry, InstantServer},
-    CoLink,
-};
+mod common;
+use colink::CoLink;
+use common::*;
 use rand::Rng;
 
 #[tokio::test]
 async fn test_storage_macro_chunk() -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>>
 {
-    let (_ir, _is, cl) = get_colink_user().await?;
+    let (_ir, _is, cl) = set_up_test_env_single_user().await?;
 
     let key_name = "storage_macro_test_chunk:$chunk";
     test_crud(&cl, key_name).await?;
@@ -18,7 +17,7 @@ async fn test_storage_macro_chunk() -> Result<(), Box<dyn std::error::Error + Se
 #[tokio::test]
 async fn test_storage_macro_redis() -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>>
 {
-    let (_ir, _is, cl) = get_colink_user().await?;
+    let (_ir, _is, cl) = set_up_test_env_single_user().await?;
 
     cl.create_entry("storage_macro_test_redis:redis_url", b"redis://localhost")
         .await?;
@@ -56,7 +55,7 @@ async fn test_crud(
 #[tokio::test]
 async fn test_storage_macro_append(
 ) -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
-    let (_ir, _is, cl) = get_colink_user().await?;
+    let (_ir, _is, cl) = set_up_test_env_single_user().await?;
 
     let key_name = "storage_macro_test_append";
     test_append(&cl, key_name, 10 as usize).await?;
@@ -89,7 +88,7 @@ async fn test_append(
 #[tokio::test]
 async fn test_storage_macro_redis_append(
 ) -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
-    let (_ir, _is, cl) = get_colink_user().await?;
+    let (_ir, _is, cl) = set_up_test_env_single_user().await?;
 
     cl.create_entry(
         "test_storage_macro_redis_append:redis_url",
@@ -105,7 +104,7 @@ async fn test_storage_macro_redis_append(
 #[tokio::test]
 async fn test_storage_macro_chunk_append(
 ) -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
-    let (_ir, _is, cl) = get_colink_user().await?;
+    let (_ir, _is, cl) = set_up_test_env_single_user().await?;
 
     let key_name = "test_storage_macro_chunk_append:$chunk";
     test_append(&cl, key_name, 5e6 as usize).await?;
@@ -119,7 +118,7 @@ async fn test_storage_macro_chunk_append(
 #[tokio::test]
 async fn test_storage_macro_redis_chunk(
 ) -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
-    let (_ir, _is, cl) = get_colink_user().await?;
+    let (_ir, _is, cl) = set_up_test_env_single_user().await?;
 
     cl.create_entry(
         "test_storage_macro_redis_chunk:redis_url",
@@ -131,13 +130,3 @@ async fn test_storage_macro_redis_chunk(
 
     Ok(())
 }
-
-async fn get_colink_user() -> Result<
-    (InstantRegistry, InstantServer, CoLink),
-    Box<dyn std::error::Error + Send + Sync + 'static>,
-> {
-    let ir = InstantRegistry::new();
-    let is = InstantServer::new();
-    let cl = is.get_colink().switch_to_generated_user().await?;
-    Ok((ir, is, cl))
-}