Adding one more spot + test.

artie-labs · Oct 31, 2024 · 266c6e9 · 266c6e9
1 parent dbd10f4
commit 266c6e9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/clients/shared/default_value.go b/clients/shared/default_value.go
@@ -28,6 +28,13 @@ func DefaultValue(column columns.Column, dialect sql.Dialect) (any, error) {
 		}
 
 		return sql.QuoteLiteral(_time.Format(ext.PostgresDateFormat)), nil
+	case typing.TimestampNTZ.Kind:
+		_time, err := ext.ParseTimestampNTZFromInterface(column.DefaultValue())
+		if err != nil {
+			return "", fmt.Errorf("failed to cast colVal as time.Time, colVal: '%v', err: %w", column.DefaultValue(), err)
+		}
+
+		return sql.QuoteLiteral(_time.Format(ext.RFC3339NoTZ)), nil
 	case typing.ETime.Kind:
 		if err := column.KindDetails.EnsureExtendedTimeDetails(); err != nil {
 			return nil, err

diff --git a/clients/shared/default_value_test.go b/clients/shared/default_value_test.go
@@ -82,6 +82,11 @@ func TestColumn_DefaultValue(t *testing.T) {
 			col:           columns.NewColumnWithDefaultValue("", typing.Date, birthdayDateTime),
 			expectedValue: "'2022-09-06'",
 		},
+		{
+			name:          "timestamp_ntz",
+			col:           columns.NewColumnWithDefaultValue("", typing.TimestampNTZ, birthdayDateTime),
+			expectedValue: "'2022-09-06T03:19:24.942'",
+		},
 		{
 			name:          "time",
 			col:           columns.NewColumnWithDefaultValue("", timeKind, birthdayDateTime),