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 }