1 files changed, 38 insertions, 0 deletions
diff --git a/internal/engine/healthcheck_test.go b/internal/engine/healthcheck_test.go
new file mode 100644
index 0000000..1d55ed0
--- /dev/null
+++ b/internal/engine/healthcheck_test.go
@@ -0,0 +1,38 @@
+package engine
+
+import (
+	"errors"
+	"testing"
+
+	"vpnem/internal/config"
+)
+
+func TestModeRequiresExitIP(t *testing.T) {
+	proxyMode := config.Mode{Name: "Full", Final: "proxy"}
+	directMode := config.Mode{Name: "Combo", Final: "direct"}
+
+	if !ModeRequiresExitIP(proxyMode) {
+		t.Fatal("expected proxy-final mode to require exit IP")
+	}
+	if ModeRequiresExitIP(directMode) {
+		t.Fatal("did not expect direct-final mode to require exit IP")
+	}
+}
+
+func TestDeepCheckRequiresRestart(t *testing.T) {
+	proxyMode := config.Mode{Name: "Full", Final: "proxy"}
+	directMode := config.Mode{Name: "Combo", Final: "direct"}
+
+	if !DeepCheckRequiresRestart(proxyMode, "", nil) {
+		t.Fatal("expected proxy-final mode without exit IP to restart")
+	}
+	if DeepCheckRequiresRestart(proxyMode, "89.124.96.166", errors.New("probe failed")) {
+		t.Fatal("did not expect proxy-final mode with exit IP to restart")
+	}
+	if !DeepCheckRequiresRestart(directMode, "", errors.New("probe failed")) {
+		t.Fatal("expected direct-final mode with failed blocked probe to restart")
+	}
+	if DeepCheckRequiresRestart(directMode, "", nil) {
+		t.Fatal("did not expect direct-final mode with successful blocked probe to restart")
+	}
+}