controller_test.go

package controller

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

	"code.fbi.h-da.de/danet/gosdn/controller/config"
	"github.com/spf13/viper"
)

const (
	configHome string = "./configs"
	configName string = "ci-testing-gosdn"
	configType string = "toml"
)

func TestInit(t *testing.T) {
	viper.SetConfigFile("./configs/ci-testing-gosdn.toml")
	viper.Set("basePNDUUID", "3e58372e-b53d-41d8-a06e-4131810c8e70")
	viper.Set("baseSouthBoundType", 1)
	viper.Set("baseSouthBoundUUID", "73b30205-7ad9-48fb-8251-0dbef649ce01")
}

func TestRun(t *testing.T) {
	TestInit(t)
	err := config.InitializeConfig()
	if err != nil {
		t.Error(err)
		return
	}

	type args struct {
		request string
	}
	tests := []struct {
		name string
		args args
		want interface{}
	}{
		{
			name: "liveliness indicator",
			args: args{request: apiEndpoint + "/livez"},
			want: http.StatusOK,
		},
		{
			name: "readyness indicator",
			args: args{request: apiEndpoint + "/readyz"},
			want: http.StatusOK,
		},
	}

	ctx, cancel := context.WithCancel(context.Background())

	viper.AddConfigPath(configHome)
	viper.SetConfigName(configName)
	viper.SetConfigType(configType)

	go func() {
		if err := Run(ctx); err != nil {
			t.Errorf("Run() error = %v", err)
		}
	}()
	time.Sleep(time.Second)
	t.Run("Controller Start HTTP API", func(t *testing.T) {
		got, err := http.Get(tests[0].args.request)
		if err != nil {
			t.Error(err)
			return
		}
		if !reflect.DeepEqual(got.StatusCode, tests[0].want) {
			t.Errorf("livez got: %v, want %v", got.StatusCode, tests[0].want)
		}
		got, err = http.Get(tests[1].args.request)
		if err != nil {
			t.Error(err)
			return
		}
		if !reflect.DeepEqual(got.StatusCode, tests[1].want) {
			t.Errorf("readyz got: %v, want %v", got.StatusCode, tests[1].want)
		}
	})

	cancel()
}