feat: add test

trustbloc · Oct 1, 2024 · 1c71b21 · 1c71b21
1 parent c7939dd
commit 1c71b21
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/wrapper/localsuite/creator_test.go b/wrapper/localsuite/creator_test.go
@@ -12,6 +12,7 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/require"
+
 	mockkms "github.com/trustbloc/kms-go/mock/kms"
 	kmsapi "github.com/trustbloc/kms-go/spi/kms"
 )
@@ -40,6 +41,23 @@ func TestKeyCreator(t *testing.T) {
 		require.IsType(t, ed25519.PublicKey{}, pubRaw)
 	})
 
+	t.Run("success export", func(t *testing.T) {
+		keyBytes, _, err := ed25519.GenerateKey(rand.Reader)
+		require.NoError(t, err)
+
+		keyID := "foo"
+
+		creator := newKeyCreator(&mockkms.KeyManager{
+			ExportPubKeyTypeValue:  kmsapi.ED25519Type,
+			ExportPubKeyBytesValue: keyBytes,
+		})
+
+		pubJWK, keyType, err := creator.ExportPubKeyBytes(keyID)
+		require.NoError(t, err)
+		require.EqualValues(t, kmsapi.ED25519Type, keyType)
+		require.NotNil(t, pubJWK)
+	})
+
 	t.Run("kms create err", func(t *testing.T) {
 		errExpected := errors.New("expected error")