diff options
Diffstat (limited to 'internal/cryptoutil/cryptoutil_test.go')
| -rw-r--r-- | internal/cryptoutil/cryptoutil_test.go | 20 |
1 files changed, 11 insertions, 9 deletions
diff --git a/internal/cryptoutil/cryptoutil_test.go b/internal/cryptoutil/cryptoutil_test.go index 1670eff..0e0560c 100644 --- a/internal/cryptoutil/cryptoutil_test.go +++ b/internal/cryptoutil/cryptoutil_test.go @@ -31,8 +31,8 @@ func TestPassword(t *testing.T) { func TestSignature(t *testing.T) { key := HMACKey(mustHex(t, "669e06ec457778b9a8133edb0a87ea82c6b141ffbbc63c038da96258175eb35c")) msg := []byte("test message") - signedMsg := key.Sign(msg) - got, ok := key.Verify(signedMsg) + signedMsg := key.Sign(msg, "info") + got, ok := key.Verify(signedMsg, "info") if !ok { t.Fatalf("Verify(%x) rejected the message", signedMsg) } @@ -64,14 +64,14 @@ func TestEncrypt(t *testing.T) { }} { t.Run(tc.desc, func(t *testing.T) { encryptedMsg := new(bytes.Buffer) - w := key.NewWriter(encryptedMsg, nil) + w := key.NewWriter(encryptedMsg, []byte("additional data")) if _, err := w.Write(tc.msg); err != nil { t.Fatalf("EncryptingWriter.Write(%q) failed: %s", tc.msg, err) } if err := w.Close(); err != nil { t.Fatalf("EncryptingWriter.Close() failed: %s", err) } - got, err := io.ReadAll(key.NewReader(bytes.NewReader(encryptedMsg.Bytes()), nil)) + got, err := io.ReadAll(key.NewReader(bytes.NewReader(encryptedMsg.Bytes()), []byte("additional data"))) if err != nil { t.Fatalf("DecryptingReader.Read(%x) failed: %s", encryptedMsg, err) } @@ -98,14 +98,14 @@ func TestDeriveKey(t *testing.T) { } msg := []byte("test message") encryptedMsg := new(bytes.Buffer) - w := key.NewWriter(encryptedMsg, nil) + w := key.NewWriter(encryptedMsg, []byte("additional data")) if _, err := w.Write(msg); err != nil { t.Fatalf("EncryptingWriter.Write(%q) failed: %s", msg, err) } if err := w.Close(); err != nil { t.Fatalf("EncryptingWriter.Close() failed: %s", err) } - got, err := io.ReadAll(key.NewReader(encryptedMsg, nil)) + got, err := io.ReadAll(key.NewReader(encryptedMsg, []byte("additional data"))) if err != nil { t.Fatalf("DecryptingReader.Read(%x) failed: %s", encryptedMsg, err) } @@ -121,8 +121,9 @@ func BenchmarkEncrypt(b *testing.B) { msg[i] = byte(i) } encryptedMsg := bytes.NewBuffer(make([]byte, 0, 35+len(msg)+16*(len(msg)+segmentSize-1)/segmentSize /* ?? */)) + additionalData := []byte("additional data") for b.Loop() { - w := key.NewWriter(encryptedMsg, nil) + w := key.NewWriter(encryptedMsg, additionalData) if _, err := w.Write(msg); err != nil { b.Fatal(err) } @@ -140,7 +141,8 @@ func BenchmarkDecrypt(b *testing.B) { msg[i] = byte(i) } encryptedMsg := new(bytes.Buffer) - w := key.NewWriter(encryptedMsg, nil) + additionalData := []byte("additional data") + w := key.NewWriter(encryptedMsg, additionalData) if _, err := w.Write(msg); err != nil { b.Fatal(err) } @@ -149,7 +151,7 @@ func BenchmarkDecrypt(b *testing.B) { } decryptedMsg := make([]byte, len(msg)) for b.Loop() { - r := key.NewReader(bytes.NewReader(encryptedMsg.Bytes()), nil) + r := key.NewReader(bytes.NewReader(encryptedMsg.Bytes()), additionalData) if _, err := io.ReadFull(r, decryptedMsg); err != nil { b.Fatal(err) } |