aboutsummaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/util.go4
-rw-r--r--lib/util_test.go3
2 files changed, 4 insertions, 3 deletions
diff --git a/lib/util.go b/lib/util.go
index b1c7b87..ca88f3f 100644
--- a/lib/util.go
+++ b/lib/util.go
@@ -3,8 +3,8 @@ package lib
import "golang.org/x/crypto/ssh"
// GetPublicKey marshals a ssh certificate to a string.
-func GetPublicKey(pub ssh.PublicKey) string {
+func GetPublicKey(pub ssh.PublicKey) []byte {
marshaled := ssh.MarshalAuthorizedKey(pub)
// Strip trailing newline
- return string(marshaled[:len(marshaled)-1])
+ return marshaled[:len(marshaled)-1]
}
diff --git a/lib/util_test.go b/lib/util_test.go
index 9e89297..b1f18a8 100644
--- a/lib/util_test.go
+++ b/lib/util_test.go
@@ -1,6 +1,7 @@
package lib
import (
+ "reflect"
"testing"
"github.com/nsheridan/cashier/testdata"
@@ -10,7 +11,7 @@ import (
func TestGetPublicKey(t *testing.T) {
t.Parallel()
c, _, _, _, _ := ssh.ParseAuthorizedKey(testdata.Cert)
- if GetPublicKey(c.(*ssh.Certificate)) != string(testdata.Cert) {
+ if !reflect.DeepEqual(GetPublicKey(c.(*ssh.Certificate)), testdata.Cert) {
t.Fail()
}
}