test: add UT for blobfuse-proxy/main.go

Author: ScottZhuMS

pkg/blobfuse-proxy/main.go

@@ -29,6 +29,7 @@ import (
 
 var (
 	blobfuseProxyEndpoint = flag.String("blobfuse-proxy-endpoint", "unix://tmp/blobfuse-proxy.sock", "blobfuse-proxy endpoint")
+	grpcServerRunner      = server.RunGRPCServer
 )
 
 func main() {
@@ -55,7 +56,7 @@ func main() {
 	mountServer := server.NewMountServiceServer()
 
 	klog.V(2).Infof("Listening for connections on address: %v\n", listener.Addr())
-	if err = server.RunGRPCServer(mountServer, false, listener); err != nil {
+	if err = grpcServerRunner(mountServer, false, listener); err != nil {
 		klog.Fatalf("Error running grpc server %v. Error: %v", listener.Addr(), err)
 	}
 }

pkg/blobfuse-proxy/main_test.go

@@ -0,0 +1,26 @@
+package main
+
+import (
+	"net"
+	"os"
+	"testing"
+
+	"sigs.k8s.io/blob-csi-driver/pkg/blobfuse-proxy/pb"
+)
+
+func mockRunGRPCServer(_ pb.MountServiceServer, _ bool, _ net.Listener) error {
+	return nil
+}
+
+func TestMain(_ *testing.T) {
+	// mock the grpcServerRunner
+	originalGRPCServerRunner := grpcServerRunner
+	grpcServerRunner = mockRunGRPCServer
+	defer func() { grpcServerRunner = originalGRPCServerRunner }()
+
+	// Set the blobfuse-proxy-endpoint
+	os.Args = []string{"cmd", "-blobfuse-proxy-endpoint=unix://tmp/test.sock"}
+
+	// Run main
+	main()
+}