ekuiper-new/internal/pkg/jwt/jwt_rsa_test.go

package jwt

import (
	"fmt"
	"testing"
)

var (
	expiredToken   = "eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJhdWQiOiJlS3VpcGVyIiwiZXhwIjoxNjM2MDExMzQxLCJpc3MiOiJzYW1wbGVfa2V5LnB1YiJ9.qm5Pq9VxDC10qbOM081U5NwScTOxYV_F5vyqbU9rXB2ebz4kDio_R2tgEgGyJ41lwD7gFl1quBjp_EgokPZNOoGRg5R1Ygf7iF8XJSDxYkspSCsBtZAuMCo3MCz3slQyvnr24qv3idUDhlwO6FPHGLaLHEyvrETSl1ZcECq2wvW01Tc2Jmg0-Kpp6TmEbH5aD-L0or5Bfy0ytBQ64nd2hKVaoADZZOXSt1iH2-1R35fEc_lBw7zs4QpCC2R--muoqYsYkESR08o6wIKAxRJvqeWab3C9k_g0zaPhwa7ZQ9wRzah-tc6PdotZkAyH7BCx-f7llO7UT47k0GnrhBe21g"
	badFormatToken = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJmb28iOiJiYXIiLCJleHAiOjE1MDAwLCJpc3MiOiJ0ZXN0In0"
)

func genToken(signKeyName, issuer, aud string) string {
	tkStr, _ := CreateToken(signKeyName, issuer, aud)
	return tkStr
}

func TestParseToken(t *testing.T) {
	type args struct {
		th string
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{
			name: "pass: have issuer public key",
			args: args{
				th: genToken("sample_key", "sample_key.pub", "eKuiper"),
			},
			wantErr: false,
		},
		{
			name: "fail: token expired",
			args: args{
				th: expiredToken,
			},
			wantErr: true,
		},
		{
			name: "fail: token sign error",
			args: args{
				th: genToken("sample_key", "sample_key.pub", "eKuiper") + "badSign",
			},
			wantErr: true,
		},
		{
			name: "fail: do not have issuer's public key",
			args: args{
				th: genToken("sample_key", "notexist.pub", "eKuiper"),
			},
			wantErr: true,
		},
		{
			name: "bad token format",
			args: args{
				th: badFormatToken,
			},
			wantErr: true,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			_, err := ParseToken(tt.args.th)
			if (err != nil) != tt.wantErr {
				t.Errorf("ParseToken() error = %v, wantErr %v", err, tt.wantErr)
				return
			}

			if err != nil {
				fmt.Printf("=====================\n")
				fmt.Printf("Validate Error %s", err)
			}
		})
	}
}