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);
+}