test: cases

couple fixes needed to get Array(Nullable(...)) & Nullable(DateTime64(...)) cases to work
ClickHouse · Oct 13, 2024 · 5eff914 · 5eff914
1 parent 9b9a1e4
commit 5eff914
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/proto/col_auto_test.go b/proto/col_auto_test.go
@@ -46,6 +46,8 @@ func TestColAuto_Infer(t *testing.T) {
 		ColumnTypeUUID,
 		ColumnTypeArray.Sub(ColumnTypeUUID),
 		ColumnTypeNullable.Sub(ColumnTypeUUID),
+		"Array(Nullable(Int8))",
+		"Nullable(DateTime64(3))",
 	} {
 		r := AutoResult("foo")
 		require.NoError(t, r.Data.(Inferable).Infer(columnType))

diff --git a/proto/col_datetime64.go b/proto/col_datetime64.go
@@ -126,6 +126,10 @@ func (c ColDateTime64) Raw() *ColDateTime64Raw {
 	return &ColDateTime64Raw{ColDateTime64: c}
 }
 
+func (c *ColDateTime64) Nullable() *ColNullable[time.Time] {
+	return &ColNullable[time.Time]{Values: c}
+}
+
 func (c *ColDateTime64) Array() *ColArr[time.Time] {
 	return &ColArr[time.Time]{Data: c}
 }

diff --git a/proto/col_nullable.go b/proto/col_nullable.go
@@ -117,6 +117,12 @@ func (c ColNullable[T]) Row(i int) Nullable[T] {
 	}
 }
 
+func (c *ColNullable[T]) Array() *ColArr[Nullable[T]] {
+	return &ColArr[Nullable[T]]{
+		Data: c,
+	}
+}
+
 func (c *ColNullable[T]) Reset() {
 	c.Nulls.Reset()
 	c.Values.Reset()