delete_test.go

package qb

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestDelete(t *testing.T) {
	sqlite := NewDialect("sqlite3")
	sqlite.SetEscaping(true)

	mysql := NewDialect("mysql")
	mysql.SetEscaping(true)

	postgres := NewDialect("postgres")
	postgres.SetEscaping(true)

	usersTable := Table(
		"users",
		Column("id", Varchar().Size(36)),
		Column("email", Varchar().Unique()),
	)

	var statement *Stmt

	statement = Delete(usersTable).
		Where(Eq(usersTable.C("id"), 5)).
		Build(sqlite)

	assert.Equal(t, statement.SQL(), "DELETE FROM users\nWHERE (users.id = ?);")
	assert.Equal(t, statement.Bindings(), []interface{}{5})

	statement = Delete(usersTable).
		Where(Eq(usersTable.C("id"), 5)).
		Build(mysql)

	assert.Equal(t, statement.SQL(), "DELETE FROM `users`\nWHERE (`users`.`id` = ?);")
	assert.Equal(t, statement.Bindings(), []interface{}{5})

	statement = Delete(usersTable).
		Where(Eq(usersTable.C("id"), 5)).
		Returning(usersTable.C("id")).
		Build(postgres)

	assert.Equal(t, statement.SQL(), "DELETE FROM \"users\"\nWHERE (\"users\".\"id\" = $1)\nRETURNING \"id\";")
	assert.Equal(t, statement.Bindings(), []interface{}{5})

	statement = Delete(usersTable).Build(sqlite)
	assert.Equal(t, statement.SQL(), "DELETE FROM users;")
}