fix: failing tests after switch to grpc.NewClient

Author: isometry

pkg/provider/grpc/grpc_test.go

@@ -10,6 +10,7 @@ import (
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/health"
 	"google.golang.org/grpc/health/grpc_health_v1"
+	"google.golang.org/grpc/resolver"
 
 	ph "github.com/isometry/platform-health/pkg/platform_health"
 	provider_grpc "github.com/isometry/platform-health/pkg/provider/grpc"
@@ -20,6 +21,9 @@ func init() {
 }
 
 func TestGetHealth(t *testing.T) {
+	// workaround for grpc resolver with Zscaler
+	resolver.SetDefaultScheme("passthrough")
+
 	listener, err := net.Listen("tcp", "localhost:0")
 	if err != nil {
 		t.Fatalf("Failed to set up test server: %v", err)

pkg/provider/satellite/satellite.go

@@ -91,6 +91,7 @@ func (i *Satellite) GetHealth(ctx context.Context) *ph.HealthCheckResponse {
 	if err != nil {
 		return component.Unhealthy(err.Error())
 	}
+	defer conn.Close()
 
 	// Propagate already visited serverIds from context to enable loop detection
 	request := &ph.HealthCheckRequest{

pkg/provider/satellite/satellite_test.go

@@ -8,6 +8,7 @@ import (
 	"time"
 
 	"github.com/stretchr/testify/assert"
+	"google.golang.org/grpc/resolver"
 
 	ph "github.com/isometry/platform-health/pkg/platform_health"
 	"github.com/isometry/platform-health/pkg/provider"
@@ -27,6 +28,9 @@ func (c *testConfig) GetInstances() []provider.Instance {
 }
 
 func TestSatelliteGetHealth(t *testing.T) {
+	// workaround for grpc resolver with Zscaler
+	resolver.SetDefaultScheme("passthrough")
+
 	// Start listener for the main server
 	listener, err := net.Listen("tcp", "localhost:0")
 	if err != nil {