34 lines
705 B
Go
34 lines
705 B
Go
|
package sshutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/sha256"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/drone/drone/Godeps/_workspace/src/github.com/franela/goblin"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSSHUtil(t *testing.T) {
|
||
|
|
|
||
|
|
g := goblin.Goblin(t)
|
||
|
|
g.Describe("sshutil", func() {
|
||
|
|
var encrypted, testMsg string
|
||
|
|
|
||
|
|
privkey, err := GeneratePrivateKey()
|
||
|
|
g.Assert(err == nil).IsTrue()
|
||
|
|
pubkey := privkey.PublicKey
|
||
|
|
sha256 := sha256.New()
|
||
|
|
testMsg = "foo=bar"
|
||
|
|
|
||
|
|
g.Before(func() {
|
||
|
|
encrypted, err = Encrypt(sha256, &pubkey, testMsg)
|
||
|
|
g.Assert(err == nil).IsTrue()
|
||
|
|
})
|
||
|
|
|
||
|
|
g.It("Can decrypt encrypted msg", func() {
|
||
|
|
decrypted, err := Decrypt(sha256, privkey, encrypted)
|
||
|
|
g.Assert(err == nil).IsTrue()
|
||
|
|
g.Assert(decrypted == testMsg).IsTrue()
|
||
|
|
})
|
||
|
|
})
|
||
|
|
}
|