diff --git a/pkg/sdk/parameters.go b/pkg/sdk/parameters.go index aea226a018..87256c44db 100644 --- a/pkg/sdk/parameters.go +++ b/pkg/sdk/parameters.go @@ -1186,7 +1186,7 @@ type AccountParameters struct { EnableTriSecretAndRekeyOptOutForImageRepository *bool `ddl:"parameter" sql:"ENABLE_TRI_SECRET_AND_REKEY_OPT_OUT_FOR_IMAGE_REPOSITORY"` EnableTriSecretAndRekeyOptOutForSpcsBlockStorage *bool `ddl:"parameter" sql:"ENABLE_TRI_SECRET_AND_REKEY_OPT_OUT_FOR_SPCS_BLOCK_STORAGE"` EnableUnhandledExceptionsReporting *bool `ddl:"parameter" sql:"ENABLE_UNHANDLED_EXCEPTIONS_REPORTING"` - EnforceNetworkRulesForInternalStages *bool `ddl:"keyword" sql:"ENFORCE_NETWORK_RULES_FOR_INTERNAL_STAGES"` + EnforceNetworkRulesForInternalStages *bool `ddl:"parameter" sql:"ENFORCE_NETWORK_RULES_FOR_INTERNAL_STAGES"` EventTable *string `ddl:"parameter,single_quotes" sql:"EVENT_TABLE"` ExternalOAuthAddPrivilegedRolesToBlockedList *bool `ddl:"parameter" sql:"EXTERNAL_OAUTH_ADD_PRIVILEGED_ROLES_TO_BLOCKED_LIST"` // InitialReplicationSizeLimitInTB is a string because values like 3.0 get rounded to 3, resulting in an error in Snowflake. diff --git a/pkg/sdk/parameters_test.go b/pkg/sdk/parameters_test.go index 93592e1f1e..50418a6bb9 100644 --- a/pkg/sdk/parameters_test.go +++ b/pkg/sdk/parameters_test.go @@ -54,6 +54,21 @@ func TestUnSetObjectParameterNetworkPolicyOnUser(t *testing.T) { }) } +func TestSetAccountParameterEnforceNetworkRulesForInternalStages(t *testing.T) { + opts := &AlterAccountOptions{ + Set: &AccountSet{ + Parameters: &AccountLevelParameters{ + AccountParameters: &AccountParameters{ + EnforceNetworkRulesForInternalStages: Bool(true), + }, + }, + }, + } + t.Run("Set Enforce Network Rules for Internal Stages", func(t *testing.T) { + assertOptsValidAndSQLEquals(t, opts, "ALTER ACCOUNT SET ENFORCE_NETWORK_RULES_FOR_INTERNAL_STAGES = true") + }) +} + func TestToAccountParameter(t *testing.T) { type test struct { input string