refactor tests

Signed-off-by: Joshua Kim <[email protected]>

Author: joshua-kim

vms/avm/vm.go

@@ -10,7 +10,6 @@ import (
 	"net/http"
 	"reflect"
 	"sync"
-	"time"
 
 	"github.com/gorilla/rpc/v2"
 	"github.com/prometheus/client_golang/prometheus"
@@ -395,7 +394,7 @@ func (vm *VM) Linearize(ctx context.Context, stopVertexID ids.ID, toEngine chan<
 		stopVertexID,
 		vm.Config.Upgrades.CortinaTime,
 	); err != nil {
-		return err
+		return fmt.Errorf("failed to initialize legacy chain state: %w", err)
 	}
 
 	stateMigration := vm.StateMigrationFactory.New(
@@ -524,10 +523,6 @@ func (vm *VM) GetUTXO(utxoID ids.ID) (*avax.UTXO, error) {
 	return vm.state.GetUTXO(utxoID)
 }
 
-func (vm *VM) GetLastAcceptedTimestamp() time.Time {
-	return vm.state.GetTimestamp()
-}
-
 /*
  ******************************************************************************
  ********************************** JSON API **********************************

vms/avm/vm_test.go

@@ -823,7 +823,6 @@ func TestVMLinearizeStateMigration(t *testing.T) {
 	}
 
 	require.Equal(numBlocks, wantBlkIDs.Len())
-	wantLastAcceptedTimestamp := vm.GetLastAcceptedTimestamp()
 
 	db, err = memdb.Copy(db)
 	require.NoError(vm.Shutdown(context.Background()))
@@ -879,12 +878,10 @@ func TestVMLinearizeStateMigration(t *testing.T) {
 		require.Equal(wantUTXO.InputID(), gotUTXO.InputID())
 	}
 
+	wantLastAcceptedBlkID, _, _ := wantBlkIDs.Newest()
 	gotLastAcceptedBlkID, err := vm.LastAccepted(context.Background())
 	require.NoError(err)
-	wantLastAcceptedBlkID, _, ok := wantBlkIDs.Newest()
-	require.True(ok)
 	require.Equal(wantLastAcceptedBlkID, gotLastAcceptedBlkID)
-	require.Equal(wantLastAcceptedTimestamp, vm.GetLastAcceptedTimestamp())
 
 	db, err = memdb.Copy(db)
 	require.NoError(vm.Shutdown(context.Background()))