From 8c49751c597cd1051d59338f898bbac63a78c87c Mon Sep 17 00:00:00 2001
From: Michael Street <5597260+MStreet3@users.noreply.github.com>
Date: Wed, 19 Feb 2025 17:43:15 -0500
Subject: [PATCH] adds mcms state test

---
 deployment/common/changeset/state_test.go | 44 +++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/deployment/common/changeset/state_test.go b/deployment/common/changeset/state_test.go
index 883079dcfcb..1537a772e1c 100644
--- a/deployment/common/changeset/state_test.go
+++ b/deployment/common/changeset/state_test.go
@@ -69,6 +69,50 @@ func TestMaybeLoadMCMSWithTimelockChainState(t *testing.T) {
 			},
 			wantErr: "",
 		},
+		{
+			name:  "OK_labelled multichain contract is selected if only option",
+			chain: defaultChain,
+			addresses: map[string]deployment.TypeAndVersion{
+				"0x123": deployment.NewTypeAndVersion(types.RBACTimelock, deployment.Version1_0_0),
+				"0x456": deployment.NewTypeAndVersion(types.CallProxy, deployment.Version1_0_0),
+				"0xabc": deployment.NewTypeAndVersion(types.CancellerManyChainMultisig, deployment.Version1_0_0),
+				"0xdef": deployment.NewTypeAndVersion(types.BypasserManyChainMultisig, deployment.Version1_0_0),
+				"0xaaa": func() deployment.TypeAndVersion {
+					tv := deployment.NewTypeAndVersion(types.ManyChainMultisig, deployment.Version1_0_0)
+					tv.Labels.Add(types.ProposerRole.String())
+					return tv
+				}(),
+			},
+			wantState: &MCMSWithTimelockState{
+				MCMSWithTimelockContracts: &proposalutils.MCMSWithTimelockContracts{
+					Timelock: func() *owner_helpers.RBACTimelock {
+						tl, err := owner_helpers.NewRBACTimelock(common.HexToAddress("0x123"), nil)
+						require.NoError(t, err)
+						return tl
+					}(),
+					CallProxy: func() *owner_helpers.CallProxy {
+						cp, err := owner_helpers.NewCallProxy(common.HexToAddress("0x456"), nil)
+						require.NoError(t, err)
+						return cp
+					}(),
+					CancellerMcm: func() *owner_helpers.ManyChainMultiSig {
+						mcms, err := owner_helpers.NewManyChainMultiSig(common.HexToAddress("0xabc"), nil)
+						require.NoError(t, err)
+						return mcms
+					}(),
+					BypasserMcm: func() *owner_helpers.ManyChainMultiSig {
+						mcms, err := owner_helpers.NewManyChainMultiSig(common.HexToAddress("0xdef"), nil)
+						require.NoError(t, err)
+						return mcms
+					}(),
+					ProposerMcm: func() *owner_helpers.ManyChainMultiSig {
+						mcms, err := owner_helpers.NewManyChainMultiSig(common.HexToAddress("0xaaa"), nil)
+						require.NoError(t, err)
+						return mcms
+					}(),
+				},
+			},
+		},
 		{
 			// A ManyChainMultiSig contract can be labeled or it can be typed.  A typed contract should be selected
 			// over a labeled contract.