test(evmengine/types): add test cases for upgrade contract

client/x/evmengine/types/upgrade_contract_test.go

@@ -0,0 +1,84 @@
+//nolint:testpackage // Ignore this linter rule because we want to test private method, so package name should be same with the target package name.
+package types
+
+import (
+	"testing"
+
+	"github.com/ethereum/go-ethereum/accounts/abi/bind"
+	"github.com/stretchr/testify/require"
+
+	"github.com/piplabs/story/contracts/bindings"
+)
+
+func Test_mustGetABI(t *testing.T) {
+	t.Parallel()
+	testCases := []struct {
+		name        string
+		input       *bind.MetaData
+		expectPanic bool
+	}{
+		{
+			name:        "No Panic with valid metadata",
+			input:       bindings.UpgradeEntrypointMetaData,
+			expectPanic: false,
+		},
+		{
+			name:        "Panics with invalid metadata",
+			input:       &bind.MetaData{ABI: "invalid ABI"},
+			expectPanic: true,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			t.Parallel()
+
+			if tc.expectPanic {
+				require.Panics(t, func() {
+					mustGetABI(tc.input)
+				})
+			} else {
+				require.NotPanics(t, func() {
+					mustGetABI(tc.input)
+				})
+			}
+		})
+	}
+}
+
+func Test_mustGetEvent(t *testing.T) {
+	t.Parallel()
+	abi := mustGetABI(bindings.UpgradeEntrypointMetaData)
+	testCases := []struct {
+		name        string
+		eventName   string
+		expectPanic bool
+	}{
+		{
+			name:        "No Panic for valid event SoftwareUpgrade",
+			eventName:   "SoftwareUpgrade",
+			expectPanic: false,
+		},
+		{
+			name:        "Panics for non-existent event",
+			eventName:   "NonExistentEvent",
+			expectPanic: true,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			t.Parallel()
+
+			if tc.expectPanic {
+				require.Panics(t, func() {
+					mustGetEvent(abi, tc.eventName)
+				})
+			} else {
+				require.NotPanics(t, func() {
+					mustGetEvent(abi, tc.eventName)
+				})
+			}
+		})
+	}
+}