test: improve coverage

.github/workflows/cd.yaml

@@ -1,5 +1,11 @@
 name: "test-and-build-on-pr"
-on: [pull_request]
+on:
+  push:
+    branches:
+    - main
+  pull_request:
+    branches:
+    - main
 
 jobs:
   test:

src-tauri/src/heartbeat.rs

@@ -28,4 +28,17 @@ mod tests {
         assert_eq!("Heartbeat status: 200 OK", result.unwrap())
     }
 
+    #[tokio::test]
+    async fn test_establish_heartbeat_connection_failure() {
+        let mut server = mockito::Server::new();
+        let url = server.url();
+
+        let mock = server.mock("POST", "/clusters/123/heartbeat")
+        .with_status(500)
+        .create();
+
+        let result = establish_heartbeat_connection("123", "token", &url).await;
+        mock.assert();
+        assert_eq!("Heartbeat status: 500 Internal Server Error", result.unwrap())
+    }
 }

src-tauri/src/util.rs

@@ -72,4 +72,10 @@ mod tests {
         let cleanup_result = super::cleanup(beiboot_name);
         assert!(cleanup_result.is_ok());
     }
+
+    #[tokio::test]
+    async fn test_check_docker_engine() {
+        let result = super::check_docker_engine().await;
+        assert!(result.is_ok());
+    }
 }