pomerium/proxy/authenticator/authenticator_test.go

package authenticator

import (
	"errors"
	"reflect"
	"testing"
	"time"
)

func TestMockAuthenticate(t *testing.T) {
	// Absurd, but I caught a typo this way.
	fixedDate := time.Date(2009, 11, 17, 20, 34, 58, 651387237, time.UTC)
	redeemResponse := &RedeemResponse{
		AccessToken:  "AccessToken",
		RefreshToken: "RefreshToken",
		Expiry:       fixedDate,
	}
	ma := &MockAuthenticate{
		RedeemError:      errors.New("RedeemError"),
		RedeemResponse:   redeemResponse,
		RefreshResponse:  "RefreshResponse",
		RefreshTime:      fixedDate,
		RefreshError:     errors.New("RefreshError"),
		ValidateResponse: true,
		ValidateError:    errors.New("ValidateError"),
		CloseError:       errors.New("CloseError"),
	}
	got, gotErr := ma.Redeem("a")
	if gotErr.Error() != "RedeemError" {
		t.Errorf("unexpected value for gotErr %s", gotErr)
	}
	if !reflect.DeepEqual(redeemResponse, got) {
		t.Errorf("unexpected value for redeemResponse %s", got)
	}
	gotToken, gotTime, gotErr := ma.Refresh("a")
	if gotErr.Error() != "RefreshError" {
		t.Errorf("unexpected value for gotErr %s", gotErr)
	}
	if !reflect.DeepEqual(gotToken, "RefreshResponse") {
		t.Errorf("unexpected value for gotToken %s", gotToken)
	}
	if !gotTime.Equal(fixedDate) {
		t.Errorf("unexpected value for gotTime %s", gotTime)
	}

	ok, gotErr := ma.Validate("a")
	if !ok {
		t.Errorf("unexpected value for ok : %t", ok)
	}
	if gotErr.Error() != "ValidateError" {
		t.Errorf("unexpected value for gotErr %s", gotErr)
	}
	gotErr = ma.Close()
	if gotErr.Error() != "CloseError" {
		t.Errorf("unexpected value for ma.CloseError %s", gotErr)
	}

}

func TestNew(t *testing.T) {
	tests := []struct {
		name        string
		serviceName string
		opts        *Options
		wantErr     bool
	}{
		{"grpc good", "grpc", &Options{Addr: "test", InternalAddr: "intranet.local", SharedSecret: "secret"}, false},
		{"grpc missing shared secret", "grpc", &Options{Addr: "test", InternalAddr: "intranet.local", SharedSecret: ""}, true},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			_, err := New(tt.serviceName, tt.opts)
			if (err != nil) != tt.wantErr {
				t.Errorf("New() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
		})
	}
}