From a7db980039813247e0bd2bd4983383d92d0d81ae Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tobias=20Fr=C3=B6lich?= <tobias@blueshoe.de>
Date: Thu, 13 Jul 2023 16:30:16 +0200
Subject: [PATCH] test: improve coverage

---
 .github/workflows/cd.yaml  |  8 +++++++-
 src-tauri/src/heartbeat.rs | 13 +++++++++++++
 src-tauri/src/util.rs      |  6 ++++++
 3 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/cd.yaml b/.github/workflows/cd.yaml
index 2723212..e026f58 100644
--- a/.github/workflows/cd.yaml
+++ b/.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:
diff --git a/src-tauri/src/heartbeat.rs b/src-tauri/src/heartbeat.rs
index 8ca83d1..14a589f 100644
--- a/src-tauri/src/heartbeat.rs
+++ b/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())
+    }
 }
diff --git a/src-tauri/src/util.rs b/src-tauri/src/util.rs
index 1d98c7c..a953319 100644
--- a/src-tauri/src/util.rs
+++ b/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());
+    }
 }