pem encode

opentdf · Aug 10, 2024 · 5f10293 · 5f10293
1 parent ff79764
commit 5f10293
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/tests/util.go b/tests/util.go
@@ -23,11 +23,18 @@ func generateKasRSAKeyPair() ([]byte, []byte, error) {
 		},
 	)
 
-	pubKeyPEM, err := x509.MarshalPKIXPublicKey(&pubKey)
+	pk, err := x509.MarshalPKIXPublicKey(&pubKey)
 	if err != nil {
 		return nil, nil, err
 	}
 
+	pubKeyPEM := pem.EncodeToMemory(
+		&pem.Block{
+			Type:  "PUBLIC KEY",
+			Bytes: pk,
+		},
+	)
+
 	return privKeyPEM, pubKeyPEM, nil
 }
 
@@ -44,11 +51,16 @@ func generateKasECDHKeyPair() ([]byte, []byte, error) {
 		Bytes: privKey.Bytes(),
 	}
 
-	pubKeyPEM, err := x509.MarshalPKIXPublicKey(pubKey)
+	pk, err := x509.MarshalPKIXPublicKey(pubKey)
 	if err != nil {
 		return nil, nil, err
 	}
 
+	pubKeyPEM := pem.EncodeToMemory(&pem.Block{
+		Type:  "PUBLIC KEY",
+		Bytes: pk,
+	})
+
 	return pem.EncodeToMemory(&privKeyPEM), pubKeyPEM, nil
 
 }