1 package goqu_test
2
3 import (
4 "testing"
5
6 "github.com/DATA-DOG/go-sqlmock"
7 "github.com/doug-martin/goqu/v9"
8 "github.com/stretchr/testify/suite"
9 )
10
11 type (
12 dialectWrapperSuite struct {
13 suite.Suite
14 }
15 )
16
17 func (dws *dialectWrapperSuite) SetupSuite() {
18 testDialect := goqu.DefaultDialectOptions()
19
20 goqu.RegisterDialect("test", testDialect)
21 }
22
23 func (dws *dialectWrapperSuite) TearDownSuite() {
24 goqu.DeregisterDialect("test")
25 }
26
27 func (dws *dialectWrapperSuite) TestFrom() {
28 dw := goqu.Dialect("test")
29 dws.Equal(goqu.From("table").WithDialect("test"), dw.From("table"))
30 }
31
32 func (dws *dialectWrapperSuite) TestSelect() {
33 dw := goqu.Dialect("test")
34 dws.Equal(goqu.Select("col").WithDialect("test"), dw.Select("col"))
35 }
36
37 func (dws *dialectWrapperSuite) TestInsert() {
38 dw := goqu.Dialect("test")
39 dws.Equal(goqu.Insert("table").WithDialect("test"), dw.Insert("table"))
40 }
41
42 func (dws *dialectWrapperSuite) TestDelete() {
43 dw := goqu.Dialect("test")
44 dws.Equal(goqu.Delete("table").WithDialect("test"), dw.Delete("table"))
45 }
46
47 func (dws *dialectWrapperSuite) TestTruncate() {
48 dw := goqu.Dialect("test")
49 dws.Equal(goqu.Truncate("table").WithDialect("test"), dw.Truncate("table"))
50 }
51
52 func (dws *dialectWrapperSuite) TestDB() {
53 mDB, _, err := sqlmock.New()
54 dws.Require().NoError(err)
55 dw := goqu.Dialect("test")
56 dws.Equal(goqu.New("test", mDB), dw.DB(mDB))
57 }
58
59 func TestDialectWrapper(t *testing.T) {
60 suite.Run(t, new(dialectWrapperSuite))
61 }
62
View as plain text