api_token_provider_test.go

package fireblocksdk_test

import (
	sdk "fireblocksdk"
	"testing"
	"time"

	"github.com/golang-jwt/jwt"
	"github.com/stretchr/testify/require"
	"github.com/stretchr/testify/suite"
)

const (
	// https://phpseclib.com/docs/rsa-keys
	privateKey = "-----BEGIN PRIVATE KEY-----\nMIIBVAIBADANBgkqhkiG9w0BAQEFAASCAT4wggE6AgEAAkEAqPfgaTEWEP3S9w0t\ngsicURfo+nLW09/0KfOPinhYZ4ouzU+3xC4pSlEp8Ut9FgL0AgqNslNaK34Kq+NZ\njO9DAQIDAQABAkAgkuLEHLaqkWhLgNKagSajeobLS3rPT0Agm0f7k55FXVt743hw\nNgkp98bMNrzy9AQ1mJGbQZGrpr4c8ZAx3aRNAiEAoxK/MgGeeLui385KJ7ZOYktj\nhLBNAB69fKwTZFsUNh0CIQEJQRpFCcydunv2bENcN/oBTRw39E8GNv2pIcNxZkcb\nNQIgbYSzn3Py6AasNj6nEtCfB+i1p3F35TK/87DlPSrmAgkCIQDJLhFoj1gbwRbH\n/bDRPrtlRUDDx44wHoEhSDRdy77eiQIgE6z/k6I+ChN1LLttwX0galITxmAYrOBh\nBVl433tgTTQ=\n-----END PRIVATE KEY-----" //nolint:lll
)

func TestAuthTokenSuite(t *testing.T) {
	suite.Run(t, new(AuthTokenSuite))
}

type AuthTokenSuite struct {
	suite.Suite
	auth         sdk.IAuthProvider
	apiKey       string
	apiSecretKey []byte
	timeProvider sdk.ITimeProvider
}

type testTimeProvider struct{}

func (tp *testTimeProvider) Now() time.Time {
	return time.Unix(1000, 0)
}

func (suite *AuthTokenSuite) SetupTest() {
	suite.apiKey = "apiKey"
	suite.apiSecretKey = []byte(privateKey)
	suite.timeProvider = &testTimeProvider{}
	suite.auth, _ = sdk.NewAuthProvider(
		suite.apiKey,
		suite.apiSecretKey,
		sdk.WithTimeProvider(suite.timeProvider),
	)
}

func (suite *AuthTokenSuite) TestWithCorrectPrivateKey() {
	token, err := suite.auth.SignJwt("", []byte(""))
	require.NoError(suite.T(), err)
	require.Equal(suite.T(), "eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJib2R5SGFzaCI6ImUzYjBjNDQyOThmYzFjMTQ5YWZiZjRjODk5NmZiOTI0MjdhZTQxZTQ2NDliOTM0Y2E0OTU5OTFiNzg1MmI4NTUiLCJleHAiOjEwMTAsImlhdCI6MTAwMCwibm9uY2UiOjEwMDAsIm5vdyI6MTAwMCwic3ViIjoiYXBpS2V5IiwidXJpIjoiIn0.ZlA5Sjlskt2eG1Zg6B3uRSCM3tq82pAihZiXYzwgVc4CCmB-PKELxSALwtIFcZKAdKsfqqkeiiGUDbkOwxhG8g", token) //nolint:lll
}

func (suite *AuthTokenSuite) TestMustFailWithNotRSAPrivateKey() {
	_, err := sdk.NewAuthProvider(
		suite.apiKey,
		[]byte("fakeKey"),
		sdk.WithTimeProvider(suite.timeProvider),
	)
	require.Error(suite.T(), err)
	require.ErrorAs(suite.T(), err, &jwt.ErrNotRSAPrivateKey)
}

func (suite *AuthTokenSuite) TestTimeProvider() {
	unix := suite.timeProvider.Now().Unix()
	require.Equal(suite.T(), int64(1000), unix)
}

func (suite *AuthTokenSuite) TestGetApiKey() {
	require.Equal(suite.T(), suite.auth.GetAPIKey(), suite.apiKey)
}