mirror of
https://github.com/gitpod-io/gitpod.git
synced 2025-12-08 17:36:30 +00:00
79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
|
|
// Licensed under the GNU Affero General Public License (AGPL).
|
|
// See License-AGPL.txt in the project root for license information.
|
|
|
|
package baseserver_test
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gitpod-io/gitpod/common-go/baseserver"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"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)
|
|
|
|
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)
|
|
|
|
readyURL := fmt.Sprintf("%s/ready", srv.HTTPAddress())
|
|
resp, err := http.Get(readyURL)
|
|
require.NoError(t, err)
|
|
require.Equal(t, http.StatusOK, resp.StatusCode)
|
|
}
|
|
|
|
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)
|
|
|
|
readyUR := fmt.Sprintf("%s/metrics", srv.HTTPAddress())
|
|
resp, err := http.Get(readyUR)
|
|
require.NoError(t, err)
|
|
require.Equal(t, http.StatusOK, resp.StatusCode)
|
|
}
|
|
|
|
func TestServer_ServesMetricsEndpointWithCustomMetricsConfig(t *testing.T) {
|
|
registry := prometheus.NewRegistry()
|
|
srv := baseserver.NewForTests(t,
|
|
baseserver.WithMetricsRegistry(registry),
|
|
)
|
|
|
|
go func(t *testing.T) {
|
|
require.NoError(t, srv.ListenAndServe())
|
|
}(t)
|
|
|
|
baseserver.WaitForServerToBeReachable(t, srv, 3*time.Second)
|
|
|
|
readyUR := fmt.Sprintf("%s/metrics", srv.HTTPAddress())
|
|
resp, err := http.Get(readyUR)
|
|
require.NoError(t, err)
|
|
require.Equal(t, http.StatusOK, resp.StatusCode)
|
|
}
|