feat(bench): add tauri_3mb_transfer test (tauri-apps#1895)

jihanlingyu · Jun 23, 2021 · f729b5e · f729b5e
1 parent 47f7558
commit f729b5e
Show file tree

Hide file tree

Showing 7 changed files with 213 additions and 44 deletions.
diff --git a/tooling/bench/src/build_benchmark_jsons.rs b/tooling/bench/src/build_benchmark_jsons.rs
@@ -39,7 +39,9 @@ fn main() {
 
   // write json's
   utils::write_json(
-    tauri_data.to_str().expect("Something wrong with tauri_data"),
+    tauri_data
+      .to_str()
+      .expect("Something wrong with tauri_data"),
     &serde_json::to_value(&all_data).expect("Unable to build final json (alls)"),
   )
   .expect(format!("Unable to write {:?}", tauri_data).as_str());

diff --git a/tooling/bench/src/run_benchmark.rs b/tooling/bench/src/run_benchmark.rs
@@ -29,16 +29,13 @@ fn get_all_benchmarks() -> Vec<(String, String)> {
         utils::get_target()
       ),
     ),
-    /*
-    FIXME: require next release of tao
     (
       "tauri_3mb_transfer".into(),
       format!(
-        "tests/target/{}/release/tauri_3mb_transfer",
+        "tests/target/{}/release/bench_files_transfer",
         utils::get_target()
       ),
     ),
-    */
   ]
 }
 
@@ -250,6 +247,16 @@ fn run_exec_time(target_dir: &Path) -> Result<HashMap<String, HashMap<String, f6
 }
 
 fn main() -> Result<()> {
+  // download big files if not present
+  let json_3mb = utils::home_path().join(".tauri_3mb.json");
+
+  if !json_3mb.exists() {
+    utils::download_file(
+      "https://github.com/lemarier/tauri-test/releases/download/v2.0.0/json_3mb.json",
+      json_3mb,
+    );
+  }
+
   println!("Starting tauri benchmark");
 
   let target_dir = utils::target_dir();

diff --git a/tooling/bench/src/utils.rs b/tooling/bench/src/utils.rs
@@ -5,7 +5,13 @@
 use anyhow::Result;
 use serde::{Deserialize, Serialize};
 use serde_json::Value;
-use std::{collections::HashMap, fs, io::{BufRead, BufReader}, path::PathBuf, process::{Command, Output, Stdio}};
+use std::{
+  collections::HashMap,
+  fs,
+  io::{BufRead, BufReader},
+  path::PathBuf,
+  process::{Command, Output, Stdio},
+};
 
 #[derive(Default, Clone, Serialize, Deserialize, Debug)]
 pub struct BenchResult {
@@ -51,6 +57,14 @@ pub fn bench_root_path() -> PathBuf {
   PathBuf::from(env!("CARGO_MANIFEST_DIR"))
 }
 
+#[allow(dead_code)]
+pub fn home_path() -> PathBuf {
+  #[cfg(any(target_os = "macos", target_os = "linux"))]
+  return PathBuf::from(env!("HOME"));
+  #[cfg(any(target_os = "windows"))]
+  return PathBuf::from(env!("HOMEPATH"));
+}
+
 #[allow(dead_code)]
 pub fn tauri_root_path() -> PathBuf {
   bench_root_path()
@@ -192,3 +206,26 @@ pub fn write_json(filename: &str, value: &Value) -> Result<()> {
   serde_json::to_writer(f, value)?;
   Ok(())
 }
+
+#[allow(dead_code)]
+pub fn download_file(url: &str, filename: PathBuf) {
+  if !url.starts_with("http:") && !url.starts_with("https:") {
+    fs::copy(url, filename).unwrap();
+    return;
+  }
+
+  // Downloading with curl this saves us from adding
+  // a Rust HTTP client dependency.
+  println!("Downloading {}", url);
+  let status = Command::new("curl")
+    .arg("-L")
+    .arg("-s")
+    .arg("-o")
+    .arg(&filename)
+    .arg(&url)
+    .status()
+    .unwrap();
+
+  assert!(status.success());
+  assert!(filename.exists());
+}
diff --git a/tooling/bench/tests/Cargo.lock b/tooling/bench/tests/Cargo.lock
diff --git a/tooling/bench/tests/files_transfer/public/index.html b/tooling/bench/tests/files_transfer/public/index.html
@@ -11,34 +11,37 @@ <h1>Welcome to Tauri!</h1>
 
     <script>
       window.addEventListener("DOMContentLoaded", (event) => {
-        window.__TAURI__.invoke("tauri", {
-          __tauriModule: "Fs",
-          message: {
-            cmd: "readBinaryFile",
-            path: ".tauri_3mb.json",
-            options: {
-              // home folder
-              dir: 11
-            }
-          },
-        }).then((_data) => {
-          window.__TAURI__.invoke("tauri", {
-            __tauriModule: "Process",
+        window.__TAURI__
+          .invoke("tauri", {
+            __tauriModule: "Fs",
             message: {
-              cmd: "exit",
-              exitCode: 0,
+              cmd: "readBinaryFile",
+              path: ".tauri_3mb.json",
+              options: {
+                // home folder
+                dir: 11,
+              },
             },
+          })
+          .then((_data) => {
+            window.__TAURI__.invoke("tauri", {
+              __tauriModule: "Process",
+              message: {
+                cmd: "exit",
+                exitCode: 0,
+              },
+            });
+          })
+          .catch((_error) => {
+            // error
+            window.__TAURI__.invoke("tauri", {
+              __tauriModule: "Process",
+              message: {
+                cmd: "exit",
+                exitCode: 1,
+              },
+            });
           });
-        }).catch((_error) => {
-          // error
-          window.__TAURI__.invoke("tauri", {
-            __tauriModule: "Process",
-            message: {
-              cmd: "exit",
-              exitCode: 1,
-            },
-          });
-        });
       });
     </script>
   </body>

diff --git a/tooling/bench/tests/files_transfer/src-tauri/Cargo.toml b/tooling/bench/tests/files_transfer/src-tauri/Cargo.toml
@@ -10,7 +10,7 @@ tauri-build = { path = "../../../../../core/tauri-build", features = [ "codegen"
 [dependencies]
 serde_json = "1.0"
 serde = { version = "1.0", features = [ "derive" ] }
-tauri = { path = "../../../../../core/tauri", features = [] }
+tauri = { path = "../../../../../core/tauri", features = ["fs-read-binary-file"] }
 
 [features]
 default = [ "custom-protocol" ]

diff --git a/tooling/bench/tests/files_transfer/src-tauri/tauri.conf.json b/tooling/bench/tests/files_transfer/src-tauri/tauri.conf.json
@@ -36,7 +36,10 @@
       }
     },
     "allowlist": {
-      "all": false
+      "all": false,
+      "fs": {
+        "readBinaryFile": true
+      }
     },
     "windows": [
       {