初版功能完成
This commit is contained in:
@@ -0,0 +1,27 @@
|
||||
package clientwg
|
||||
|
||||
import (
|
||||
"encoding/base64"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestParseKeyBase64(t *testing.T) {
|
||||
t.Parallel()
|
||||
raw := make([]byte, keySize)
|
||||
for index := range raw {
|
||||
raw[index] = byte(index + 1)
|
||||
}
|
||||
key, err := ParseKeyBase64(base64.StdEncoding.EncodeToString(raw))
|
||||
if err != nil {
|
||||
t.Fatalf("ParseKeyBase64() error = %v", err)
|
||||
}
|
||||
if key[0] != 1 || key[31] != 32 {
|
||||
t.Fatalf("ParseKeyBase64() returned unexpected bytes: first=%d last=%d", key[0], key[31])
|
||||
}
|
||||
if _, err := ParseKeyBase64("not-base64"); err == nil {
|
||||
t.Fatal("ParseKeyBase64() accepted invalid base64")
|
||||
}
|
||||
if _, err := ParseKeyBase64(base64.StdEncoding.EncodeToString([]byte{1, 2, 3})); err == nil {
|
||||
t.Fatal("ParseKeyBase64() accepted a short key")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user