@@ -11,7 +11,6 @@ import (
1111 "github.com/ava-labs/avalanchego/ids"
1212 agoUtils "github.com/ava-labs/avalanchego/utils"
1313 "github.com/ava-labs/avalanchego/utils/set"
14- "github.com/ava-labs/avalanchego/vms/platformvm/warp"
1514 avalancheWarp "github.com/ava-labs/avalanchego/vms/platformvm/warp"
1615 "github.com/ava-labs/avalanchego/vms/platformvm/warp/payload"
1716 "github.com/ava-labs/libevm/common"
@@ -183,9 +182,9 @@ func TestGetVerifiedWarpMessage(t *testing.T) {
183182 packagedPayloadBytes ,
184183 )
185184 require .NoError (t , err )
186- unsignedWarpMsg , err := warp .NewUnsignedMessage (networkID , sourceChainID , addressedPayload .Bytes ())
185+ unsignedWarpMsg , err := avalancheWarp .NewUnsignedMessage (networkID , sourceChainID , addressedPayload .Bytes ())
187186 require .NoError (t , err )
188- warpMessage , err := warp .NewMessage (unsignedWarpMsg , & warp .BitSetSignature {}) // Create message with empty signature for testing
187+ warpMessage , err := avalancheWarp .NewMessage (unsignedWarpMsg , & avalancheWarp .BitSetSignature {}) // Create message with empty signature for testing
189188 require .NoError (t , err )
190189 warpMessagePredicateBytes := predicate .PackPredicate (warpMessage .Bytes ())
191190 getVerifiedWarpMsg , err := PackGetVerifiedWarpMessage (0 )
@@ -407,9 +406,9 @@ func TestGetVerifiedWarpMessage(t *testing.T) {
407406 Caller : callerAddr ,
408407 InputFn : func (t testing.TB ) []byte { return getVerifiedWarpMsg },
409408 BeforeHook : func (t testing.TB , state contract.StateDB ) {
410- unsignedMessage , err := warp .NewUnsignedMessage (networkID , sourceChainID , []byte {1 , 2 , 3 }) // Invalid addressed payload
409+ unsignedMessage , err := avalancheWarp .NewUnsignedMessage (networkID , sourceChainID , []byte {1 , 2 , 3 }) // Invalid addressed payload
411410 require .NoError (t , err )
412- warpMessage , err := warp .NewMessage (unsignedMessage , & warp .BitSetSignature {})
411+ warpMessage , err := avalancheWarp .NewMessage (unsignedMessage , & avalancheWarp .BitSetSignature {})
413412 require .NoError (t , err )
414413
415414 state .SetPredicateStorageSlots (ContractAddress , [][]byte {predicate .PackPredicate (warpMessage .Bytes ())})
@@ -464,9 +463,9 @@ func TestGetVerifiedWarpBlockHash(t *testing.T) {
464463 blockHash := ids .GenerateTestID ()
465464 blockHashPayload , err := payload .NewHash (blockHash )
466465 require .NoError (t , err )
467- unsignedWarpMsg , err := warp .NewUnsignedMessage (networkID , sourceChainID , blockHashPayload .Bytes ())
466+ unsignedWarpMsg , err := avalancheWarp .NewUnsignedMessage (networkID , sourceChainID , blockHashPayload .Bytes ())
468467 require .NoError (t , err )
469- warpMessage , err := warp .NewMessage (unsignedWarpMsg , & warp .BitSetSignature {}) // Create message with empty signature for testing
468+ warpMessage , err := avalancheWarp .NewMessage (unsignedWarpMsg , & avalancheWarp .BitSetSignature {}) // Create message with empty signature for testing
470469 require .NoError (t , err )
471470 warpMessagePredicateBytes := predicate .PackPredicate (warpMessage .Bytes ())
472471 getVerifiedWarpBlockHash , err := PackGetVerifiedWarpBlockHash (0 )
@@ -685,9 +684,9 @@ func TestGetVerifiedWarpBlockHash(t *testing.T) {
685684 Caller : callerAddr ,
686685 InputFn : func (t testing.TB ) []byte { return getVerifiedWarpBlockHash },
687686 BeforeHook : func (t testing.TB , state contract.StateDB ) {
688- unsignedMessage , err := warp .NewUnsignedMessage (networkID , sourceChainID , []byte {1 , 2 , 3 }) // Invalid block hash payload
687+ unsignedMessage , err := avalancheWarp .NewUnsignedMessage (networkID , sourceChainID , []byte {1 , 2 , 3 }) // Invalid block hash payload
689688 require .NoError (t , err )
690- warpMessage , err := warp .NewMessage (unsignedMessage , & warp .BitSetSignature {})
689+ warpMessage , err := avalancheWarp .NewMessage (unsignedMessage , & avalancheWarp .BitSetSignature {})
691690 require .NoError (t , err )
692691
693692 state .SetPredicateStorageSlots (ContractAddress , [][]byte {predicate .PackPredicate (warpMessage .Bytes ())})
0 commit comments