chore(coprocessor): add mask_database_url unit test

Author: goshawk-3

coprocessor/fhevm-engine/fhevm-engine-common/src/utils.rs

@@ -122,14 +122,14 @@ pub struct DatabaseURL(String);
 impl From<&str> for DatabaseURL {
     fn from(s: &str) -> Self {
         let url: Self = Self(s.to_owned());
-        url.parse().expect("DatabaseURL is valid");
+        url.parse().expect("DatabaseURL should be valid");
         url
     }
 }
 impl From<String> for DatabaseURL {
     fn from(s: String) -> Self {
         let url: Self = Self(s);
-        url.parse().expect("DatabaseURL is valid");
+        url.parse().expect("DatabaseURL should be valid");
         url
     }
 }
@@ -160,7 +160,7 @@ impl DatabaseURL {
             }
         }
         let url: Self = Self(url);
-        let _ = url.parse().expect("DatabaseURL is valid");
+        let _ = url.parse().expect("DatabaseURL should be valid");
         url
     }
 

coprocessor/fhevm-engine/fhevm-engine-common/tests/utils.rs

@@ -0,0 +1,33 @@
+use fhevm_engine_common::utils::DatabaseURL;
+
+#[tokio::test]
+async fn mask_database_url() {
+    let db_url: DatabaseURL = "postgres://postgres:mypassword@localhost:5432/coprocessor".into();
+
+    let debug_fmt = format!("{:?}", db_url);
+    assert!(!debug_fmt.contains("mypassword"));
+
+    let display_fmt = format!("{}", db_url);
+    assert!(!display_fmt.contains("mypassword"));
+    println!("DatabaseURL: {}", db_url);
+
+    let db_url: DatabaseURL = DatabaseURL::new_with_app_name(
+        "postgres://user:secret@dbhost:5432/mydb?sslmode=require",
+        "tfhe-worker",
+    );
+
+    assert_eq!(
+        db_url.as_str(),
+        "postgres://user:secret@dbhost:5432/mydb?sslmode=require&application_name=tfhe-worker"
+    );
+
+    let db_url: DatabaseURL =
+        DatabaseURL::new_with_app_name("postgres://user:secret@dbhost:5432/mydb", "tfhe-worker");
+
+    assert_eq!(
+        db_url.as_str(),
+        "postgres://user:secret@dbhost:5432/mydb?application_name=tfhe-worker"
+    );
+
+    println!("DatabaseURL: {}", db_url);
+}

coprocessor/fhevm-engine/stress-test-generator/data/json/minitest_002_erc20.json

@@ -9,8 +9,8 @@
         "user_address": "0xa0534e99d86F081E8D3868A8C4732C8f65dfdB07",
         "scenario": [
             [
-                0.1,
-                5
+                20.0,
+                120
             ]
         ]
     }

coprocessor/fhevm-engine/stress-test-generator/src/utils.rs

@@ -346,7 +346,7 @@ impl EnvConfig {
         };
         let acl_contract_address: String = match env::var("ACL_CONTRACT_ADDRESS") {
             Ok(val) => val,
-            Err(_) => "0x617E46088ffe8d0f27A123574119b9C3F3b81073".to_string(),
+            Err(_) => "0x05fD9B5EFE0a996095f42Ed7e77c390810CF660c".to_string(),
         };
         let chain_id: i64 = match env::var("CHAIN_ID") {
             Ok(val) => val.parse::<i64>().unwrap(),