github.com/prysmaticlabs/prysm@v1.4.4/tools/unencrypted-keys-gen/main_test.go (about)

     1  package main
     2  
     3  import (
     4  	"bytes"
     5  	"encoding/json"
     6  	"testing"
     7  
     8  	"github.com/prysmaticlabs/prysm/shared/testutil/assert"
     9  	"github.com/prysmaticlabs/prysm/shared/testutil/require"
    10  	"github.com/prysmaticlabs/prysm/tools/unencrypted-keys-gen/keygen"
    11  )
    12  
    13  func TestSavesUnencryptedKeys(t *testing.T) {
    14  	keys := 2
    15  	numKeys = &keys
    16  	ctnr := generateUnencryptedKeys(0 /* start index */)
    17  	buf := new(bytes.Buffer)
    18  	require.NoError(t, keygen.SaveUnencryptedKeysToFile(buf, ctnr))
    19  	enc := buf.Bytes()
    20  	dec := &keygen.UnencryptedKeysContainer{}
    21  	require.NoError(t, json.Unmarshal(enc, dec))
    22  	assert.DeepEqual(t, ctnr, dec)
    23  }