From ca604628aa8558b77ce24b11d5255e6f828d9222 Mon Sep 17 00:00:00 2001
From: Mskxn <118117161+Mskxn@users.noreply.github.com>
Date: Wed, 19 Apr 2023 07:56:40 +0800
Subject: [PATCH] stubserver: Stop server when StartClient failed (#6190)

---
 internal/stubserver/stubserver.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/internal/stubserver/stubserver.go b/internal/stubserver/stubserver.go
index 012021dc0..94ef56482 100644
--- a/internal/stubserver/stubserver.go
+++ b/internal/stubserver/stubserver.go
@@ -86,7 +86,11 @@ func (ss *StubServer) Start(sopts []grpc.ServerOption, dopts ...grpc.DialOption)
 	if err := ss.StartServer(sopts...); err != nil {
 		return err
 	}
-	return ss.StartClient(dopts...)
+	if err := ss.StartClient(dopts...); err != nil {
+		ss.Stop()
+		return err
+	}
+	return nil
 }
 
 // StartServer only starts the server. It does not create a client to it.