netbird/client/internal/certproof/store_other_test.go

//go:build ((!darwin && !windows) || ios) && !js

package certproof

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/netbirdio/netbird/client/internal/pkcs11"
)

func TestStoreWithToken(t *testing.T) {
	dir := storeDir(t)
	t.Setenv(StoreDirEnv, dir)

	files, ok := storeWithToken(Config{}).(*fileStore)
	require.True(t, ok, "no token configured reads the PEM directory alone")
	assert.Equal(t, dir, files.dir, "the directory comes from NB_CERT_STORE_DIR")

	assert.IsType(t, &fileStore{}, storeWithToken(Config{PKCS11: PKCS11Config{URI: "not-a-pkcs11-uri"}}), "an invalid URI must not hide the PEM directory")
	assert.IsType(t, &fileStore{}, storeWithToken(Config{PKCS11: PKCS11Config{PIN: "1234"}}), "a PIN naming no token is refused and leaves the PEM directory")

	configured := map[string]PKCS11Config{
		"env pin with token uri": {URI: "pkcs11:token=netbird", PIN: "1234"},
		"uri alone":              {URI: "pkcs11:token=netbird?pin-value=1234"},
	}
	if !pkcs11.Supported() {
		for name, cfg := range configured {
			assert.IsType(t, &fileStore{}, storeWithToken(Config{PKCS11: cfg}),
				"%s: a build without PKCS#11 support reads the PEM directory alone", name)
		}
		return
	}
	for name, cfg := range configured {
		store, ok := storeWithToken(Config{PKCS11: cfg}).(Stores)
		require.True(t, ok, "%s joins the token to the PEM directory", name)
		require.Len(t, store, 2, name)
		token, ok := store[1].(*pkcs11Store)
		require.True(t, ok, name)
		assert.Equal(t, dir, token.certDir, "%s: the token pairs certificates from the same directory", name)
	}
}