sqlingo/delete_test.go

40 lines
1.1 KiB
Go

package sqlingo
import (
"context"
"errors"
"testing"
)
func TestDelete(t *testing.T) {
errorExpression := expression{
builder: func(scope scope) (string, error) {
return "", errors.New("error")
},
}
db := newMockDatabase()
if _, err := db.DeleteFrom(Table1).Where(staticExpression("##", 1, false)).Execute(); err != nil {
t.Error(err)
}
assertLastSql(t, "DELETE FROM `table1` WHERE ##")
if _, err := db.DeleteFrom(Table1).Where(errorExpression).Execute(); err == nil {
t.Error("should get error here")
}
if _, err := db.DeleteFrom(Table1).Where(Raw("#1#")).Limit(3).Execute(); err != nil {
t.Error(err)
}
assertLastSql(t, "DELETE FROM `table1` WHERE #1# LIMIT 3")
if _, err := db.DeleteFrom(Table1).Where(Raw("#1#")).OrderBy(Raw("#2#")).Limit(3).Execute(); err != nil {
t.Error(err)
}
assertLastSql(t, "DELETE FROM `table1` WHERE #1# ORDER BY #2# LIMIT 3")
if _, err := db.DeleteFrom(Table1).Where(Raw("#1#")).WithContext(context.Background()).Execute(); err != nil {
t.Error(err)
}
assertLastSql(t, "DELETE FROM `table1` WHERE #1#")
}