Add test for using BulkSQL with a SELECT

nyaruka · Aug 28, 2023 · c616cdf · c616cdf
1 parent bf60e41
commit c616cdf
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/dbutil/query_test.go b/dbutil/query_test.go
@@ -57,6 +57,14 @@ func TestBulkSQL(t *testing.T) {
 	assert.NoError(t, err)
 	assert.Equal(t, `INSERT INTO foo (id, name) VALUES($1, $2),($3, $4),($5, $6)`, query)
 	assert.Equal(t, []any{1, "Bob", 2, "Cathy", 3, "George"}, args)
+
+	// try a select
+	sql = `SELECT * FROM foo WHERE (id, name) IN (VALUES(:id, :name))`
+
+	query, args, err = dbutil.BulkSQL(db, sql, []any{contact{ID: 1, Name: "Bob"}, contact{ID: 2, Name: "Cathy"}})
+	assert.NoError(t, err)
+	assert.Equal(t, `SELECT * FROM foo WHERE (id, name) IN (VALUES($1, $2),($3, $4))`, query)
+	assert.Equal(t, []any{1, "Bob", 2, "Cathy"}, args)
 }
 
 func TestBulkQuery(t *testing.T) {