- Test coverage for probation server module

internal/probation/server_test.go

@@ -7,6 +7,8 @@ package probation
 
 import (
 	"github.com/nginxinc/kubernetes-nginx-ingress/test/mocks"
+	"github.com/sirupsen/logrus"
+	"net/http"
 	"testing"
 )
 
@@ -51,3 +53,21 @@ func TestHealthServer_HandleFailCheck(t *testing.T) {
 		t.Errorf("Expected 'Service Not Available', got %v", body)
 	}
 }
+
+func TestHealthServer_Start(t *testing.T) {
+	server := NewHealthServer()
+	server.Start()
+
+	response, err := http.Get("http://localhost:51031/livez")
+	if err != nil {
+		t.Error(err)
+	}
+
+	if response.StatusCode != http.StatusOK {
+		t.Errorf("Expected status code %v, got %v", http.StatusAccepted, response.StatusCode)
+	}
+
+	logrus.Infof("recevied a response from the probe server: %v", response)
+
+	server.Stop()
+}