nil check contract in view generator

smartcontractkit · Sep 23, 2024 · 26b5860 · 26b5860
1 parent 4c3613c
commit 26b5860
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/integration-tests/deployment/ccip/view/v1_2/commit_store.go b/integration-tests/deployment/ccip/view/v1_2/commit_store.go
@@ -24,6 +24,9 @@ type CommitStoreView struct {
 }
 
 func GenerateCommitStoreView(c *commit_store.CommitStore) (CommitStoreView, error) {
+	if c == nil {
+		return CommitStoreView{}, fmt.Errorf("cannot generate view for nil CommitStore")
+	}
 	meta, err := types.NewContractMetaData(c, c.Address())
 	if err != nil {
 		return CommitStoreView{}, fmt.Errorf("failed to generate contract metadata for CommitStore: %w", err)

diff --git a/integration-tests/deployment/ccip/view/v1_6/rmn_proxy_contract.go b/integration-tests/deployment/ccip/view/v1_6/rmn_proxy_contract.go
@@ -15,6 +15,9 @@ type RMNProxyView struct {
 }
 
 func GenerateRMNProxyView(r *rmn_proxy_contract.RMNProxyContract) (RMNProxyView, error) {
+	if r == nil {
+		return RMNProxyView{}, fmt.Errorf("cannot generate view for nil RMNProxy")
+	}
 	meta, err := types.NewContractMetaData(r, r.Address())
 	if err != nil {
 		return RMNProxyView{}, fmt.Errorf("failed to generate contract metadata for RMNProxy: %w", err)