lnd.xprv/shachain/producer_test.go

package shachain

import (
	"bytes"
	"testing"

	"github.com/roasbeef/btcd/chaincfg/chainhash"
)

// TestShaChainProducerRestore checks the ability of shachain producer to be
// properly recreated from binary representation.
func TestShaChainProducerRestore(t *testing.T) {
	var err error

	seed := chainhash.DoubleHashH([]byte("shachaintest"))
	sender := NewRevocationProducer(seed)

	s1, err := sender.AtIndex(0)
	if err != nil {
		t.Fatal(err)
	}

	var b bytes.Buffer
	if err := sender.Encode(&b); err != nil {
		t.Fatal(err)
	}

	sender, err = NewRevocationProducerFromBytes(b.Bytes())
	if err != nil {
		t.Fatal(err)
	}

	s3, err := sender.AtIndex(0)
	if err != nil {
		t.Fatal(err)
	}

	if !s1.IsEqual(s3) {
		t.Fatalf("secrets should match: %v:%v", s1.String(), s3.String())
	}
}