package sqlite

import (
	"context"
	"github.com/go-jet/jet/v2/qrm"
	"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) {

	deleteStmt := Link.
		DELETE().
		WHERE(Link.Name.IN(String("Bing"), String("Yahoo")))

	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Microsecond)
	defer cancel()

	time.Sleep(20 * time.Millisecond)

	testutils.ExecuteInTxAndRollback(t, sampleDB, func(tx qrm.DB) {
		var dest []model.Link
		err := deleteStmt.QueryContext(ctx, tx, &dest)
		require.Error(t, err, "context deadline exceeded")
	})

	testutils.ExecuteInTxAndRollback(t, sampleDB, func(tx qrm.DB) {
		_, err := deleteStmt.ExecContext(ctx, tx)
		require.Error(t, err, "context deadline exceeded")
	})

	requireLogged(t, deleteStmt)
}