delete_test.go

package dat

import (
	"testing"

	"gopkg.in/stretchr/testify.v1/assert"
)

func BenchmarkDeleteSql(b *testing.B) {
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		DeleteFrom("alpha").Where("a", "b").Limit(1).OrderBy("id").ToSQL()
	}
}

func TestDeleteAllToSql(t *testing.T) {
	sql, _ := DeleteFrom("a").ToSQL()

	assert.Equal(t, sql, "DELETE FROM a")
}

func TestDeleteSingleToSql(t *testing.T) {
	sql, args := DeleteFrom("a").Where("id = $1", 1).ToSQL()

	assert.Equal(t, sql, "DELETE FROM a WHERE (id = $1)")
	assert.Equal(t, args, []interface{}{1})
}

func TestDeleteTenStaringFromTwentyToSql(t *testing.T) {
	sql, _ := DeleteFrom("a").Limit(10).Offset(20).OrderBy("id").ToSQL()

	assert.Equal(t, sql, "DELETE FROM a ORDER BY id LIMIT 10 OFFSET 20")
}

func TestDeleteWhereExprSql(t *testing.T) {
	expr := Expr("id=$1", 100)
	sql, args := DeleteFrom("a").Where("foo = $1", "bar").Where(expr).ToSQL()
	assert.Equal(t, sql, `DELETE FROM a WHERE (foo = $1) AND (id=$2)`)
	assert.Exactly(t, args, []interface{}{"bar", 100})
}