PoA e2e test

scripts/abi_bindings.sh

@@ -16,7 +16,7 @@ export ARCH=$(uname -m)
 [ $ARCH = x86_64 ] && ARCH=amd64
 echo "ARCH set to $ARCH"
 
-DEFAULT_CONTRACT_LIST="TeleporterMessenger TeleporterRegistry ExampleERC20 TestMessenger ValidatorSetSig NativeTokenStakingManager ERC20TokenStakingManager"
+DEFAULT_CONTRACT_LIST="TeleporterMessenger TeleporterRegistry ExampleERC20 TestMessenger ValidatorSetSig NativeTokenStakingManager ERC20TokenStakingManager PoAValidatorManager"
 
 CONTRACT_LIST=
 HELP=

tests/flows/errors.go

@@ -1,5 +1,5 @@
 package flows
 
 const (
-	errTxReverted = "execution reverted"
+	ErrTxReverted = "execution reverted"
 )

tests/flows/validator_set_sig.go → tests/flows/governance/validator_set_sig.go

@@ -1,4 +1,4 @@
-package flows
+package governance
 
 import (
 	"context"

tests/flows/erc20_token_staking.go → tests/flows/staking/erc20_token_staking.go

@@ -1,4 +1,4 @@
-package flows
+package staking
 
 import (
 	"context"
@@ -82,7 +82,6 @@ func ERC20TokenStakingManager(network interfaces.LocalNetwork) {
 		// (Sending to the P-Chain will be skipped for now)
 		signedWarpMessage := network.ConstructSignedWarpMessage(context.Background(), receipt, subnetAInfo, pChainInfo)
 
-		// Validate the Warp message, (this will be done on the P-Chain in the future)
 		// Validate the Warp message, (this will be done on the P-Chain in the future)
 		utils.ValidateRegisterSubnetValidatorMessage(
 			signedWarpMessage,

tests/flows/native_token_staking.go → tests/flows/staking/native_token_staking.go

@@ -1,4 +1,4 @@
-package flows
+package staking
 
 import (
 	"context"

tests/flows/staking/poa_validator_manager.go

@@ -0,0 +1,206 @@
+package staking
+
+import (
+	"context"
+	"math/big"
+	"time"
+
+	"github.com/ava-labs/avalanchego/ids"
+	"github.com/ava-labs/avalanchego/utils/crypto/bls"
+	"github.com/ava-labs/subnet-evm/accounts/abi/bind"
+	"github.com/ava-labs/subnet-evm/core/types"
+	"github.com/ava-labs/teleporter/tests/interfaces"
+	"github.com/ava-labs/teleporter/tests/utils"
+	"github.com/ethereum/go-ethereum/crypto"
+	. "github.com/onsi/gomega"
+)
+
+/*
+ * Register a PoA validator manager on a L1. The steps are as follows:
+ * - Generate random address to be the owner address
+ * - Fund native assets to the owner address
+ * - Deploy the PoAValidatorManager contract
+ * - Attempt to initiate with non owner and check that it fails
+ * - Initiate validator registration
+ * - Deliver the Warp message to the P-Chain (not implemented)
+ * - Aggregate P-Chain signatures on the response Warp message
+ * - Deliver the Warp message to the L1
+ * - Verify that the validator is registered in the validator manager contract
+ *
+ * Delists the validator from the L1. The steps are as follows:
+ * - Attempt to initiate with non owner and check that it fails
+ * - Initiate validator delisting
+ * - Deliver the Warp message to the P-Chain (not implemented)
+ * - Aggregate P-Chain signatures on the response Warp message
+ * - Deliver the Warp message to the L1
+ * - Verify that the validator is delisted from the validator manager contract
+ */
+func PoAValidatorManager(network interfaces.LocalNetwork) {
+	cChainInfo := network.GetPrimaryNetworkInfo()
+	subnetAInfo, _ := utils.GetTwoSubnets(network)
+	_, fundedKey := network.GetFundedAccountInfo()
+	pChainInfo := utils.GetPChainInfo(cChainInfo)
+
+	signatureAggregator := utils.NewSignatureAggregator(
+		cChainInfo.NodeURIs[0],
+		[]ids.ID{
+			subnetAInfo.SubnetID,
+			ids.Empty, // Primary network subnet ID
+		},
+	)
+
+	// Generate random address to be the owner address
+	ownerKey, err := crypto.GenerateKey()
+	Expect(err).Should(BeNil())
+	ownerAddress := crypto.PubkeyToAddress(ownerKey.PublicKey)
+
+	// Transfer native assets to the owner account
+	ctx := context.Background()
+	fundAmount := big.NewInt(1e18) // 1avax
+	utils.SendNativeTransfer(
+		ctx,
+		subnetAInfo,
+		fundedKey,
+		ownerAddress,
+		fundAmount,
+	)
+
+	validatorManagerAddress, validatorManager := utils.DeployAndInitializePoAValidatorManager(
+		ctx,
+		fundedKey,
+		subnetAInfo,
+		pChainInfo,
+		ownerAddress,
+	)
+
+	var validationID ids.ID // To be used in the delisting step
+	nodeID := ids.GenerateTestID()
+	blsPublicKey := [bls.PublicKeyLen]byte{}
+	weight := uint64(1)
+
+	{
+		// Try to call with invalid owner
+		opts, err := bind.NewKeyedTransactorWithChainID(fundedKey, subnetAInfo.EVMChainID)
+		Expect(err).Should(BeNil())
+
+		_, err = validatorManager.InitializeValidatorRegistration(
+			opts,
+			weight,
+			nodeID,
+			uint64(time.Now().Add(24*time.Hour).Unix()),
+			blsPublicKey[:],
+		)
+		Expect(err).ShouldNot(BeNil())
+
+		// Initiate validator registration
+		var receipt *types.Receipt
+		receipt, validationID = utils.InitializePoAValidatorRegistration(
+			ownerKey,
+			subnetAInfo,
+			weight,
+			nodeID,
+			blsPublicKey,
+			validatorManager,
+		)
+
+		// Gather subnet-evm Warp signatures for the RegisterSubnetValidatorMessage & relay to the P-Chain
+		// (Sending to the P-Chain will be skipped for now)
+		signedWarpMessage := network.ConstructSignedWarpMessage(context.Background(), receipt, subnetAInfo, pChainInfo)
+
+		// Validate the Warp message, (this will be done on the P-Chain in the future)
+		utils.ValidateRegisterSubnetValidatorMessage(
+			signedWarpMessage,
+			nodeID,
+			weight,
+			subnetAInfo.SubnetID,
+			blsPublicKey,
+		)
+
+		// Construct a SubnetValidatorRegistrationMessage Warp message from the P-Chain
+		registrationSignedMessage := utils.ConstructSubnetValidatorRegistrationMessage(
+			validationID,
+			true,
+			subnetAInfo,
+			pChainInfo,
+			network,
+			signatureAggregator,
+		)
+
+		// Deliver the Warp message to the subnet
+		receipt = utils.CompletePoAValidatorRegistration(
+			fundedKey,
+			subnetAInfo,
+			validatorManagerAddress,
+			registrationSignedMessage,
+		)
+		// Check that the validator is registered in the staking contract
+		registrationEvent, err := utils.GetEventFromLogs(
+			receipt.Logs,
+			validatorManager.ParseValidationPeriodRegistered,
+		)
+		Expect(err).Should(BeNil())
+		Expect(registrationEvent.ValidationID[:]).Should(Equal(validationID[:]))
+	}
+
+	//
+	// Delist the validator
+	//
+	{
+		// Try with invalid validator owner
+		opts, err := bind.NewKeyedTransactorWithChainID(fundedKey, subnetAInfo.EVMChainID)
+		Expect(err).Should(BeNil())
+		_, err = validatorManager.InitializeEndValidation(
+			opts,
+			validationID,
+		)
+		Expect(err).ShouldNot(BeNil())
+
+		receipt := utils.InitializeEndPoAValidation(
+			ownerKey,
+			subnetAInfo,
+			validatorManager,
+			validationID,
+		)
+		validatorRemovalEvent, err := utils.GetEventFromLogs(
+			receipt.Logs,
+			validatorManager.ParseValidatorRemovalInitialized,
+		)
+		Expect(err).Should(BeNil())
+		Expect(validatorRemovalEvent.ValidationID[:]).Should(Equal(validationID[:]))
+		Expect(validatorRemovalEvent.Weight.Uint64()).Should(Equal(weight))
+
+		// Gather subnet-evm Warp signatures for the SetSubnetValidatorWeightMessage & relay to the P-Chain
+		// (Sending to the P-Chain will be skipped for now)
+		signedWarpMessage := network.ConstructSignedWarpMessage(context.Background(), receipt, subnetAInfo, pChainInfo)
+		Expect(err).Should(BeNil())
+
+		// Validate the Warp message, (this will be done on the P-Chain in the future)
+		utils.ValidateSetSubnetValidatorWeightMessage(signedWarpMessage, validationID, 0, 0)
+
+		// Construct a SubnetValidatorRegistrationMessage Warp message from the P-Chain
+		registrationSignedMessage := utils.ConstructSubnetValidatorRegistrationMessage(
+			validationID,
+			false,
+			subnetAInfo,
+			pChainInfo,
+			network,
+			signatureAggregator,
+		)
+
+		// Deliver the Warp message to the subnet
+		receipt = utils.CompleteEndPoAValidation(
+			ownerKey,
+			subnetAInfo,
+			validatorManagerAddress,
+			registrationSignedMessage,
+		)
+
+		// Check that the validator is has been delisted from the staking contract
+		registrationEvent, err := utils.GetEventFromLogs(
+			receipt.Logs,
+			validatorManager.ParseValidationPeriodEnded,
+		)
+		Expect(err).Should(BeNil())
+		Expect(registrationEvent.ValidationID[:]).Should(Equal(validationID[:]))
+	}
+}

tests/flows/add_fee_amount.go → tests/flows/teleporter/add_fee_amount.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/basic_send_receive.go → tests/flows/teleporter/basic_send_receive.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/deliver_to_nonexistent_contract.go → tests/flows/teleporter/deliver_to_nonexistent_contract.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/deliver_to_wrong_chain.go → tests/flows/teleporter/deliver_to_wrong_chain.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/insufficient_gas.go → tests/flows/teleporter/insufficient_gas.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/check_upgrade_access.go → tests/flows/teleporter/registry/check_upgrade_access.go

@@ -1,13 +1,15 @@
-package flows
+package registry
 
 import (
 	"context"
 	"math/big"
 
 	"github.com/ava-labs/subnet-evm/accounts/abi/bind"
+	"github.com/ava-labs/teleporter/tests/flows"
 	"github.com/ava-labs/teleporter/tests/interfaces"
 	"github.com/ava-labs/teleporter/tests/utils"
 	"github.com/ethereum/go-ethereum/crypto"
+
 	. "github.com/onsi/gomega"
 )
 
@@ -53,7 +55,7 @@ func CheckUpgradeAccess(network interfaces.Network) {
 	Expect(err).Should(BeNil())
 	_, err = testMessenger.PauseTeleporterAddress(nonOwnerOpts, teleporterAddress)
 	Expect(err).ShouldNot(BeNil())
-	Expect(err.Error()).Should(ContainSubstring(errTxReverted))
+	Expect(err.Error()).Should(ContainSubstring(flows.ErrTxReverted))
 
 	// Check that the teleporter address is not paused, because previous call should have failed
 	isPaused, err := testMessenger.IsTeleporterAddressPaused(&bind.CallOpts{}, teleporterAddress)
@@ -84,7 +86,7 @@ func CheckUpgradeAccess(network interfaces.Network) {
 	// Try to call unpauseTeleporterAddress from the previous owner account
 	_, err = testMessenger.UnpauseTeleporterAddress(ownerOpts, teleporterAddress)
 	Expect(err).ShouldNot(BeNil())
-	Expect(err.Error()).Should(ContainSubstring(errTxReverted))
+	Expect(err.Error()).Should(ContainSubstring(flows.ErrTxReverted))
 
 	// Make sure the teleporter address is still paused
 	isPaused, err = testMessenger.IsTeleporterAddressPaused(&bind.CallOpts{}, teleporterAddress)

tests/flows/pause_teleporter.go → tests/flows/teleporter/registry/pause_teleporter.go

@@ -1,4 +1,4 @@
-package flows
+package registry
 
 import (
 	"context"

tests/flows/teleporter_registry.go → tests/flows/teleporter/registry/teleporter_registry.go

@@ -1,4 +1,4 @@
-package flows
+package registry
 
 import (
 	"context"

tests/flows/relay_message_twice.go → tests/flows/teleporter/relay_message_twice.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/relayer_modifies_message.go → tests/flows/teleporter/relayer_modifies_message.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/resubmit_altered_message.go → tests/flows/teleporter/resubmit_altered_message.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/retry_successful_execution.go → tests/flows/teleporter/retry_successful_execution.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/send_specific_receipts.go → tests/flows/teleporter/send_specific_receipts.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"bytes"

tests/flows/teleporter_message_ids.go → tests/flows/teleporter/teleporter_message_ids.go

@@ -1,7 +1,7 @@
 // (c) 2024, Ava Labs, Inc. All rights reserved.
 // See the file LICENSE for licensing terms.
 
-package flows
+package teleporter
 
 import (
 	"math/big"

tests/flows/unallowed_relayer.go → tests/flows/teleporter/unallowed_relayer.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"

tests/flows/validator_churn.go → tests/flows/teleporter/validator_churn.go

@@ -1,4 +1,4 @@
-package flows
+package teleporter
 
 import (
 	"context"