[PM-16908][TEST] Add some test functions

crates/bitwarden-wasm-internal/rewrite_wasm_types.js

@@ -131,16 +131,7 @@ const modified = ts.visitNode(ast, visitor);
 if (!ts.isSourceFile(modified)) {
   throw new Error("Modified AST is not a source file");
 }
-const result = ts.createPrinter({ newLine: ts.NewLineKind.LineFeed }).printFile(modified);
-prettier
-  .format(result, {
-    parser: "typescript",
-    tabWidth: 2,
-    useTabs: false,
-  })
-  .then((formatted) => {
-    fs.writeFileSync(tsFilename, formatted, "utf-8");
-  });
+const tsResult = ts.createPrinter({ newLine: ts.NewLineKind.LineFeed }).printFile(modified);
 
 // Save the sub-clients table to the types file
 const SEPARATOR = "/* The following code is generated by the rewrite_wasm_types.js script */";
@@ -151,10 +142,19 @@ ${SEPARATOR}
 export const SUB_CLIENT_METHODS = ${JSON.stringify(subClients, null, 2)};
 `,
 );
-fs.writeFileSync(
-  tsFilename,
-  `${tsCode.split(SEPARATOR)[0]}
+
+const tsResultWithSubClients = `${tsResult.split(SEPARATOR)[0]}
 ${SEPARATOR}
 export declare const SUB_CLIENT_METHODS: Record<string, any>;
-`,
-);
+`;
+
+// Format and save the modified types file
+prettier
+  .format(tsResultWithSubClients, {
+    parser: "typescript",
+    tabWidth: 2,
+    useTabs: false,
+  })
+  .then((formatted) => {
+    fs.writeFileSync(tsFilename, formatted, "utf-8");
+  });

crates/bitwarden-wasm-internal/src/client.rs

@@ -40,6 +40,10 @@
         res.text().await.map_err(|e| e.to_string())
     }
 
+    pub fn test(&self) -> crate::test::TestClient {
+        crate::test::TestClient::new(self.0.clone())
+    }
+
     pub fn crypto(&self) -> CryptoClient {
         CryptoClient::new(self.0.clone())
     }

crates/bitwarden-wasm-internal/src/lib.rs

@@ -6,6 +6,7 @@ mod custom_types;
 mod init;
 mod pure_crypto;
 mod ssh;
+mod test;
 mod vault;
 
 pub use bitwarden_ipc::wasm::*;

crates/bitwarden-wasm-internal/src/test.rs

@@ -0,0 +1,68 @@
+use std::rc::Rc;
+
+use bitwarden_core::Client;
+use js_sys::Promise;
+use wasm_bindgen::prelude::*;
+
+#[wasm_bindgen]
+pub struct TestClient(Rc<Client>);
+
+impl TestClient {
+    pub fn new(client: Rc<Client>) -> Self {
+        Self(client)
+    }
+}
+
+#[wasm_bindgen]
+pub struct TestSub1Client(Rc<Client>);
+
+#[wasm_bindgen]
+pub struct TestSub2Client(Rc<Client>);
+
+#[allow(clippy::unused_async)]
+#[wasm_bindgen]
+impl TestClient {
+    pub fn sub1(&self) -> TestSub1Client {
+        TestSub1Client(self.0.clone())
+    }
+
+    #[allow(clippy::unused_async)]
+    pub async fn async_echo(&self, msg: String) -> String {
+        format!("ECHOED: '{}'", msg.to_uppercase())
+    }
+}
+
+#[allow(clippy::unused_async)]
+#[wasm_bindgen]
+impl TestSub1Client {
+    pub fn sub2(&self) -> TestSub2Client {
+        TestSub2Client(self.0.clone())
+    }
+
+    pub async fn async_echo_cb(
+        &self,
+        #[wasm_bindgen(unchecked_param_type = "(text: string) => Promise<string>")]
+        f: &js_sys::Function,
+    ) -> Result<String, JsValue> {
+        let this = JsValue::null();
+        let val = f.call1(&this, &"hello".into())?;
+
+        let pr: Promise = val.dyn_into()?;
+        let fut = wasm_bindgen_futures::JsFuture::from(pr);
+        let val = fut
+            .await?
+            .as_string()
+            .ok_or_else(|| js_sys::Error::new("result is not a string"))?;
+
+        Ok(format!("Result async: '{}'", val.to_uppercase()))
+    }
+}
+
+#[allow(clippy::unused_async)]
+#[wasm_bindgen]
+impl TestSub2Client {
+    #[allow(clippy::unused_async)]
+    pub async fn get_flags(&self) -> String {
+        format!("{:?}", self.0.internal.get_flags())
+    }
+}