From 324d025048c84550f48d458a8bf7c5744a59c5e0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jan=20Cie=C5=9Blak?= Date: Tue, 24 Sep 2024 14:13:41 +0200 Subject: [PATCH] Refactor task parameters --- .../gen/object_parameters_def.go | 76 +-- .../task_parameters_snowflake_gen.go | 140 +++- .../database_resource_parameters_gen.go | 12 +- .../gen/templates/definition.tmpl | 13 +- .../task_resource_parameters_gen.go | 622 ++++++++++++++++++ .../user_resource_parameters_gen.go | 12 +- .../warehouse_resource_parameters_gen.go | 12 +- pkg/sdk/parameters.go | 8 +- pkg/sdk/testint/tasks_gen_integration_test.go | 49 +- pkg/sdk/testint/users_integration_test.go | 1 + 10 files changed, 812 insertions(+), 133 deletions(-) create mode 100644 pkg/acceptance/bettertestspoc/assert/resourceparametersassert/task_resource_parameters_gen.go diff --git a/pkg/acceptance/bettertestspoc/assert/objectparametersassert/gen/object_parameters_def.go b/pkg/acceptance/bettertestspoc/assert/objectparametersassert/gen/object_parameters_def.go index a2141417ffa..e9d8b080920 100644 --- a/pkg/acceptance/bettertestspoc/assert/objectparametersassert/gen/object_parameters_def.go +++ b/pkg/acceptance/bettertestspoc/assert/objectparametersassert/gen/object_parameters_def.go @@ -133,106 +133,46 @@ var allObjectsParameters = []SnowflakeObjectParameters{ IdType: "sdk.SchemaObjectIdentifier", Level: sdk.ParameterTypeTask, Parameters: []SnowflakeParameter{ - // ABORT_DETACHED_QUERY - // ACTIVE_PYTHON_PROFILER + {ParameterName: string(sdk.TaskParameterSuspendTaskAfterNumFailures), ParameterType: "int", DefaultValue: "10", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, + {ParameterName: string(sdk.TaskParameterTaskAutoRetryAttempts), ParameterType: "int", DefaultValue: "0", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, + {ParameterName: string(sdk.TaskParameterUserTaskManagedInitialWarehouseSize), ParameterType: "sdk.WarehouseSize", DefaultValue: `sdk.WarehouseSize("Medium")`, DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, + {ParameterName: string(sdk.TaskParameterUserTaskMinimumTriggerIntervalInSeconds), ParameterType: "int", DefaultValue: "30", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, + {ParameterName: string(sdk.TaskParameterUserTaskTimeoutMs), ParameterType: "int", DefaultValue: "3600000", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, + {ParameterName: string(sdk.TaskParameterAbortDetachedQuery), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterAutocommit), ParameterType: "bool", DefaultValue: "true", DefaultLevel: "sdk.ParameterTypeAccount"}, - // AUTOCOMMIT_API_SUPPORTED {ParameterName: string(sdk.TaskParameterBinaryInputFormat), ParameterType: "sdk.BinaryInputFormat", DefaultValue: "sdk.BinaryInputFormatHex", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterBinaryOutputFormat), ParameterType: "sdk.BinaryOutputFormat", DefaultValue: "sdk.BinaryOutputFormatHex", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // CLIENT_ENABLE_CONSERVATIVE_MEMORY_USAGE - // CLIENT_ENABLE_DEFAULT_OVERWRITE_IN_PUT - // CLIENT_ENABLE_LOG_INFO_STATEMENT_PARAMETERS {ParameterName: string(sdk.TaskParameterClientMemoryLimit), ParameterType: "int", DefaultValue: "1536", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterClientMetadataRequestUseConnectionCtx), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // CLIENT_METADATA_USE_SESSION_DATABASE {ParameterName: string(sdk.TaskParameterClientPrefetchThreads), ParameterType: "int", DefaultValue: "4", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterClientResultChunkSize), ParameterType: "int", DefaultValue: "160", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterClientResultColumnCaseInsensitive), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // CLIENT_SESSION_CLONE {ParameterName: string(sdk.TaskParameterClientSessionKeepAlive), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterClientSessionKeepAliveHeartbeatFrequency), ParameterType: "int", DefaultValue: "3600", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterClientTimestampTypeMapping), ParameterType: "sdk.ClientTimestampTypeMapping", DefaultValue: "sdk.ClientTimestampTypeMappingLtz", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // CSV_TIMESTAMP_FORMAT - // C_API_QUERY_RESULT_FORMAT {ParameterName: string(sdk.TaskParameterDateInputFormat), ParameterType: "string", DefaultValue: "AUTO", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterDateOutputFormat), ParameterType: "string", DefaultValue: "YYYY-MM-DD", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // DOTNET_QUERY_RESULT_FORMAT - // DYNAMIC_TABLES_VIEW_VERSION - // DYNAMIC_TABLE_GRAPH_HISTORY_VIEW_VERSION - // DYNAMIC_TABLE_REFRESH_HISTORY_VIEW_VERSION - // ENABLE_CONSOLE_OUTPUT - // ENABLE_PROVIDER_LISTING_PROGRAMMATIC_ACCESS_DESCRIBE_LISTING - // ENABLE_SNOW_API_FOR_COMPUTE_POOL - // ENABLE_SNOW_API_FOR_COPILOT - // ENABLE_SNOW_API_FOR_DATABASE - // ENABLE_SNOW_API_FOR_FUNCTION - // ENABLE_SNOW_API_FOR_GRANT - // ENABLE_SNOW_API_FOR_ICEBERG - // ENABLE_SNOW_API_FOR_IMAGE_REPOSITORY - // ENABLE_SNOW_API_FOR_RESULT - // ENABLE_SNOW_API_FOR_ROLE - // ENABLE_SNOW_API_FOR_SCHEMA - // ENABLE_SNOW_API_FOR_SERVICE - // ENABLE_SNOW_API_FOR_SESSION - // ENABLE_SNOW_API_FOR_STAGE - // ENABLE_SNOW_API_FOR_TABLE - // ENABLE_SNOW_API_FOR_TASK - // ENABLE_SNOW_API_FOR_USER - // ENABLE_SNOW_API_FOR_WAREHOUSE {ParameterName: string(sdk.TaskParameterEnableUnloadPhysicalTypeOptimization), ParameterType: "bool", DefaultValue: "true", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterErrorOnNondeterministicMerge), ParameterType: "bool", DefaultValue: "true", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterErrorOnNondeterministicUpdate), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterGeographyOutputFormat), ParameterType: "sdk.GeographyOutputFormat", DefaultValue: "sdk.GeographyOutputFormatGeoJSON", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterGeometryOutputFormat), ParameterType: "sdk.GeometryOutputFormat", DefaultValue: "sdk.GeometryOutputFormatGeoJSON", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // GO_QUERY_RESULT_FORMAT - // HYBRID_TABLE_LOCK_TIMEOUT - // INCLUDE_DT_WITH_TABLE_KIND_IN_SHOW_OBJECTS - // INCLUDE_DYNAMIC_TABLES_WITH_TABLE_KIND - // JDBC_FORMAT_DATE_WITH_TIMEZONE - // JDBC_QUERY_RESULT_FORMAT {ParameterName: string(sdk.TaskParameterJdbcTreatTimestampNtzAsUtc), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterJdbcUseSessionTimezone), ParameterType: "bool", DefaultValue: "true", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterJsonIndent), ParameterType: "int", DefaultValue: "2", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // LANGUAGE {ParameterName: string(sdk.TaskParameterLockTimeout), ParameterType: "int", DefaultValue: "43200", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterLogLevel), ParameterType: "sdk.LogLevel", DefaultValue: "sdk.LogLevelOff", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // METRIC_LEVEL {ParameterName: string(sdk.TaskParameterMultiStatementCount), ParameterType: "int", DefaultValue: "1", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterNoorderSequenceAsDefault), ParameterType: "bool", DefaultValue: "true", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // ODBC_QUERY_RESULT_FORMAT {ParameterName: string(sdk.TaskParameterOdbcTreatDecimalAsInt), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // PYTHON_CONNECTOR_QUERY_RESULT_FORMAT - // PYTHON_CONNECTOR_USE_NANOARROW - // PYTHON_PROFILER_MODULES - // PYTHON_PROFILER_TARGET_STAGE - // PYTHON_SNOWPARK_AUTO_CLEAN_UP_TEMP_TABLE_ENABLED - // PYTHON_SNOWPARK_COMPILATION_STAGE_ENABLED - // PYTHON_SNOWPARK_ELIMINATE_NUMERIC_SQL_VALUE_CAST_ENABLED - // PYTHON_SNOWPARK_USE_CTE_OPTIMIZATION - // PYTHON_SNOWPARK_USE_LARGE_QUERY_BREAKDOWN_OPTIMIZATION - // PYTHON_SNOWPARK_USE_LOGICAL_TYPE_FOR_CREATE_DATAFRAME - // PYTHON_SNOWPARK_USE_SCOPED_TEMP_OBJECTS - // PYTHON_SNOWPARK_USE_SQL_SIMPLIFIER - // QA_TEST_NAME - // QUERY_RESULT_FORMAT {ParameterName: string(sdk.TaskParameterQueryTag), ParameterType: "string", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterQuotedIdentifiersIgnoreCase), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeAccount"}, {ParameterName: string(sdk.TaskParameterRowsPerResultset), ParameterType: "int", DefaultValue: "0", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterS3StageVpceDnsName), ParameterType: "string", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterSearchPath), ParameterType: "string", DefaultValue: "$current, $public", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // SHOW_EXTERNAL_TABLE_KIND_AS_TABLE - // SNOWPARK_HIDE_INTERNAL_ALIAS - // SNOWPARK_LAZY_ANALYSIS - // SNOWPARK_REQUEST_TIMEOUT_IN_SECONDS - // SNOWPARK_STORED_PROC_IS_FINAL_TABLE_QUERY - // SNOWPARK_USE_SCOPED_TEMP_OBJECTS - // SQL_API_NULLABLE_IN_RESULT_SET - // SQL_API_QUERY_RESULT_FORMAT {ParameterName: string(sdk.TaskParameterStatementQueuedTimeoutInSeconds), ParameterType: "int", DefaultValue: "0", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterStatementTimeoutInSeconds), ParameterType: "int", DefaultValue: "172800", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterStrictJsonOutput), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // TODO: SUSPEND_TASK_AFTER_NUM_FAILURES - // TODO: TASK_AUTO_RETRY_ATTEMPTS {ParameterName: string(sdk.TaskParameterTimestampDayIsAlways24h), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterTimestampInputFormat), ParameterType: "string", DefaultValue: "AUTO", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterTimestampLtzOutputFormat), ParameterType: "string", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, @@ -247,12 +187,8 @@ var allObjectsParameters = []SnowflakeObjectParameters{ {ParameterName: string(sdk.TaskParameterTransactionAbortOnError), ParameterType: "bool", DefaultValue: "false", DefaultLevel: "sdk.ParameterTypeAccount"}, {ParameterName: string(sdk.TaskParameterTransactionDefaultIsolationLevel), ParameterType: "sdk.TransactionDefaultIsolationLevel", DefaultValue: "sdk.TransactionDefaultIsolationLevelReadCommitted", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterTwoDigitCenturyStart), ParameterType: "int", DefaultValue: "1970", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // TODO UI_QUERY_RESULT_FORMAT // TODO [SNOW-1501905]: quick workaround for now: lowercase for ignore in snowflake by default but uppercase for FAIL {ParameterName: string(sdk.TaskParameterUnsupportedDdlAction), ParameterType: "sdk.UnsupportedDDLAction", DefaultValue: "sdk.UnsupportedDDLAction(strings.ToLower(string(sdk.UnsupportedDDLActionIgnore)))", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - {ParameterName: string(sdk.TaskParameterUserTaskManagedInitialWarehouseSize), ParameterType: "sdk.WarehouseSize", DefaultValue: "Medium", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // {ParameterName: string(sdk.UserParameterUnsupportedUserTaskMinimumTriggerIntervalInSeconds), ParameterType: "int", DefaultValue: "30", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, - // {ParameterName: string(sdk.UserParameterUnsupportedUserTaskTimeoutMs), ParameterType: "int", DefaultValue: "3600000", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterUseCachedResult), ParameterType: "bool", DefaultValue: "true", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterWeekOfYearPolicy), ParameterType: "int", DefaultValue: "0", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, {ParameterName: string(sdk.TaskParameterWeekStart), ParameterType: "int", DefaultValue: "0", DefaultLevel: "sdk.ParameterTypeSnowflakeDefault"}, diff --git a/pkg/acceptance/bettertestspoc/assert/objectparametersassert/task_parameters_snowflake_gen.go b/pkg/acceptance/bettertestspoc/assert/objectparametersassert/task_parameters_snowflake_gen.go index 8feaea52b19..24ac1f78bd5 100644 --- a/pkg/acceptance/bettertestspoc/assert/objectparametersassert/task_parameters_snowflake_gen.go +++ b/pkg/acceptance/bettertestspoc/assert/objectparametersassert/task_parameters_snowflake_gen.go @@ -68,6 +68,12 @@ func (t *TaskParametersAssert) HasDefaultParameterValueOnLevel(parameterName sdk // - have an expected level func (t *TaskParametersAssert) HasAllDefaults() *TaskParametersAssert { return t. + HasDefaultParameterValueOnLevel(sdk.TaskParameterSuspendTaskAfterNumFailures, sdk.ParameterTypeSnowflakeDefault). + HasDefaultParameterValueOnLevel(sdk.TaskParameterTaskAutoRetryAttempts, sdk.ParameterTypeSnowflakeDefault). + HasDefaultParameterValueOnLevel(sdk.TaskParameterUserTaskManagedInitialWarehouseSize, sdk.ParameterTypeSnowflakeDefault). + HasDefaultParameterValueOnLevel(sdk.TaskParameterUserTaskMinimumTriggerIntervalInSeconds, sdk.ParameterTypeSnowflakeDefault). + HasDefaultParameterValueOnLevel(sdk.TaskParameterUserTaskTimeoutMs, sdk.ParameterTypeSnowflakeDefault). + HasDefaultParameterValueOnLevel(sdk.TaskParameterAbortDetachedQuery, sdk.ParameterTypeSnowflakeDefault). HasDefaultParameterValueOnLevel(sdk.TaskParameterAutocommit, sdk.ParameterTypeAccount). HasDefaultParameterValueOnLevel(sdk.TaskParameterBinaryInputFormat, sdk.ParameterTypeSnowflakeDefault). HasDefaultParameterValueOnLevel(sdk.TaskParameterBinaryOutputFormat, sdk.ParameterTypeSnowflakeDefault). @@ -117,7 +123,6 @@ func (t *TaskParametersAssert) HasAllDefaults() *TaskParametersAssert { HasDefaultParameterValueOnLevel(sdk.TaskParameterTransactionDefaultIsolationLevel, sdk.ParameterTypeSnowflakeDefault). HasDefaultParameterValueOnLevel(sdk.TaskParameterTwoDigitCenturyStart, sdk.ParameterTypeSnowflakeDefault). HasDefaultParameterValueOnLevel(sdk.TaskParameterUnsupportedDdlAction, sdk.ParameterTypeSnowflakeDefault). - HasDefaultParameterValueOnLevel(sdk.TaskParameterUserTaskManagedInitialWarehouseSize, sdk.ParameterTypeSnowflakeDefault). HasDefaultParameterValueOnLevel(sdk.TaskParameterUseCachedResult, sdk.ParameterTypeSnowflakeDefault). HasDefaultParameterValueOnLevel(sdk.TaskParameterWeekOfYearPolicy, sdk.ParameterTypeSnowflakeDefault). HasDefaultParameterValueOnLevel(sdk.TaskParameterWeekStart, sdk.ParameterTypeSnowflakeDefault) @@ -125,6 +130,12 @@ func (t *TaskParametersAssert) HasAllDefaults() *TaskParametersAssert { func (t *TaskParametersAssert) HasAllDefaultsExplicit() *TaskParametersAssert { return t. + HasDefaultSuspendTaskAfterNumFailuresValueExplicit(). + HasDefaultTaskAutoRetryAttemptsValueExplicit(). + HasDefaultUserTaskManagedInitialWarehouseSizeValueExplicit(). + HasDefaultUserTaskMinimumTriggerIntervalInSecondsValueExplicit(). + HasDefaultUserTaskTimeoutMsValueExplicit(). + HasDefaultAbortDetachedQueryValueExplicit(). HasDefaultAutocommitValueExplicit(). HasDefaultBinaryInputFormatValueExplicit(). HasDefaultBinaryOutputFormatValueExplicit(). @@ -174,7 +185,6 @@ func (t *TaskParametersAssert) HasAllDefaultsExplicit() *TaskParametersAssert { HasDefaultTransactionDefaultIsolationLevelValueExplicit(). HasDefaultTwoDigitCenturyStartValueExplicit(). HasDefaultUnsupportedDdlActionValueExplicit(). - HasDefaultUserTaskManagedInitialWarehouseSizeValueExplicit(). HasDefaultUseCachedResultValueExplicit(). HasDefaultWeekOfYearPolicyValueExplicit(). HasDefaultWeekStartValueExplicit() @@ -184,6 +194,36 @@ func (t *TaskParametersAssert) HasAllDefaultsExplicit() *TaskParametersAssert { // Parameter value checks // //////////////////////////// +func (t *TaskParametersAssert) HasSuspendTaskAfterNumFailures(expected int) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterIntValueSet(sdk.TaskParameterSuspendTaskAfterNumFailures, expected)) + return t +} + +func (t *TaskParametersAssert) HasTaskAutoRetryAttempts(expected int) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterIntValueSet(sdk.TaskParameterTaskAutoRetryAttempts, expected)) + return t +} + +func (t *TaskParametersAssert) HasUserTaskManagedInitialWarehouseSize(expected sdk.WarehouseSize) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterStringUnderlyingValueSet(sdk.TaskParameterUserTaskManagedInitialWarehouseSize, expected)) + return t +} + +func (t *TaskParametersAssert) HasUserTaskMinimumTriggerIntervalInSeconds(expected int) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterIntValueSet(sdk.TaskParameterUserTaskMinimumTriggerIntervalInSeconds, expected)) + return t +} + +func (t *TaskParametersAssert) HasUserTaskTimeoutMs(expected int) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterIntValueSet(sdk.TaskParameterUserTaskTimeoutMs, expected)) + return t +} + +func (t *TaskParametersAssert) HasAbortDetachedQuery(expected bool) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterBoolValueSet(sdk.TaskParameterAbortDetachedQuery, expected)) + return t +} + func (t *TaskParametersAssert) HasAutocommit(expected bool) *TaskParametersAssert { t.AddAssertion(assert.SnowflakeParameterBoolValueSet(sdk.TaskParameterAutocommit, expected)) return t @@ -429,11 +469,6 @@ func (t *TaskParametersAssert) HasUnsupportedDdlAction(expected sdk.UnsupportedD return t } -func (t *TaskParametersAssert) HasUserTaskManagedInitialWarehouseSize(expected sdk.WarehouseSize) *TaskParametersAssert { - t.AddAssertion(assert.SnowflakeParameterStringUnderlyingValueSet(sdk.TaskParameterUserTaskManagedInitialWarehouseSize, expected)) - return t -} - func (t *TaskParametersAssert) HasUseCachedResult(expected bool) *TaskParametersAssert { t.AddAssertion(assert.SnowflakeParameterBoolValueSet(sdk.TaskParameterUseCachedResult, expected)) return t @@ -453,6 +488,36 @@ func (t *TaskParametersAssert) HasWeekStart(expected int) *TaskParametersAssert // Parameter level checks // //////////////////////////// +func (t *TaskParametersAssert) HasSuspendTaskAfterNumFailuresLevel(expected sdk.ParameterType) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterLevelSet(sdk.TaskParameterSuspendTaskAfterNumFailures, expected)) + return t +} + +func (t *TaskParametersAssert) HasTaskAutoRetryAttemptsLevel(expected sdk.ParameterType) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterLevelSet(sdk.TaskParameterTaskAutoRetryAttempts, expected)) + return t +} + +func (t *TaskParametersAssert) HasUserTaskManagedInitialWarehouseSizeLevel(expected sdk.ParameterType) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterLevelSet(sdk.TaskParameterUserTaskManagedInitialWarehouseSize, expected)) + return t +} + +func (t *TaskParametersAssert) HasUserTaskMinimumTriggerIntervalInSecondsLevel(expected sdk.ParameterType) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterLevelSet(sdk.TaskParameterUserTaskMinimumTriggerIntervalInSeconds, expected)) + return t +} + +func (t *TaskParametersAssert) HasUserTaskTimeoutMsLevel(expected sdk.ParameterType) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterLevelSet(sdk.TaskParameterUserTaskTimeoutMs, expected)) + return t +} + +func (t *TaskParametersAssert) HasAbortDetachedQueryLevel(expected sdk.ParameterType) *TaskParametersAssert { + t.AddAssertion(assert.SnowflakeParameterLevelSet(sdk.TaskParameterAbortDetachedQuery, expected)) + return t +} + func (t *TaskParametersAssert) HasAutocommitLevel(expected sdk.ParameterType) *TaskParametersAssert { t.AddAssertion(assert.SnowflakeParameterLevelSet(sdk.TaskParameterAutocommit, expected)) return t @@ -698,11 +763,6 @@ func (t *TaskParametersAssert) HasUnsupportedDdlActionLevel(expected sdk.Paramet return t } -func (t *TaskParametersAssert) HasUserTaskManagedInitialWarehouseSizeLevel(expected sdk.ParameterType) *TaskParametersAssert { - t.AddAssertion(assert.SnowflakeParameterLevelSet(sdk.TaskParameterUserTaskManagedInitialWarehouseSize, expected)) - return t -} - func (t *TaskParametersAssert) HasUseCachedResultLevel(expected sdk.ParameterType) *TaskParametersAssert { t.AddAssertion(assert.SnowflakeParameterLevelSet(sdk.TaskParameterUseCachedResult, expected)) return t @@ -722,6 +782,30 @@ func (t *TaskParametersAssert) HasWeekStartLevel(expected sdk.ParameterType) *Ta // Parameter default value checks // //////////////////////////////////// +func (t *TaskParametersAssert) HasDefaultSuspendTaskAfterNumFailuresValue() *TaskParametersAssert { + return t.HasDefaultParameterValue(sdk.TaskParameterSuspendTaskAfterNumFailures) +} + +func (t *TaskParametersAssert) HasDefaultTaskAutoRetryAttemptsValue() *TaskParametersAssert { + return t.HasDefaultParameterValue(sdk.TaskParameterTaskAutoRetryAttempts) +} + +func (t *TaskParametersAssert) HasDefaultUserTaskManagedInitialWarehouseSizeValue() *TaskParametersAssert { + return t.HasDefaultParameterValue(sdk.TaskParameterUserTaskManagedInitialWarehouseSize) +} + +func (t *TaskParametersAssert) HasDefaultUserTaskMinimumTriggerIntervalInSecondsValue() *TaskParametersAssert { + return t.HasDefaultParameterValue(sdk.TaskParameterUserTaskMinimumTriggerIntervalInSeconds) +} + +func (t *TaskParametersAssert) HasDefaultUserTaskTimeoutMsValue() *TaskParametersAssert { + return t.HasDefaultParameterValue(sdk.TaskParameterUserTaskTimeoutMs) +} + +func (t *TaskParametersAssert) HasDefaultAbortDetachedQueryValue() *TaskParametersAssert { + return t.HasDefaultParameterValue(sdk.TaskParameterAbortDetachedQuery) +} + func (t *TaskParametersAssert) HasDefaultAutocommitValue() *TaskParametersAssert { return t.HasDefaultParameterValue(sdk.TaskParameterAutocommit) } @@ -918,10 +1002,6 @@ func (t *TaskParametersAssert) HasDefaultUnsupportedDdlActionValue() *TaskParame return t.HasDefaultParameterValue(sdk.TaskParameterUnsupportedDdlAction) } -func (t *TaskParametersAssert) HasDefaultUserTaskManagedInitialWarehouseSizeValue() *TaskParametersAssert { - return t.HasDefaultParameterValue(sdk.TaskParameterUserTaskManagedInitialWarehouseSize) -} - func (t *TaskParametersAssert) HasDefaultUseCachedResultValue() *TaskParametersAssert { return t.HasDefaultParameterValue(sdk.TaskParameterUseCachedResult) } @@ -938,6 +1018,30 @@ func (t *TaskParametersAssert) HasDefaultWeekStartValue() *TaskParametersAssert // Parameter explicit default value checks // ///////////////////////////////////////////// +func (t *TaskParametersAssert) HasDefaultSuspendTaskAfterNumFailuresValueExplicit() *TaskParametersAssert { + return t.HasSuspendTaskAfterNumFailures(10) +} + +func (t *TaskParametersAssert) HasDefaultTaskAutoRetryAttemptsValueExplicit() *TaskParametersAssert { + return t.HasTaskAutoRetryAttempts(0) +} + +func (t *TaskParametersAssert) HasDefaultUserTaskManagedInitialWarehouseSizeValueExplicit() *TaskParametersAssert { + return t.HasUserTaskManagedInitialWarehouseSize("Medium") +} + +func (t *TaskParametersAssert) HasDefaultUserTaskMinimumTriggerIntervalInSecondsValueExplicit() *TaskParametersAssert { + return t.HasUserTaskMinimumTriggerIntervalInSeconds(30) +} + +func (t *TaskParametersAssert) HasDefaultUserTaskTimeoutMsValueExplicit() *TaskParametersAssert { + return t.HasUserTaskTimeoutMs(3600000) +} + +func (t *TaskParametersAssert) HasDefaultAbortDetachedQueryValueExplicit() *TaskParametersAssert { + return t.HasAbortDetachedQuery(false) +} + func (t *TaskParametersAssert) HasDefaultAutocommitValueExplicit() *TaskParametersAssert { return t.HasAutocommit(true) } @@ -1134,10 +1238,6 @@ func (t *TaskParametersAssert) HasDefaultUnsupportedDdlActionValueExplicit() *Ta return t.HasUnsupportedDdlAction(sdk.UnsupportedDDLAction(strings.ToLower(string(sdk.UnsupportedDDLActionIgnore)))) } -func (t *TaskParametersAssert) HasDefaultUserTaskManagedInitialWarehouseSizeValueExplicit() *TaskParametersAssert { - return t.HasUserTaskManagedInitialWarehouseSize("Medium") -} - func (t *TaskParametersAssert) HasDefaultUseCachedResultValueExplicit() *TaskParametersAssert { return t.HasUseCachedResult(true) } diff --git a/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/database_resource_parameters_gen.go b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/database_resource_parameters_gen.go index 97cf0ea7e87..652b02de44f 100644 --- a/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/database_resource_parameters_gen.go +++ b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/database_resource_parameters_gen.go @@ -16,21 +16,21 @@ type DatabaseResourceParametersAssert struct { func DatabaseResourceParameters(t *testing.T, name string) *DatabaseResourceParametersAssert { t.Helper() - d := DatabaseResourceParametersAssert{ + resourceParameterAssert := DatabaseResourceParametersAssert{ ResourceAssert: assert.NewResourceAssert(name, "parameters"), } - d.AddAssertion(assert.ValueSet("parameters.#", "1")) - return &d + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert } func ImportedDatabaseResourceParameters(t *testing.T, id string) *DatabaseResourceParametersAssert { t.Helper() - d := DatabaseResourceParametersAssert{ + resourceParameterAssert := DatabaseResourceParametersAssert{ ResourceAssert: assert.NewImportedResourceAssert(id, "imported parameters"), } - d.AddAssertion(assert.ValueSet("parameters.#", "1")) - return &d + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert } //////////////////////////// diff --git a/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/gen/templates/definition.tmpl b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/gen/templates/definition.tmpl index a6c550eab65..2db132d83bd 100644 --- a/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/gen/templates/definition.tmpl +++ b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/gen/templates/definition.tmpl @@ -2,7 +2,6 @@ {{- $assertName := .Name | printf "%sResourceParametersAssert" -}} {{- $nameLowerCase := FirstLetterLowercase .Name -}} -{{- $assertVar := FirstLetter $nameLowerCase }} type {{ $assertName }} struct { *assert.ResourceAssert } @@ -10,19 +9,19 @@ type {{ $assertName }} struct { func {{ .Name }}ResourceParameters(t *testing.T, name string) *{{ $assertName }} { t.Helper() - {{ $assertVar }} := {{ $assertName }}{ + resourceParameterAssert := {{ $assertName }}{ ResourceAssert: assert.NewResourceAssert(name, "parameters"), } - {{ $assertVar }}.AddAssertion(assert.ValueSet("parameters.#", "1")) - return &{{ $assertVar }} + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert } func Imported{{ .Name }}ResourceParameters(t *testing.T, id string) *{{ $assertName }} { t.Helper() - {{ $assertVar }} := {{ $assertName }}{ + resourceParameterAssert := {{ $assertName }}{ ResourceAssert: assert.NewImportedResourceAssert(id, "imported parameters"), } - {{ $assertVar }}.AddAssertion(assert.ValueSet("parameters.#", "1")) - return &{{ $assertVar }} + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert } diff --git a/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/task_resource_parameters_gen.go b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/task_resource_parameters_gen.go new file mode 100644 index 00000000000..0a8b93fc372 --- /dev/null +++ b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/task_resource_parameters_gen.go @@ -0,0 +1,622 @@ +// Code generated by assertions generator; DO NOT EDIT. + +package resourceparametersassert + +import ( + "testing" + + "github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/acceptance/bettertestspoc/assert" + "github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/sdk" +) + +type TaskResourceParametersAssert struct { + *assert.ResourceAssert +} + +func TaskResourceParameters(t *testing.T, name string) *TaskResourceParametersAssert { + t.Helper() + + resourceParameterAssert := TaskResourceParametersAssert{ + ResourceAssert: assert.NewResourceAssert(name, "parameters"), + } + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert +} + +func ImportedTaskResourceParameters(t *testing.T, id string) *TaskResourceParametersAssert { + t.Helper() + + resourceParameterAssert := TaskResourceParametersAssert{ + ResourceAssert: assert.NewImportedResourceAssert(id, "imported parameters"), + } + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert +} + +//////////////////////////// +// Parameter value checks // +//////////////////////////// + +func (t *TaskResourceParametersAssert) HasSuspendTaskAfterNumFailures(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterSuspendTaskAfterNumFailures, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTaskAutoRetryAttempts(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterTaskAutoRetryAttempts, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUserTaskManagedInitialWarehouseSize(expected sdk.WarehouseSize) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterUserTaskManagedInitialWarehouseSize, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUserTaskMinimumTriggerIntervalInSeconds(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterUserTaskMinimumTriggerIntervalInSeconds, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUserTaskTimeoutMs(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterUserTaskTimeoutMs, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasAbortDetachedQuery(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterAbortDetachedQuery, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasAutocommit(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterAutocommit, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasBinaryInputFormat(expected sdk.BinaryInputFormat) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterBinaryInputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasBinaryOutputFormat(expected sdk.BinaryOutputFormat) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterBinaryOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientMemoryLimit(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterClientMemoryLimit, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientMetadataRequestUseConnectionCtx(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterClientMetadataRequestUseConnectionCtx, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientPrefetchThreads(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterClientPrefetchThreads, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientResultChunkSize(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterClientResultChunkSize, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientResultColumnCaseInsensitive(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterClientResultColumnCaseInsensitive, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientSessionKeepAlive(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterClientSessionKeepAlive, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientSessionKeepAliveHeartbeatFrequency(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterClientSessionKeepAliveHeartbeatFrequency, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientTimestampTypeMapping(expected sdk.ClientTimestampTypeMapping) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterClientTimestampTypeMapping, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasDateInputFormat(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterDateInputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasDateOutputFormat(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterDateOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasEnableUnloadPhysicalTypeOptimization(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterEnableUnloadPhysicalTypeOptimization, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasErrorOnNondeterministicMerge(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterErrorOnNondeterministicMerge, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasErrorOnNondeterministicUpdate(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterErrorOnNondeterministicUpdate, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasGeographyOutputFormat(expected sdk.GeographyOutputFormat) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterGeographyOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasGeometryOutputFormat(expected sdk.GeometryOutputFormat) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterGeometryOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasJdbcTreatTimestampNtzAsUtc(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterJdbcTreatTimestampNtzAsUtc, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasJdbcUseSessionTimezone(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterJdbcUseSessionTimezone, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasJsonIndent(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterJsonIndent, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasLockTimeout(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterLockTimeout, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasLogLevel(expected sdk.LogLevel) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterLogLevel, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasMultiStatementCount(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterMultiStatementCount, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasNoorderSequenceAsDefault(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterNoorderSequenceAsDefault, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasOdbcTreatDecimalAsInt(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterOdbcTreatDecimalAsInt, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasQueryTag(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterQueryTag, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasQuotedIdentifiersIgnoreCase(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterQuotedIdentifiersIgnoreCase, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasRowsPerResultset(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterRowsPerResultset, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasS3StageVpceDnsName(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterS3StageVpceDnsName, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasSearchPath(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterSearchPath, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasStatementQueuedTimeoutInSeconds(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterStatementQueuedTimeoutInSeconds, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasStatementTimeoutInSeconds(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterStatementTimeoutInSeconds, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasStrictJsonOutput(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterStrictJsonOutput, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampDayIsAlways24h(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterTimestampDayIsAlways24h, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampInputFormat(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterTimestampInputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampLtzOutputFormat(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterTimestampLtzOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampNtzOutputFormat(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterTimestampNtzOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampOutputFormat(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterTimestampOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampTypeMapping(expected sdk.TimestampTypeMapping) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterTimestampTypeMapping, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampTzOutputFormat(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterTimestampTzOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimezone(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterTimezone, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimeInputFormat(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterTimeInputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimeOutputFormat(expected string) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterValueSet(sdk.TaskParameterTimeOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTraceLevel(expected sdk.TraceLevel) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterTraceLevel, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTransactionAbortOnError(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterTransactionAbortOnError, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTransactionDefaultIsolationLevel(expected sdk.TransactionDefaultIsolationLevel) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterTransactionDefaultIsolationLevel, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTwoDigitCenturyStart(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterTwoDigitCenturyStart, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUnsupportedDdlAction(expected sdk.UnsupportedDDLAction) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterStringUnderlyingValueSet(sdk.TaskParameterUnsupportedDdlAction, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUseCachedResult(expected bool) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterBoolValueSet(sdk.TaskParameterUseCachedResult, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasWeekOfYearPolicy(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterWeekOfYearPolicy, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasWeekStart(expected int) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterIntValueSet(sdk.TaskParameterWeekStart, expected)) + return t +} + +//////////////////////////// +// Parameter level checks // +//////////////////////////// + +func (t *TaskResourceParametersAssert) HasSuspendTaskAfterNumFailuresLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterSuspendTaskAfterNumFailures, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTaskAutoRetryAttemptsLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTaskAutoRetryAttempts, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUserTaskManagedInitialWarehouseSizeLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterUserTaskManagedInitialWarehouseSize, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUserTaskMinimumTriggerIntervalInSecondsLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterUserTaskMinimumTriggerIntervalInSeconds, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUserTaskTimeoutMsLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterUserTaskTimeoutMs, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasAbortDetachedQueryLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterAbortDetachedQuery, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasAutocommitLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterAutocommit, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasBinaryInputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterBinaryInputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasBinaryOutputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterBinaryOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientMemoryLimitLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterClientMemoryLimit, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientMetadataRequestUseConnectionCtxLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterClientMetadataRequestUseConnectionCtx, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientPrefetchThreadsLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterClientPrefetchThreads, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientResultChunkSizeLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterClientResultChunkSize, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientResultColumnCaseInsensitiveLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterClientResultColumnCaseInsensitive, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientSessionKeepAliveLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterClientSessionKeepAlive, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientSessionKeepAliveHeartbeatFrequencyLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterClientSessionKeepAliveHeartbeatFrequency, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasClientTimestampTypeMappingLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterClientTimestampTypeMapping, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasDateInputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterDateInputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasDateOutputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterDateOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasEnableUnloadPhysicalTypeOptimizationLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterEnableUnloadPhysicalTypeOptimization, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasErrorOnNondeterministicMergeLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterErrorOnNondeterministicMerge, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasErrorOnNondeterministicUpdateLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterErrorOnNondeterministicUpdate, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasGeographyOutputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterGeographyOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasGeometryOutputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterGeometryOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasJdbcTreatTimestampNtzAsUtcLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterJdbcTreatTimestampNtzAsUtc, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasJdbcUseSessionTimezoneLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterJdbcUseSessionTimezone, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasJsonIndentLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterJsonIndent, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasLockTimeoutLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterLockTimeout, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasLogLevelLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterLogLevel, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasMultiStatementCountLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterMultiStatementCount, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasNoorderSequenceAsDefaultLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterNoorderSequenceAsDefault, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasOdbcTreatDecimalAsIntLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterOdbcTreatDecimalAsInt, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasQueryTagLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterQueryTag, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasQuotedIdentifiersIgnoreCaseLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterQuotedIdentifiersIgnoreCase, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasRowsPerResultsetLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterRowsPerResultset, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasS3StageVpceDnsNameLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterS3StageVpceDnsName, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasSearchPathLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterSearchPath, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasStatementQueuedTimeoutInSecondsLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterStatementQueuedTimeoutInSeconds, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasStatementTimeoutInSecondsLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterStatementTimeoutInSeconds, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasStrictJsonOutputLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterStrictJsonOutput, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampDayIsAlways24hLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimestampDayIsAlways24h, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampInputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimestampInputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampLtzOutputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimestampLtzOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampNtzOutputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimestampNtzOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampOutputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimestampOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampTypeMappingLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimestampTypeMapping, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimestampTzOutputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimestampTzOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimezoneLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimezone, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimeInputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimeInputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTimeOutputFormatLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTimeOutputFormat, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTraceLevelLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTraceLevel, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTransactionAbortOnErrorLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTransactionAbortOnError, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTransactionDefaultIsolationLevelLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTransactionDefaultIsolationLevel, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasTwoDigitCenturyStartLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterTwoDigitCenturyStart, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUnsupportedDdlActionLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterUnsupportedDdlAction, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasUseCachedResultLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterUseCachedResult, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasWeekOfYearPolicyLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterWeekOfYearPolicy, expected)) + return t +} + +func (t *TaskResourceParametersAssert) HasWeekStartLevel(expected sdk.ParameterType) *TaskResourceParametersAssert { + t.AddAssertion(assert.ResourceParameterLevelSet(sdk.TaskParameterWeekStart, expected)) + return t +} diff --git a/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/user_resource_parameters_gen.go b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/user_resource_parameters_gen.go index 44e722144ca..429525699f9 100644 --- a/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/user_resource_parameters_gen.go +++ b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/user_resource_parameters_gen.go @@ -16,21 +16,21 @@ type UserResourceParametersAssert struct { func UserResourceParameters(t *testing.T, name string) *UserResourceParametersAssert { t.Helper() - u := UserResourceParametersAssert{ + resourceParameterAssert := UserResourceParametersAssert{ ResourceAssert: assert.NewResourceAssert(name, "parameters"), } - u.AddAssertion(assert.ValueSet("parameters.#", "1")) - return &u + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert } func ImportedUserResourceParameters(t *testing.T, id string) *UserResourceParametersAssert { t.Helper() - u := UserResourceParametersAssert{ + resourceParameterAssert := UserResourceParametersAssert{ ResourceAssert: assert.NewImportedResourceAssert(id, "imported parameters"), } - u.AddAssertion(assert.ValueSet("parameters.#", "1")) - return &u + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert } //////////////////////////// diff --git a/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/warehouse_resource_parameters_gen.go b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/warehouse_resource_parameters_gen.go index b5b283946a2..36a414a41bd 100644 --- a/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/warehouse_resource_parameters_gen.go +++ b/pkg/acceptance/bettertestspoc/assert/resourceparametersassert/warehouse_resource_parameters_gen.go @@ -16,21 +16,21 @@ type WarehouseResourceParametersAssert struct { func WarehouseResourceParameters(t *testing.T, name string) *WarehouseResourceParametersAssert { t.Helper() - w := WarehouseResourceParametersAssert{ + resourceParameterAssert := WarehouseResourceParametersAssert{ ResourceAssert: assert.NewResourceAssert(name, "parameters"), } - w.AddAssertion(assert.ValueSet("parameters.#", "1")) - return &w + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert } func ImportedWarehouseResourceParameters(t *testing.T, id string) *WarehouseResourceParametersAssert { t.Helper() - w := WarehouseResourceParametersAssert{ + resourceParameterAssert := WarehouseResourceParametersAssert{ ResourceAssert: assert.NewImportedResourceAssert(id, "imported parameters"), } - w.AddAssertion(assert.ValueSet("parameters.#", "1")) - return &w + resourceParameterAssert.AddAssertion(assert.ValueSet("parameters.#", "1")) + return &resourceParameterAssert } //////////////////////////// diff --git a/pkg/sdk/parameters.go b/pkg/sdk/parameters.go index 763cd7668c1..8f275752e1b 100644 --- a/pkg/sdk/parameters.go +++ b/pkg/sdk/parameters.go @@ -659,7 +659,11 @@ type TaskParameter string // TODO(SNOW-1348116 - next prs): Handle task parameters const ( // Task Parameters - TaskParameterUserTaskManagedInitialWarehouseSize TaskParameter = "USER_TASK_MANAGED_INITIAL_WAREHOUSE_SIZE" + TaskParameterSuspendTaskAfterNumFailures TaskParameter = "SUSPEND_TASK_AFTER_NUM_FAILURES" + TaskParameterTaskAutoRetryAttempts TaskParameter = "TASK_AUTO_RETRY_ATTEMPTS" + TaskParameterUserTaskManagedInitialWarehouseSize TaskParameter = "USER_TASK_MANAGED_INITIAL_WAREHOUSE_SIZE" + TaskParameterUserTaskMinimumTriggerIntervalInSeconds TaskParameter = "USER_TASK_MINIMUM_TRIGGER_INTERVAL_IN_SECONDS" + TaskParameterUserTaskTimeoutMs TaskParameter = "USER_TASK_TIMEOUT_MS" // Session Parameters (inherited) TaskParameterAbortDetachedQuery TaskParameter = "ABORT_DETACHED_QUERY" @@ -681,7 +685,6 @@ const ( TaskParameterErrorOnNondeterministicUpdate TaskParameter = "ERROR_ON_NONDETERMINISTIC_UPDATE" TaskParameterGeographyOutputFormat TaskParameter = "GEOGRAPHY_OUTPUT_FORMAT" TaskParameterGeometryOutputFormat TaskParameter = "GEOMETRY_OUTPUT_FORMAT" - TaskParameterJdbcTreatDecimalAsInt TaskParameter = "JDBC_TREAT_DECIMAL_AS_INT" TaskParameterJdbcTreatTimestampNtzAsUtc TaskParameter = "JDBC_TREAT_TIMESTAMP_NTZ_AS_UTC" TaskParameterJdbcUseSessionTimezone TaskParameter = "JDBC_USE_SESSION_TIMEZONE" TaskParameterJsonIndent TaskParameter = "JSON_INDENT" @@ -695,7 +698,6 @@ const ( TaskParameterRowsPerResultset TaskParameter = "ROWS_PER_RESULTSET" TaskParameterS3StageVpceDnsName TaskParameter = "S3_STAGE_VPCE_DNS_NAME" TaskParameterSearchPath TaskParameter = "SEARCH_PATH" - TaskParameterSimulatedDataSharingConsumer TaskParameter = "SIMULATED_DATA_SHARING_CONSUMER" TaskParameterStatementQueuedTimeoutInSeconds TaskParameter = "STATEMENT_QUEUED_TIMEOUT_IN_SECONDS" TaskParameterStatementTimeoutInSeconds TaskParameter = "STATEMENT_TIMEOUT_IN_SECONDS" TaskParameterStrictJsonOutput TaskParameter = "STRICT_JSON_OUTPUT" diff --git a/pkg/sdk/testint/tasks_gen_integration_test.go b/pkg/sdk/testint/tasks_gen_integration_test.go index 23ec833cd4e..edf5ba09aa9 100644 --- a/pkg/sdk/testint/tasks_gen_integration_test.go +++ b/pkg/sdk/testint/tasks_gen_integration_test.go @@ -141,6 +141,8 @@ func TestInt_Tasks(t *testing.T) { require.NoError(t, err) assertTask(t, task, id, "") + + assertions.AssertThat(t, objectparametersassert.TaskParameters(t, id).HasAllDefaults()) }) t.Run("create task: with initial warehouse", func(t *testing.T) { @@ -184,6 +186,11 @@ func TestInt_Tasks(t *testing.T) { require.NoError(t, err) assertTaskWithOptions(t, task, id, "some comment", testClientHelper().Ids.WarehouseId().Name(), "10 MINUTE", `SYSTEM$STREAM_HAS_DATA('MYSTREAM')`, true, `{"output_dir": "/temp/test_directory/", "learning_rate": 0.1}`, nil, &errorIntegrationId) + assertions.AssertThat(t, objectparametersassert.TaskParameters(t, id). + HasJsonIndent(4). + HasUserTaskTimeoutMs(500). + HasSuspendTaskAfterNumFailures(3), + ) }) t.Run("create task: with after", func(t *testing.T) { @@ -422,8 +429,8 @@ func TestInt_Tasks(t *testing.T) { WithAllowOverlappingExecution(true). WithUserTaskTimeoutMs(10). WithSessionParameters(sdk.SessionParameters{ - // TODO(SNOW-1348116 - next prs): fill and assert parameters - Autocommit: sdk.Bool(true), + Autocommit: sdk.Bool(false), + AbortDetachedQuery: sdk.Bool(true), }). WithSuspendTaskAfterNumFailures(15). WithComment("some_comment"). @@ -446,6 +453,13 @@ func TestInt_Tasks(t *testing.T) { HasComment("some_comment"). HasTaskRelations(sdk.TaskRelations{}), ) + assertions.AssertThat(t, objectparametersassert.TaskParameters(t, task.ID()). + HasAutocommit(false). + HasAbortDetachedQuery(true). + HasUserTaskTimeoutMs(10). + HasSuspendTaskAfterNumFailures(15). + HasTaskAutoRetryAttempts(15), + ) err = client.Tasks.CreateOrAlter(ctx, sdk.NewCreateOrAlterTaskRequest(id, sql)) require.NoError(t, err) @@ -462,6 +476,13 @@ func TestInt_Tasks(t *testing.T) { HasComment(""). HasTaskRelations(sdk.TaskRelations{}), ) + assertions.AssertThat(t, objectparametersassert.TaskParameters(t, task.ID()). + HasDefaultAutocommitValue(). + HasDefaultAbortDetachedQueryValue(). + HasDefaultUserTaskTimeoutMsValue(). + HasDefaultSuspendTaskAfterNumFailuresValue(). + HasDefaultTaskAutoRetryAttemptsValue(), + ) require.Equal(t, createdOn, alteredTask.CreatedOn) }) @@ -491,9 +512,8 @@ func TestInt_Tasks(t *testing.T) { // WithWarehouse(testClientHelper().Ids.WarehouseId()). WithErrorNotificationIntegration(errorIntegrationId). WithSessionParameters(sdk.SessionParameters{ - Autocommit: sdk.Bool(true), - ClientSessionKeepAlive: sdk.Bool(true), - // TODO(SNOW-1348116 - next prs): fill and assert parameters + Autocommit: sdk.Bool(false), + AbortDetachedQuery: sdk.Bool(true), }). WithSchedule("10 MINUTE"). WithConfig(`$${"output_dir": "/temp/test_directory/", "learning_rate": 0.1}$$`). @@ -506,11 +526,6 @@ func TestInt_Tasks(t *testing.T) { )) require.NoError(t, err) - // TODO(SNOW-1348116 - next prs): Assert parameters - // assertions.AssertThat(t, objectparametersassert.TaskParameters(t, task.ID()). - // HasUserTaskManagedInitialWarehouseSize() - // ) - assertions.AssertThat(t, objectassert.Task(t, task.ID()). // HasWarehouse(testClientHelper().Ids.WarehouseId().Name()). HasErrorIntegration(sdk.Pointer(errorIntegrationId)). @@ -519,6 +534,14 @@ func TestInt_Tasks(t *testing.T) { HasAllowOverlappingExecution(true). HasComment("new comment"), ) + assertions.AssertThat(t, objectparametersassert.TaskParameters(t, task.ID()). + HasAutocommit(false). + HasAbortDetachedQuery(true). + HasUserTaskTimeoutMs(1000). + HasSuspendTaskAfterNumFailures(100). + HasTaskAutoRetryAttempts(10). + HasUserTaskMinimumTriggerIntervalInSeconds(15), + ) err = client.Tasks.Alter(ctx, sdk.NewAlterTaskRequest(task.ID()).WithUnset(*sdk.NewTaskUnsetRequest(). WithErrorIntegration(true). @@ -545,11 +568,7 @@ func TestInt_Tasks(t *testing.T) { HasAllowOverlappingExecution(false). HasComment(""), ) - - // TODO(SNOW-1348116 - next prs): Assert parameters - // assertions.AssertThat(t, objectparametersassert.TaskParameters(t, task.ID()). - // HasUserTaskManagedInitialWarehouseSize() - // ) + assertions.AssertThat(t, objectparametersassert.TaskParameters(t, task.ID()).HasAllDefaults()) }) t.Run("alter task: set and unset tag", func(t *testing.T) { diff --git a/pkg/sdk/testint/users_integration_test.go b/pkg/sdk/testint/users_integration_test.go index ea5388c341d..7899c195562 100644 --- a/pkg/sdk/testint/users_integration_test.go +++ b/pkg/sdk/testint/users_integration_test.go @@ -158,6 +158,7 @@ func TestInt_Users(t *testing.T) { user, err := client.Users.ShowByID(ctx, id) require.NoError(t, err) + objectassert.DatabaseFromObject() assertions.AssertThatObject(t, objectassert.UserFromObject(t, user). HasName(id.Name()). HasHasPassword(true).