[pull] develop from smartcontractkit:develop

core/capabilities/integration_tests/framework/capabilities_registry.go

@@ -0,0 +1,125 @@
+package framework
+
+import (
+	"context"
+	"testing"
+
+	"github.com/ethereum/go-ethereum"
+	"github.com/ethereum/go-ethereum/accounts/abi/bind"
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/stretchr/testify/require"
+	"google.golang.org/protobuf/proto"
+
+	"github.com/smartcontractkit/chainlink-common/pkg/capabilities/pb"
+	kcr "github.com/smartcontractkit/chainlink-evm/gethwrappers/keystone/generated/capabilities_registry_1_1_0"
+	p2ptypes "github.com/smartcontractkit/chainlink/v2/core/services/p2p/types"
+)
+
+type CapabilitiesRegistry struct {
+	t              *testing.T
+	backend        *EthBlockchain
+	contract       *kcr.CapabilitiesRegistry
+	addr           common.Address
+	nodeOperatorID uint32
+	nextDonID      int
+}
+
+func NewCapabilitiesRegistry(ctx context.Context, t *testing.T, backend *EthBlockchain) *CapabilitiesRegistry {
+	addr, _, contract, err := kcr.DeployCapabilitiesRegistry(backend.transactionOpts, backend.Client())
+	require.NoError(t, err)
+	backend.Commit()
+
+	_, err = contract.AddNodeOperators(backend.transactionOpts, []kcr.CapabilitiesRegistryNodeOperator{
+		{
+			Admin: backend.transactionOpts.From,
+			Name:  "TEST_NODE_OPERATOR",
+		},
+	})
+	require.NoError(t, err)
+	blockHash := backend.Commit()
+
+	logs, err := backend.Client().FilterLogs(ctx, ethereum.FilterQuery{
+		BlockHash: &blockHash,
+		FromBlock: nil,
+		ToBlock:   nil,
+		Addresses: nil,
+		Topics:    nil,
+	})
+
+	require.NoError(t, err)
+
+	recLog, err := contract.ParseNodeOperatorAdded(logs[0])
+	require.NoError(t, err)
+
+	nopID := recLog.NodeOperatorId
+
+	return &CapabilitiesRegistry{t: t, addr: addr, contract: contract, backend: backend, nodeOperatorID: nopID}
+}
+
+func (r *CapabilitiesRegistry) getAddress() common.Address {
+	return r.addr
+}
+
+type capability struct {
+	donCapabilityConfig *pb.CapabilityConfig
+	registryConfig      kcr.CapabilitiesRegistryCapability
+	// internalOnly is true if the capability is published in the registry but not made available outside the DON in which it runs
+	internalOnly bool
+}
+
+// SetupDON sets up a new DON with the given capabilities and returns the DON ID
+func (r *CapabilitiesRegistry) setupDON(donInfo DonConfiguration, capabilities []capability) int {
+	hashedCapabilityIDs := make([][32]byte, 0, len(capabilities))
+
+	for _, c := range capabilities {
+		id, err := r.contract.GetHashedCapabilityId(&bind.CallOpts{}, c.registryConfig.LabelledName, c.registryConfig.Version)
+		require.NoError(r.t, err)
+		hashedCapabilityIDs = append(hashedCapabilityIDs, id)
+	}
+
+	registryCapabilities := make([]kcr.CapabilitiesRegistryCapability, 0, len(capabilities))
+	for _, c := range capabilities {
+		registryCapabilities = append(registryCapabilities, c.registryConfig)
+	}
+
+	_, err := r.contract.AddCapabilities(r.backend.transactionOpts, registryCapabilities)
+	require.NoError(r.t, err)
+
+	r.backend.Commit()
+
+	peerIDs := make([][32]byte, 0, len(donInfo.p2pKeys))
+	nodes := make([]kcr.CapabilitiesRegistryNodeParams, 0, len(donInfo.p2pKeys))
+	for i, p2pkey := range donInfo.p2pKeys {
+		signer, innerErr := getSignerStringFromOCRKeyBundle(donInfo.KeyBundles[i])
+		require.NoError(r.t, innerErr)
+		peer := peerIDAndOCRSigner{PeerID: p2ptypes.PeerID(p2pkey.PeerID()), Signer: signer}
+		peerIDs = append(peerIDs, p2pkey.PeerID())
+		n, innerErr := peerToNode(r.nodeOperatorID, peer)
+		require.NoError(r.t, innerErr)
+
+		n.HashedCapabilityIds = hashedCapabilityIDs
+		nodes = append(nodes, n)
+	}
+
+	_, err = r.contract.AddNodes(r.backend.transactionOpts, nodes)
+	require.NoError(r.t, err)
+	r.backend.Commit()
+
+	capabilityConfigurations := make([]kcr.CapabilitiesRegistryCapabilityConfiguration, 0, len(capabilities))
+	for i, c := range capabilities {
+		configBinary, err2 := proto.Marshal(c.donCapabilityConfig)
+		require.NoError(r.t, err2)
+
+		capabilityConfigurations = append(capabilityConfigurations, kcr.CapabilitiesRegistryCapabilityConfiguration{
+			CapabilityId: hashedCapabilityIDs[i],
+			Config:       configBinary,
+		})
+	}
+
+	_, err = r.contract.AddDON(r.backend.transactionOpts, peerIDs, capabilityConfigurations, true, donInfo.AcceptsWorkflows, donInfo.F)
+	require.NoError(r.t, err)
+	r.backend.Commit()
+
+	r.nextDonID++
+	return r.nextDonID
+}