package varsig_test

import (
	"encoding/base64"
	"fmt"
	"testing"

	"github.com/libp2p/go-libp2p/core/crypto/pb"
	"github.com/stretchr/testify/assert"

	"github.com/ucan-wg/go-ucan/tokens/internal/varsig"
)

func TestDecode(t *testing.T) {
	t.Parallel()

	notAHeader := base64.RawStdEncoding.EncodeToString([]byte("not a header"))
	keyType, err := varsig.Decode([]byte(notAHeader))
	assert.Equal(t, pb.KeyType(-1), keyType)
	assert.ErrorIs(t, err, varsig.ErrUnknownHeader)
}

func ExampleDecode() {
	hdr, err := base64.RawStdEncoding.DecodeString("NIUkEoACcQ")
	if err != nil {
		fmt.Println(err.Error())

		return
	}

	keyType, _ := varsig.Decode(hdr)
	fmt.Println(keyType.String())
	// Output:
	// RSA
}

func TestEncode(t *testing.T) {
	t.Parallel()

	header, err := varsig.Encode(pb.KeyType(99))
	assert.Nil(t, header)
	assert.ErrorIs(t, err, varsig.ErrUnknownKeyType)
}

func ExampleEncode() {
	header, _ := varsig.Encode(pb.KeyType_RSA)
	fmt.Println(base64.RawStdEncoding.EncodeToString(header))

	// Output:
	// NIUkEoACcQ
}