[baseserver] Helper for starting server in tests

Author: easyCZ

components/common-go/baseserver/server_test.go

@@ -11,32 +11,22 @@ import (
 	"github.com/stretchr/testify/require"
 	"net/http"
 	"testing"
-	"time"
 )
 
 func TestServer_StartStop(t *testing.T) {
 	// We don't use the helper NewForTests, because we want to control stopping ourselves.
 	srv, err := baseserver.New("server_test", baseserver.WithHTTPPort(8765), baseserver.WithGRPCPort(8766))
 	require.NoError(t, err)
+	baseserver.StartServerForTests(t, srv)
 
-	go func() {
-		require.NoError(t, srv.ListenAndServe())
-	}()
-
-	baseserver.WaitForServerToBeReachable(t, srv, 3*time.Second)
 	require.Equal(t, "http://localhost:8765", srv.HTTPAddress())
 	require.Equal(t, "localhost:8766", srv.GRPCAddress())
 	require.NoError(t, srv.Close())
 }
 
 func TestServer_ServesReady(t *testing.T) {
 	srv := baseserver.NewForTests(t)
-
-	go func(t *testing.T) {
-		require.NoError(t, srv.ListenAndServe())
-	}(t)
-
-	baseserver.WaitForServerToBeReachable(t, srv, 3*time.Second)
+	baseserver.StartServerForTests(t, srv)
 
 	readyURL := fmt.Sprintf("%s/ready", srv.HTTPAddress())
 	resp, err := http.Get(readyURL)
@@ -47,11 +37,7 @@ func TestServer_ServesReady(t *testing.T) {
 func TestServer_ServesMetricsEndpointWithDefaultConfig(t *testing.T) {
 	srv := baseserver.NewForTests(t)
 
-	go func(t *testing.T) {
-		require.NoError(t, srv.ListenAndServe())
-	}(t)
-
-	baseserver.WaitForServerToBeReachable(t, srv, 3*time.Second)
+	baseserver.StartServerForTests(t, srv)
 
 	readyUR := fmt.Sprintf("%s/metrics", srv.HTTPAddress())
 	resp, err := http.Get(readyUR)
@@ -65,11 +51,7 @@ func TestServer_ServesMetricsEndpointWithCustomMetricsConfig(t *testing.T) {
 		baseserver.WithMetricsRegistry(registry),
 	)
 
-	go func(t *testing.T) {
-		require.NoError(t, srv.ListenAndServe())
-	}(t)
-
-	baseserver.WaitForServerToBeReachable(t, srv, 3*time.Second)
+	baseserver.StartServerForTests(t, srv)
 
 	readyUR := fmt.Sprintf("%s/metrics", srv.HTTPAddress())
 	resp, err := http.Get(readyUR)

components/common-go/baseserver/testing.go

@@ -34,7 +34,20 @@ func NewForTests(t *testing.T, opts ...Option) *Server {
 	return srv
 }
 
-func WaitForServerToBeReachable(t *testing.T, srv *Server, timeout time.Duration) {
+// StartServerForTests starts the server for test purposes.
+// This is a helper which also ensures the server is reachable before returning.
+func StartServerForTests(t *testing.T, srv *Server) {
+	t.Helper()
+
+	go func() {
+		err := srv.ListenAndServe()
+		require.NoError(t, err)
+	}()
+
+	waitForServerToBeReachable(t, srv, 3*time.Second)
+}
+
+func waitForServerToBeReachable(t *testing.T, srv *Server, timeout time.Duration) {
 	ctx, cancel := context.WithTimeout(context.Background(), timeout)
 	defer cancel()
 

components/public-api-server/integration_test.go

@@ -14,19 +14,14 @@ import (
 	"google.golang.org/grpc/credentials/insecure"
 	"google.golang.org/grpc/status"
 	"testing"
-	"time"
 )
 
 func TestPublicAPIServer_v1_WorkspaceService(t *testing.T) {
 	ctx := context.Background()
 	srv := baseserver.NewForTests(t)
-	require.NoError(t, register(srv))
 
-	go func() {
-		require.NoError(t, srv.ListenAndServe())
-	}()
-
-	baseserver.WaitForServerToBeReachable(t, srv, 1*time.Second)
+	require.NoError(t, register(srv))
+	baseserver.StartServerForTests(t, srv)
 
 	conn, err := grpc.Dial(srv.GRPCAddress(), grpc.WithTransportCredentials(insecure.NewCredentials()))
 	require.NoError(t, err)