package main import ( "fmt" "io/ioutil" "os" "testing" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" "github.com/tendermint/tendermint/privval" ) const lastSignBytes = "750802110500000000000000220B08B398F3E00510F48DA6402A480A20FC25" + "8973076512999C3E6839A22E9FBDB1B77CF993E8A9955412A41A59D4CAD312240A20C971B286ACB8AA" + "A6FCA0365EB0A660B189EDC08B46B5AF2995DEFA51A28D215B10013211746573742D636861696E2D533245415533" const oldPrivvalContent = `{ "address": "1D8089FAFDFAE4A637F3D616E17B92905FA2D91D", "pub_key": { "type": "tendermint/PubKeyEd25519", "value": "r3Yg2AhDZ745CNTpavsGU+mRZ8WpRXqoJuyqjN8mJq0=" }, "last_height": "5", "last_round": "0", "last_step": 3, "last_signature": "CTr7b9ZQlrJJf+12rPl5t/YSCUc/KqV7jQogCfFJA24e7hof69X6OMT7eFLVQHyodPjD/QTA298XHV5ejxInDQ==", "last_signbytes": "` + lastSignBytes + `", "priv_key": { "type": "tendermint/PrivKeyEd25519", "value": "7MwvTGEWWjsYwjn2IpRb+GYsWi9nnFsw8jPLLY1UtP6vdiDYCENnvjkI1Olq+wZT6ZFnxalFeqgm7KqM3yYmrQ==" } }` func TestLoadAndUpgrade(t *testing.T) { oldFilePath := initTmpOldFile(t) defer os.Remove(oldFilePath) newStateFile, err := ioutil.TempFile("", "priv_validator_state*.json") defer os.Remove(newStateFile.Name()) require.NoError(t, err) newKeyFile, err := ioutil.TempFile("", "priv_validator_key*.json") defer os.Remove(newKeyFile.Name()) require.NoError(t, err) emptyOldFile, err := ioutil.TempFile("", "priv_validator_empty*.json") require.NoError(t, err) defer os.Remove(emptyOldFile.Name()) type args struct { oldPVPath string newPVKeyPath string newPVStatePath string } tests := []struct { name string args args wantErr bool wantPanic bool }{ {"successful upgrade", args{oldPVPath: oldFilePath, newPVKeyPath: newKeyFile.Name(), newPVStatePath: newStateFile.Name()}, false, false, }, {"unsuccessful upgrade: empty old privval file", args{oldPVPath: emptyOldFile.Name(), newPVKeyPath: newKeyFile.Name(), newPVStatePath: newStateFile.Name()}, true, false, }, {"unsuccessful upgrade: invalid new paths (1/3)", args{oldPVPath: oldFilePath, newPVKeyPath: "", newPVStatePath: newStateFile.Name()}, false, true, }, {"unsuccessful upgrade: invalid new paths (2/3)", args{oldPVPath: oldFilePath, newPVKeyPath: newKeyFile.Name(), newPVStatePath: ""}, false, true, }, {"unsuccessful upgrade: invalid new paths (3/3)", args{oldPVPath: oldFilePath, newPVKeyPath: "", newPVStatePath: ""}, false, true, }, } for _, tt := range tests { tt := tt t.Run(tt.name, func(t *testing.T) { // need to re-write the file everytime because upgrading renames it err := ioutil.WriteFile(oldFilePath, []byte(oldPrivvalContent), 0600) require.NoError(t, err) if tt.wantPanic { require.Panics(t, func() { loadAndUpgrade(tt.args.oldPVPath, tt.args.newPVKeyPath, tt.args.newPVStatePath) }) } else { err = loadAndUpgrade(tt.args.oldPVPath, tt.args.newPVKeyPath, tt.args.newPVStatePath) if tt.wantErr { assert.Error(t, err) fmt.Println("ERR", err) } else { assert.NoError(t, err) upgradedPV := privval.LoadFilePV(tt.args.newPVKeyPath, tt.args.newPVStatePath) oldPV, err := privval.LoadOldFilePV(tt.args.oldPVPath + ".bak") require.NoError(t, err) assert.Equal(t, oldPV.Address, upgradedPV.Key.Address) assert.Equal(t, oldPV.Address, upgradedPV.GetAddress()) assert.Equal(t, oldPV.PubKey, upgradedPV.Key.PubKey) assert.Equal(t, oldPV.PubKey, upgradedPV.GetPubKey()) assert.Equal(t, oldPV.PrivKey, upgradedPV.Key.PrivKey) assert.Equal(t, oldPV.LastHeight, upgradedPV.LastSignState.Height) assert.Equal(t, oldPV.LastRound, upgradedPV.LastSignState.Round) assert.Equal(t, oldPV.LastSignature, upgradedPV.LastSignState.Signature) assert.Equal(t, oldPV.LastSignBytes, upgradedPV.LastSignState.SignBytes) assert.Equal(t, oldPV.LastStep, upgradedPV.LastSignState.Step) } } }) } } func initTmpOldFile(t *testing.T) string { tmpfile, err := ioutil.TempFile("", "priv_validator_*.json") require.NoError(t, err) t.Logf("created test file %s", tmpfile.Name()) _, err = tmpfile.WriteString(oldPrivvalContent) require.NoError(t, err) return tmpfile.Name() }