chore: adds unit test on GetContractSet

deployment/keystone/changeset/internal/state_test.go

@@ -0,0 +1,137 @@
+package internal
+
+import (
+	"testing"
+
+	"github.com/ethereum/go-ethereum/common"
+	chainsel "github.com/smartcontractkit/chain-selectors"
+	"github.com/smartcontractkit/chainlink-common/pkg/logger"
+	"github.com/smartcontractkit/chainlink/deployment"
+	commonchangeset "github.com/smartcontractkit/chainlink/deployment/common/changeset"
+	"github.com/smartcontractkit/chainlink/deployment/common/proposalutils"
+	kcr "github.com/smartcontractkit/chainlink/v2/core/gethwrappers/keystone/generated/capabilities_registry_1_1_0"
+
+	"github.com/stretchr/testify/require"
+)
+
+func Test_GetContractSet(t *testing.T) {
+	type testCase struct {
+		name        string
+		giveRequest *GetContractSetsRequest
+		wantResp    *GetContractSetsResponse
+		wantErr     string
+	}
+	lggr := logger.Test(t)
+	chain := chainsel.ETHEREUM_TESTNET_SEPOLIA
+
+	tt := []testCase{
+		{
+			name: "OK_picks unlabeled contracts by default",
+			giveRequest: func() *GetContractSetsRequest {
+				giveAB := map[uint64]map[string]deployment.TypeAndVersion{
+					chain.Selector: {
+						"0xabc": deployment.NewTypeAndVersion(
+							deployment.ContractType("CapabilitiesRegistry"),
+							deployment.Version1_0_0,
+						),
+						"0x123": func() deployment.TypeAndVersion {
+							tv := deployment.NewTypeAndVersion(
+								deployment.ContractType("CapabilitiesRegistry"),
+								deployment.Version1_0_0,
+							)
+							tv.Labels.Add("SA")
+							return tv
+						}(),
+					},
+				}
+				ab := deployment.NewMemoryAddressBookFromMap(
+					giveAB,
+				)
+				req := &GetContractSetsRequest{
+					Chains: map[uint64]deployment.Chain{
+						chain.Selector: {
+							Selector: chain.Selector,
+						},
+					},
+					AddressBook: ab,
+				}
+				return req
+			}(),
+			wantResp: &GetContractSetsResponse{
+				ContractSets: map[uint64]ContractSet{
+					chain.Selector: {
+						MCMSWithTimelockState: commonchangeset.MCMSWithTimelockState{
+							MCMSWithTimelockContracts: &proposalutils.MCMSWithTimelockContracts{},
+						},
+						CapabilitiesRegistry: func() *kcr.CapabilitiesRegistry {
+							cr, err := kcr.NewCapabilitiesRegistry(common.HexToAddress("0xabc"), nil)
+							require.NoError(t, err)
+							return cr
+						}(),
+					},
+				},
+			},
+		},
+		{
+			name: "OK_resolves labeled contracts",
+			giveRequest: func() *GetContractSetsRequest {
+				giveAB := map[uint64]map[string]deployment.TypeAndVersion{
+					chain.Selector: {
+						"0xabc": deployment.NewTypeAndVersion(
+							deployment.ContractType("CapabilitiesRegistry"),
+							deployment.Version1_0_0,
+						),
+						"0x123": func() deployment.TypeAndVersion {
+							tv := deployment.NewTypeAndVersion(
+								deployment.ContractType("CapabilitiesRegistry"),
+								deployment.Version1_0_0,
+							)
+							tv.Labels.Add("SA")
+							return tv
+						}(),
+					},
+				}
+				ab := deployment.NewMemoryAddressBookFromMap(
+					giveAB,
+				)
+				req := &GetContractSetsRequest{
+					Chains: map[uint64]deployment.Chain{
+						chain.Selector: {
+							Selector: chain.Selector,
+						},
+					},
+					Labels:      []string{"SA"},
+					AddressBook: ab,
+				}
+				return req
+			}(),
+			wantResp: &GetContractSetsResponse{
+				ContractSets: map[uint64]ContractSet{
+					chain.Selector: {
+						MCMSWithTimelockState: commonchangeset.MCMSWithTimelockState{
+							MCMSWithTimelockContracts: &proposalutils.MCMSWithTimelockContracts{},
+						},
+						CapabilitiesRegistry: func() *kcr.CapabilitiesRegistry {
+							cr, err := kcr.NewCapabilitiesRegistry(common.HexToAddress("0x123"), nil)
+							require.NoError(t, err)
+							return cr
+						}(),
+					},
+				},
+			},
+		},
+	}
+
+	for _, tc := range tt {
+		t.Run(tc.name, func(t *testing.T) {
+			gotResp, err := GetContractSets(lggr, tc.giveRequest)
+			if tc.wantErr != "" {
+				require.Error(t, err)
+				require.ErrorContains(t, err, tc.wantErr)
+				return
+			}
+
+			require.Equal(t, tc.wantResp, gotResp)
+		})
+	}
+}