Vardhaman SuranaandGitHub
2a40b4538c
- added manager to wait for all go routines to end before exit - code refactor - renamed Manafer to Interface and GoRoutineManager to GroupManager - replaced some go routine calls with manager Add func - added unit tests for manager
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package httpserver
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/hashicorp/go-retryablehttp"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/fission/fission/pkg/utils/loggerfactory"
|
|
"github.com/fission/fission/pkg/utils/manager"
|
|
)
|
|
|
|
func TestStartServer(t *testing.T) {
|
|
mgr := manager.New()
|
|
defer mgr.Wait()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
logger := loggerfactory.GetLogger()
|
|
m := mux.NewRouter()
|
|
m.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
_, err := w.Write([]byte("test handler"))
|
|
if err != nil {
|
|
logger.Error("failed to write response", zap.Error(err))
|
|
}
|
|
}))
|
|
|
|
mgr.Add(ctx, func(ctx context.Context) {
|
|
StartServer(ctx, logger, mgr, "test", "8999", m)
|
|
})
|
|
|
|
tests := []struct {
|
|
Name string
|
|
URL string
|
|
StatusCode int
|
|
Body string
|
|
}{
|
|
{
|
|
Name: "test handler",
|
|
URL: "http://localhost:8999",
|
|
StatusCode: http.StatusOK,
|
|
Body: "test handler",
|
|
},
|
|
{
|
|
Name: "not found",
|
|
URL: "http://localhost:8999/notfound",
|
|
StatusCode: http.StatusNotFound,
|
|
Body: "404 page not found\n",
|
|
},
|
|
}
|
|
client := retryablehttp.NewClient()
|
|
for _, test := range tests {
|
|
t.Run(test.Name, func(t *testing.T) {
|
|
resp, err := client.Get(test.URL)
|
|
require.NoError(t, err, "failed to make get request %s", test.URL)
|
|
defer resp.Body.Close()
|
|
require.Equal(t, test.StatusCode, resp.StatusCode)
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
require.Equal(t, string(body), test.Body)
|
|
})
|
|
}
|
|
}
|