Remove dagState and GetUTXOFromID (ava-labs#1632)

joshua-kim · Jun 21, 2023 · 0073875 · 0073875
1 parent 94302bc
commit 0073875
Show file tree

Hide file tree

Showing 12 changed files with 101 additions and 417 deletions.
diff --git a/vms/avm/dag_state.go b/vms/avm/dag_state.go
diff --git a/vms/avm/index_test.go b/vms/avm/index_test.go
@@ -94,7 +94,7 @@ func TestIndexTransaction_Ordered(t *testing.T) {
 
 		var inputUTXOs []*avax.UTXO
 		for _, utxoID := range uniqueParsedTX.InputUTXOs() {
-			utxo, err := vm.dagState.GetUTXOFromID(utxoID)
+			utxo, err := vm.state.GetUTXO(utxoID.InputID())
 			require.NoError(err)
 
 			inputUTXOs = append(inputUTXOs, utxo)
@@ -170,7 +170,7 @@ func TestIndexTransaction_MultipleTransactions(t *testing.T) {
 
 		var inputUTXOs []*avax.UTXO
 		for _, utxoID := range uniqueParsedTX.InputUTXOs() {
-			utxo, err := vm.dagState.GetUTXOFromID(utxoID)
+			utxo, err := vm.state.GetUTXO(utxoID.InputID())
 			require.NoError(err)
 
 			inputUTXOs = append(inputUTXOs, utxo)
@@ -235,7 +235,7 @@ func TestIndexTransaction_MultipleAddresses(t *testing.T) {
 
 	var inputUTXOs []*avax.UTXO //nolint:prealloc
 	for _, utxoID := range tx.Unsigned.InputUTXOs() {
-		utxo, err := vm.dagState.GetUTXOFromID(utxoID)
+		utxo, err := vm.state.GetUTXO(utxoID.InputID())
 		require.NoError(err)
 
 		inputUTXOs = append(inputUTXOs, utxo)
@@ -303,7 +303,7 @@ func TestIndexTransaction_UnorderedWrites(t *testing.T) {
 
 		var inputUTXOs []*avax.UTXO
 		for _, utxoID := range uniqueParsedTX.InputUTXOs() {
-			utxo, err := vm.dagState.GetUTXOFromID(utxoID)
+			utxo, err := vm.state.GetUTXO(utxoID.InputID())
 			require.NoError(err)
 
 			inputUTXOs = append(inputUTXOs, utxo)