[PRIV-128] Add DONForCapability to capability registry

Author: cedric-cordenier

pkg/loop/internal/core/services/capability/capabilities_registry.go

@@ -78,6 +78,22 @@ func (cr *capabilitiesRegistryClient) NodeByPeerID(ctx context.Context, peerID p
 	return cr.nodeFromNodeReply(res), nil
 }
 
+func (cr *capabilitiesRegistryClient) DONForCapability(ctx context.Context, capabilityID string) (capabilities.DON, []capabilities.Node, error) {
+	res, err := cr.grpc.DONForCapability(ctx, &pb.DONForCapabilityRequest{CapabilityID: capabilityID})
+	if err != nil {
+		return capabilities.DON{}, nil, err
+	}
+
+	don := toDON(res.Don)
+
+	var nodes []capabilities.Node
+	for _, n := range res.Nodes {
+		nodes = append(nodes, cr.nodeFromNodeReply(n))
+	}
+
+	return don, nodes, nil
+}
+
 func (cr *capabilitiesRegistryClient) nodeFromNodeReply(nodeReply *pb.NodeReply) capabilities.Node {
 	var pid *p2ptypes.PeerID
 	if len(nodeReply.PeerID) > 0 {
@@ -365,6 +381,23 @@ func (c *capabilitiesRegistryServer) NodeByPeerID(ctx context.Context, nodeReque
 	return c.nodeReplyFromNode(node), nil
 }
 
+func (c *capabilitiesRegistryServer) DONForCapability(ctx context.Context, req *pb.DONForCapabilityRequest) (*pb.DONForCapabilityReply, error) {
+	don, nodes, err := c.impl.DONForCapability(ctx, req.CapabilityID)
+	if err != nil {
+		return nil, err
+	}
+
+	var nodesPb []*pb.NodeReply
+	for _, n := range nodes {
+		nodesPb = append(nodesPb, c.nodeReplyFromNode(n))
+	}
+
+	return &pb.DONForCapabilityReply{
+		Don:   toPbDON(don),
+		Nodes: nodesPb,
+	}, nil
+}
+
 func (c *capabilitiesRegistryServer) nodeReplyFromNode(node capabilities.Node) *pb.NodeReply {
 	workflowDONpb := toPbDON(node.WorkflowDON)
 

pkg/loop/internal/core/services/capability/capabilities_registry_test.go

@@ -496,6 +496,70 @@ func TestCapabilitiesRegistry_ConfigForCapability_RemoteExecutableConfig(t *test
 	assert.Equal(t, 2*time.Minute, capConf.RemoteExecutableConfig.RegistrationExpiry)
 }
 
+func TestCapabilitiesRegistry_DONForCapability(t *testing.T) {
+	stopCh := make(chan struct{})
+	logger := logger.Test(t)
+	reg := mocks.NewCapabilitiesRegistry(t)
+
+	pluginName := "registry-test"
+	client, server := plugin.TestPluginGRPCConn(
+		t,
+		true,
+		map[string]plugin.Plugin{
+			pluginName: &testRegistryPlugin{
+				impl: reg,
+				brokerExt: &net.BrokerExt{
+					BrokerConfig: net.BrokerConfig{
+						StopCh: stopCh,
+						Logger: logger,
+					},
+				},
+			},
+		},
+	)
+
+	defer client.Close()
+	defer server.Stop()
+
+	regClient, err := client.Dispense(pluginName)
+	require.NoError(t, err)
+
+	rc, ok := regClient.(*capabilitiesRegistryClient)
+	require.True(t, ok)
+
+	capID := "[email protected]"
+
+	donID := uint32(1)
+	expectedDON := capabilities.DON{
+		ID: donID,
+		F:  1,
+		Members: []p2ptypes.PeerID{
+			[32]byte{0: 1},
+			[32]byte{0: 2},
+		},
+	}
+	expectedNodes := []capabilities.Node{
+		{
+			PeerID:              &p2ptypes.PeerID{0: 1},
+			NodeOperatorID:      1,
+			EncryptionPublicKey: [32]byte{0: 1},
+			CapabilityDONs:      []capabilities.DON{},
+		},
+		{
+			PeerID:              &p2ptypes.PeerID{0: 2},
+			NodeOperatorID:      2,
+			EncryptionPublicKey: [32]byte{0: 2},
+			CapabilityDONs:      []capabilities.DON{},
+		},
+	}
+	reg.On("DONForCapability", mock.Anything, capID).Once().Return(expectedDON, expectedNodes, nil)
+
+	don, nodes, err := rc.DONForCapability(t.Context(), capID)
+	require.NoError(t, err)
+	assert.Equal(t, expectedDON, don)
+	assert.Equal(t, expectedNodes, nodes)
+}
+
 func ensureEqual(t *testing.T, expectedNode, actualNode capabilities.Node) {
 	// check local node struct
 	require.Equal(t, expectedNode.PeerID, actualNode.PeerID)

pkg/loop/internal/pb/capabilities_registry.pb.go

@@ -11,6 +11,7 @@ service CapabilitiesRegistry {
   rpc LocalNode (google.protobuf.Empty) returns (NodeReply) {}
   rpc NodeByPeerID(NodeRequest) returns (NodeReply) {}
   rpc ConfigForCapability (ConfigForCapabilityRequest) returns (ConfigForCapabilityReply) {}
+  rpc DONForCapability (DONForCapabilityRequest) returns (DONForCapabilityReply) {}
   rpc Get (GetRequest) returns (GetReply) {}
   rpc GetTrigger (GetTriggerRequest) returns (GetTriggerReply) {}
   rpc GetExecutable (GetExecutableRequest) returns (GetExecutableReply) {}
@@ -112,3 +113,12 @@ message ConfigForCapabilityRequest {
 message ConfigForCapabilityReply {
   loop.CapabilityConfig capability_config = 1;
 }
+
+message DONForCapabilityRequest {
+  string capabilityID = 1;
+}
+
+message DONForCapabilityReply {
+  DON don = 1;
+  repeated NodeReply nodes = 2;
+}