Add SQLBuilder support for SQLite databases.

tests/sqlite/delete_test.go

@@ -0,0 +1,83 @@
+package sqlite
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/go-jet/jet/v2/internal/testutils"
+	. "github.com/go-jet/jet/v2/sqlite"
+	"github.com/go-jet/jet/v2/tests/.gentestdata/sqlite/test_sample/model"
+	. "github.com/go-jet/jet/v2/tests/.gentestdata/sqlite/test_sample/table"
+	"github.com/stretchr/testify/require"
+)
+
+func TestDelete_WHERE_RETURNING(t *testing.T) {
+	tx := beginSampleDBTx(t)
+	defer tx.Rollback()
+
+	var expectedSQL = `
+DELETE FROM link
+WHERE link.name IN ('Bing', 'Yahoo')
+RETURNING link.id AS "link.id",
+          link.url AS "link.url",
+          link.name AS "link.name",
+          link.description AS "link.description";
+`
+	deleteStmt := Link.DELETE().
+		WHERE(Link.Name.IN(String("Bing"), String("Yahoo"))).
+		RETURNING(Link.AllColumns)
+
+	testutils.AssertDebugStatementSql(t, deleteStmt, expectedSQL, "Bing", "Yahoo")
+	var dest []model.Link
+	err := deleteStmt.Query(tx, &dest)
+	require.NoError(t, err)
+	require.Len(t, dest, 2)
+	requireLogged(t, deleteStmt)
+}
+
+func TestDeleteWithWhereOrderByLimit(t *testing.T) {
+	t.SkipNow() // Until https://github.com/mattn/go-sqlite3/pull/802 is fixed
+	tx := beginSampleDBTx(t)
+	defer tx.Rollback()
+
+	sampleDB.Stats()
+
+	var expectedSQL = `
+DELETE FROM link
+WHERE link.name IN ('Bing', 'Yahoo')
+ORDER BY link.name
+LIMIT 1;
+`
+	deleteStmt := Link.DELETE().
+		WHERE(Link.Name.IN(String("Bing"), String("Yahoo"))).
+		ORDER_BY(Link.Name).
+		LIMIT(1)
+
+	testutils.AssertDebugStatementSql(t, deleteStmt, expectedSQL, "Bing", "Yahoo", int64(1))
+	testutils.AssertExec(t, deleteStmt, tx, 1)
+	requireLogged(t, deleteStmt)
+}
+
+func TestDeleteContextDeadlineExceeded(t *testing.T) {
+	tx := beginSampleDBTx(t)
+	defer tx.Rollback()
+
+	deleteStmt := Link.
+		DELETE().
+		WHERE(Link.Name.IN(String("Bing"), String("Yahoo")))
+
+	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Microsecond)
+	defer cancel()
+
+	time.Sleep(10 * time.Millisecond)
+
+	dest := []model.Link{}
+	err := deleteStmt.QueryContext(ctx, tx, &dest)
+	require.Error(t, err, "context deadline exceeded")
+
+	_, err = deleteStmt.ExecContext(ctx, tx)
+	require.Error(t, err, "context deadline exceeded")
+
+	requireLogged(t, deleteStmt)
+}