controller_test.go

package nucleus

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

func TestRun(t *testing.T) {
	type args struct {
		request string
	}
	tests := []struct {
		name    string
		args    args
		want    interface{}
		wantErr bool
	}{
		{
			name:    "liveliness indicator",
			args:    args{request: apiEndpoint + "/livez"},
			want:    http.StatusOK,
			wantErr: false,
		},
		{
			name:    "readyness indicator",
			args:    args{request: apiEndpoint + "/readyz"},
			want:    http.StatusOK,
			wantErr: false,
		},
		{
			name:    "init",
			args:    args{request: apiEndpoint + "/api?q=init"},
			want:    http.StatusOK,
			wantErr: false,
		},
	}
	for _, tt := range tests {
		ctx, cancel := context.WithCancel(context.Background())
		go func() {
			if err := Run(ctx); (err != nil) != tt.wantErr {
				t.Errorf("Run() error = %v, wantErr %v", err, tt.wantErr)
			}
		}()
		t.Run(tt.name, func(t *testing.T) {
			got, err := http.Get(tt.args.request)
			if err != nil {
				t.Error(err)
			}
			if !reflect.DeepEqual(got.StatusCode, tt.want) {
				t.Errorf("Run() got: %v, want %v", got.StatusCode, tt.want)
			}
		})
		cancel()
	}
}