server_test.go

package orchestra

import (
	"net/http"
	"reflect"
	"testing"
	"time"
)

func TestNewServerPlayer(t *testing.T) {
	type args struct {
		srv  *http.Server
		opts []ServerPlayerOption
	}
	tests := []struct {
		name string
		args args
		want *ServerPlayer
	}{
		{
			name: "default",
			args: args{},
			want: &ServerPlayer{server: &http.Server{}, shutdownTimeout: time.Second * 10},
		},
		{
			name: "set timeout to 5s",
			args: args{opts: []ServerPlayerOption{
				WithShutdownTimeout(time.Second * 5),
			}},
			want: &ServerPlayer{server: &http.Server{}, shutdownTimeout: time.Second * 5},
		},
		{
			name: "replace default http server",
			args: args{srv: &http.Server{Addr: ":4321"}},
			want: &ServerPlayer{server: &http.Server{Addr: ":4321"}, shutdownTimeout: time.Second * 10},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := NewServerPlayer(tt.args.srv, tt.args.opts...); !reflect.DeepEqual(got, tt.want) {
				t.Errorf("NewServerPlayer() = %v, want %v", got, tt.want)
			}
		})
	}
}