github.com/evdatsion/aphelion-dpos-bft@v0.32.1/privval/file_deprecated_test.go (about)

     1  package privval_test
     2  
     3  import (
     4  	"io/ioutil"
     5  	"os"
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  	"github.com/stretchr/testify/require"
    10  
    11  	"github.com/evdatsion/aphelion-dpos-bft/privval"
    12  )
    13  
    14  const oldPrivvalContent = `{
    15    "address": "1D8089FAFDFAE4A637F3D616E17B92905FA2D91D",
    16    "pub_key": {
    17      "type": "tendermint/PubKeyEd25519",
    18      "value": "r3Yg2AhDZ745CNTpavsGU+mRZ8WpRXqoJuyqjN8mJq0="
    19    },
    20    "last_height": "5",
    21    "last_round": "0",
    22    "last_step": 3,
    23    "last_signature": "CTr7b9ZQlrJJf+12rPl5t/YSCUc/KqV7jQogCfFJA24e7hof69X6OMT7eFLVQHyodPjD/QTA298XHV5ejxInDQ==",
    24    "last_signbytes": "750802110500000000000000220B08B398F3E00510F48DA6402A480A20FC258973076512999C3E6839A22E9FBDB1B77CF993E8A9955412A41A59D4CAD312240A20C971B286ACB8AAA6FCA0365EB0A660B189EDC08B46B5AF2995DEFA51A28D215B10013211746573742D636861696E2D533245415533",
    25    "priv_key": {
    26      "type": "tendermint/PrivKeyEd25519",
    27      "value": "7MwvTGEWWjsYwjn2IpRb+GYsWi9nnFsw8jPLLY1UtP6vdiDYCENnvjkI1Olq+wZT6ZFnxalFeqgm7KqM3yYmrQ=="
    28    }
    29  }`
    30  
    31  func TestLoadAndUpgrade(t *testing.T) {
    32  
    33  	oldFilePath := initTmpOldFile(t)
    34  	defer os.Remove(oldFilePath)
    35  	newStateFile, err := ioutil.TempFile("", "priv_validator_state*.json")
    36  	defer os.Remove(newStateFile.Name())
    37  	require.NoError(t, err)
    38  	newKeyFile, err := ioutil.TempFile("", "priv_validator_key*.json")
    39  	defer os.Remove(newKeyFile.Name())
    40  	require.NoError(t, err)
    41  
    42  	oldPV, err := privval.LoadOldFilePV(oldFilePath)
    43  	assert.NoError(t, err)
    44  	newPV := oldPV.Upgrade(newKeyFile.Name(), newStateFile.Name())
    45  
    46  	assertEqualPV(t, oldPV, newPV)
    47  	assert.NoError(t, err)
    48  	upgradedPV := privval.LoadFilePV(newKeyFile.Name(), newStateFile.Name())
    49  	assertEqualPV(t, oldPV, upgradedPV)
    50  	oldPV, err = privval.LoadOldFilePV(oldFilePath + ".bak")
    51  	require.NoError(t, err)
    52  	assertEqualPV(t, oldPV, upgradedPV)
    53  }
    54  
    55  func assertEqualPV(t *testing.T, oldPV *privval.OldFilePV, newPV *privval.FilePV) {
    56  	assert.Equal(t, oldPV.Address, newPV.Key.Address)
    57  	assert.Equal(t, oldPV.Address, newPV.GetAddress())
    58  	assert.Equal(t, oldPV.PubKey, newPV.Key.PubKey)
    59  	assert.Equal(t, oldPV.PubKey, newPV.GetPubKey())
    60  	assert.Equal(t, oldPV.PrivKey, newPV.Key.PrivKey)
    61  
    62  	assert.Equal(t, oldPV.LastHeight, newPV.LastSignState.Height)
    63  	assert.Equal(t, oldPV.LastRound, newPV.LastSignState.Round)
    64  	assert.Equal(t, oldPV.LastSignature, newPV.LastSignState.Signature)
    65  	assert.Equal(t, oldPV.LastSignBytes, newPV.LastSignState.SignBytes)
    66  	assert.Equal(t, oldPV.LastStep, newPV.LastSignState.Step)
    67  }
    68  
    69  func initTmpOldFile(t *testing.T) string {
    70  	tmpfile, err := ioutil.TempFile("", "priv_validator_*.json")
    71  	require.NoError(t, err)
    72  	t.Logf("created test file %s", tmpfile.Name())
    73  	_, err = tmpfile.WriteString(oldPrivvalContent)
    74  	require.NoError(t, err)
    75  
    76  	return tmpfile.Name()
    77  }