Skip to content

Commit cd2eefa

Browse files
committed
obfuscation_test: test obfuscator serialization
1 parent 57bfe90 commit cd2eefa

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

obfuscation_test.go

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package sphinx
33
import (
44
"bytes"
55
"encoding/hex"
6+
"reflect"
67
"testing"
78

89
"github.com/roasbeef/btcd/btcec"
@@ -201,6 +202,21 @@ func TestOnionFailureSpecVector(t *testing.T) {
201202
sharedSecret: sharedSecrets[len(sharedSecrets)-1-i],
202203
}
203204

205+
var b bytes.Buffer
206+
if err := obfuscator.Encode(&b); err != nil {
207+
t.Fatalf("unable to encode obfuscator: %v", err)
208+
}
209+
210+
obfuscator2 := &OnionErrorEncrypter{}
211+
obfuscatorReader := bytes.NewReader(b.Bytes())
212+
if err := obfuscator2.Decode(obfuscatorReader); err != nil {
213+
t.Fatalf("unable to decode obfuscator: %v", err)
214+
}
215+
216+
if !reflect.DeepEqual(obfuscator, obfuscator2) {
217+
t.Fatalf("unable to reconstruct obfuscator: %v", err)
218+
}
219+
204220
if !bytes.Equal(expectedSharedSecret, obfuscator.sharedSecret[:]) {
205221
t.Fatalf("shared secret not match with spec: expected "+
206222
"%x, got %x", expectedSharedSecret,

0 commit comments

Comments
 (0)