add marshal test for public key

1 files changed, 6 insertions, 2 deletions

diff --git a/private_key_test.go b/private_key_test.go
index 0a3fd00..1d8f5ba 100644
--- a/private_key_test.go
+++ b/private_key_test.go
@@ -11,8 +11,8 @@ var (
 )
 
 // run the marshal test
-func RunMarshalTest(pk_type string, pk Pemmer, label string, t *testing.T) ([]byte, error) {
-  marsh_pem, err := pk.MarshalPem()
+func RunMarshalTest(pk_type string, pe Pemmer, label string, t *testing.T) ([]byte, error) {
+  marsh_pem, err := pe.MarshalPem()
   if err != nil {
     t.Errorf("%s: marshal pem not working: %s", pk_type, err)
     return nil, err
@@ -30,6 +30,10 @@ func RunMarshalTest(pk_type string, pk Pemmer, label string, t *testing.T) ([]by
 func RunPrivateKeyTests(pk_type string, pk PrivateKey, t *testing.T) {
   pu := pk.Public()
 
+  // TODO check return result of the marshalled public key
+  _, err := RunMarshalTest(pk_type + "-public", pu, PemLabelPublic, t)
+  if err != nil { return }
+
   signature, err := pk.Sign(SignatureMessage)
   if err != nil { t.Errorf("%s: error creating a signature: %s", pk_type, err) }