X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=src%2Fgovpn%2Fpeer_test.go;h=4e6f5879e5596ab72a01fc071927e39df932be2d;hb=933703051138e099c2fde03861278477058c3517;hp=dfdb941991b88858a6e3f5adc9c7613eb4a432cd;hpb=020a4f02b6ffe3bffd17ca6fd078278010e53025;p=govpn.git

diff --git a/src/govpn/peer_test.go b/src/govpn/peer_test.go
index dfdb941..4e6f587 100644
--- a/src/govpn/peer_test.go
+++ b/src/govpn/peer_test.go
@@ -20,6 +20,7 @@ package govpn
 
 import (
 	"testing"
+	"testing/quick"
 	"time"
 )
 
@@ -56,6 +57,40 @@ func init() {
 	plaintext = make([]byte, 789)
 }
 
+func TestSymmetric(t *testing.T) {
+	peerd := newPeer(true, "foo", Dummy{nil}, conf, new([SSize]byte))
+	f := func(payload []byte) bool {
+		if len(payload) == 0 {
+			return true
+		}
+		peer.EthProcess(payload)
+		return peerd.PktProcess(ciphertext, Dummy{nil}, true)
+	}
+	if err := quick.Check(f, nil); err != nil {
+		t.Error(err)
+	}
+}
+
+func TestSymmetricEncLess(t *testing.T) {
+	peerd := newPeer(true, "foo", Dummy{nil}, conf, new([SSize]byte))
+	peer.NoiseEnable = true
+	peer.EncLess = true
+	peerd.EncLess = true
+	peerd.NoiseEnable = true
+	f := func(payload []byte) bool {
+		if len(payload) == 0 {
+			return true
+		}
+		peer.EthProcess(payload)
+		return peerd.PktProcess(ciphertext, Dummy{nil}, true)
+	}
+	if err := quick.Check(f, nil); err != nil {
+		t.Error(err)
+	}
+	peer.NoiseEnable = false
+	peer.EncLess = false
+}
+
 func BenchmarkEnc(b *testing.B) {
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {