delete blocks

Signed-off-by: Joshua Kim <20001595+joshua-kim@users.noreply.github.com>

Author: joshua-kim

vms/avm/state.go

@@ -225,6 +225,14 @@ func (g *gForkStateMigration) Migrate(
 		if err := next.state.BlockDB.Put(itr.Key(), itr.Value()); err != nil {
 			return nil, fmt.Errorf("failed to migrate block: %w", err)
 		}
+
+		if err := prevBlockDB.Delete(itr.Key()); err != nil {
+			return nil, fmt.Errorf("failed to delete migrated block: %w", err)
+		}
+
+		if err := g.commit(baseDB); err != nil {
+			return nil, fmt.Errorf("failed to commit db: %w", err)
+		}
 	}
 
 	g.log.Debug("migrating height index")

vms/avm/vm_test.go

@@ -901,7 +901,10 @@ func TestVMLinearizeStateMigration(t *testing.T) {
 
 	for itr := wantBlkIDs.NewIterator(); itr.Next(); {
 		_, err := vm.GetBlock(context.Background(), itr.Key())
-		require.NoError(err)
+		require.ErrorIs(err, database.ErrNotFound)
+
+		_, err = vm.GetBlockIDAtHeight(context.Background(), itr.Value())
+		require.NoError(err, database.ErrNotFound)
 	}
 
 	for _, txID := range wantTxs {