feat: add set ejector function (#540)

Co-authored-by: Tomás Grüner <[email protected]>

chainio/clients/avsregistry/writer.go

@@ -697,3 +697,27 @@ func (w *ChainWriter) SetChurnApprover(
 	}
 	return receipt, nil
 }
+
+// Sets the ejector address to the one received by parameter. This address is the only one
+// that can eject operators. Returns the receipt of the transaction in case of success.
+func (w *ChainWriter) SetEjector(
+	ctx context.Context,
+	ejectorAddress gethcommon.Address,
+	waitForReceipt bool,
+) (*gethtypes.Receipt, error) {
+	w.logger.Info("setting ejector with address ", ejectorAddress)
+
+	noSendTxOpts, err := w.txMgr.GetNoSendTxOpts()
+	if err != nil {
+		return nil, err
+	}
+	tx, err := w.registryCoordinator.SetEjector(noSendTxOpts, ejectorAddress)
+	if err != nil {
+		return nil, err
+	}
+	receipt, err := w.txMgr.Send(ctx, tx, waitForReceipt)
+	if err != nil {
+		return nil, utils.WrapError("failed to send SetEjector tx with err", err.Error())
+	}
+	return receipt, nil
+}

chainio/clients/avsregistry/writer_test.go

@@ -535,3 +535,37 @@ func TestSetChurnApprover(t *testing.T) {
 	require.NoError(t, err)
 	assert.Equal(t, newApprover.String(), testutils.ANVIL_SECOND_ADDRESS)
 }
+
+func TestSetEjector(t *testing.T) {
+	// Test set up
+	clients, anvilHttpEndpoint := testclients.BuildTestClients(t)
+
+	contractAddrs := testutils.GetContractAddressesFromContractRegistry(anvilHttpEndpoint)
+
+	chainWriter := clients.AvsRegistryChainWriter
+
+	ejectorAddress := gethcommon.HexToAddress(testutils.ANVIL_SECOND_ADDRESS)
+
+	ethHttpClient := clients.EthHttpClient
+
+	registryCoordinatorContract, err := regcoord.NewContractRegistryCoordinator(
+		contractAddrs.RegistryCoordinator,
+		ethHttpClient,
+	)
+	require.NoError(t, err)
+
+	// At first, ejector is anvil first address
+	ejector, err := registryCoordinatorContract.Ejector(&bind.CallOpts{})
+	require.NoError(t, err)
+	assert.Equal(t, ejector.String(), testutils.ANVIL_FIRST_ADDRESS)
+
+	// Set a new ejector
+	receipt, err := chainWriter.SetEjector(context.Background(), ejectorAddress, true)
+	require.NoError(t, err)
+	require.Equal(t, receipt.Status, gethtypes.ReceiptStatusSuccessful)
+
+	// After change, ejector is the setted value
+	newEjector, err := registryCoordinatorContract.Ejector(&bind.CallOpts{})
+	require.NoError(t, err)
+	assert.Equal(t, newEjector.String(), testutils.ANVIL_SECOND_ADDRESS)
+}