mirror of
https://github.com/int128/kubelogin.git
synced 2026-05-20 14:52:49 +00:00
70 lines
2.0 KiB
Go
70 lines
2.0 KiB
Go
package authentication
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-test/deep"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/int128/kubelogin/pkg/adaptors/env/mock_env"
|
|
"github.com/int128/kubelogin/pkg/adaptors/logger/mock_logger"
|
|
"github.com/int128/kubelogin/pkg/adaptors/oidcclient"
|
|
"github.com/int128/kubelogin/pkg/adaptors/oidcclient/mock_oidcclient"
|
|
)
|
|
|
|
var nonNil = gomock.Not(gomock.Nil())
|
|
|
|
func TestAuthCodeKeyboard_Do(t *testing.T) {
|
|
dummyTokenClaims := map[string]string{"sub": "YOUR_SUBJECT"}
|
|
futureTime := time.Now().Add(time.Hour) //TODO: inject time service
|
|
timeout := 5 * time.Second
|
|
|
|
t.Run("Success", func(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
ctx, cancel := context.WithTimeout(context.TODO(), timeout)
|
|
defer cancel()
|
|
mockOIDCClient := mock_oidcclient.NewMockInterface(ctrl)
|
|
mockOIDCClient.EXPECT().
|
|
GetAuthCodeURL(nonNil).
|
|
Return("https://issuer.example.com/auth")
|
|
mockOIDCClient.EXPECT().
|
|
ExchangeAuthCode(nonNil, nonNil).
|
|
Do(func(_ context.Context, in oidcclient.ExchangeAuthCodeInput) {
|
|
if in.Code != "YOUR_AUTH_CODE" {
|
|
t.Errorf("Code wants YOUR_AUTH_CODE but was %s", in.Code)
|
|
}
|
|
}).
|
|
Return(&oidcclient.TokenSet{
|
|
IDToken: "YOUR_ID_TOKEN",
|
|
RefreshToken: "YOUR_REFRESH_TOKEN",
|
|
IDTokenSubject: "YOUR_SUBJECT",
|
|
IDTokenExpiry: futureTime,
|
|
IDTokenClaims: dummyTokenClaims,
|
|
}, nil)
|
|
mockEnv := mock_env.NewMockInterface(ctrl)
|
|
mockEnv.EXPECT().
|
|
ReadString(authCodeKeyboardPrompt).
|
|
Return("YOUR_AUTH_CODE", nil)
|
|
u := AuthCodeKeyboard{
|
|
Env: mockEnv,
|
|
Logger: mock_logger.New(t),
|
|
}
|
|
out, err := u.Do(ctx, nil, mockOIDCClient)
|
|
if err != nil {
|
|
t.Errorf("Do returned error: %+v", err)
|
|
}
|
|
want := &Output{
|
|
IDToken: "YOUR_ID_TOKEN",
|
|
RefreshToken: "YOUR_REFRESH_TOKEN",
|
|
IDTokenSubject: "YOUR_SUBJECT",
|
|
IDTokenExpiry: futureTime,
|
|
IDTokenClaims: dummyTokenClaims,
|
|
}
|
|
if diff := deep.Equal(want, out); diff != nil {
|
|
t.Error(diff)
|
|
}
|
|
})
|
|
}
|