[AB#1669514] arrow test

snowflakedb · Nov 21, 2024 · ca44e68 · ca44e68
1 parent 1516dd4
commit ca44e68
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/driver_test.go b/driver_test.go
@@ -927,18 +927,23 @@ func (uuid testUUID) Value() (driver.Value, error) {
 
 func TestUUID(t *testing.T) {
 	t.Run("Basic Type", func(t *testing.T) {
-		testUuid(t, false)
+		testUuid(t, false, false)
 	})
 
 	t.Run("JSON", func(t *testing.T) {
-		testUuid(t, true)
+		testUuid(t, true, false)
+	})
+	t.Run("Arrow", func(t *testing.T) {
+		testUuid(t, false, true)
 	})
 }
 
-func testUuid(t *testing.T, json bool) {
+func testUuid(t *testing.T, json, arrow bool) {
 	runDBTest(t, func(dbt *DBTest) {
 		if json {
 			dbt.mustExec(forceJSON)
+		} else if arrow {
+			dbt.mustExec(forceARROW)
 		}
 
 		types := []string{"CHAR(255)", "VARCHAR(255)", "TEXT", "STRING"}