sqlutil/helper_test.go at main · allisson/sqlutil · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
package sqlutil

import (
	"context"
	"regexp"
	"testing"

	"github.com/DATA-DOG/go-sqlmock"
	"github.com/stretchr/testify/assert"
)

type player struct {
	ID   int    `db:"id" fieldtag:"insert"`
	Name string `db:"name" fieldtag:"insert,update"`
}

func TestGet(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.Nil(t, err)
	defer db.Close()

	rows := sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "Ronaldinho Gaúcho")
	mock.ExpectQuery(regexp.QuoteMeta(`SELECT * FROM players WHERE id = $1`)).
		WithArgs("R10").
		WillReturnRows(rows)

	options := NewFindOptions(PostgreSQLFlavor).WithFilter("id", "R10")
	p := player{}
	err = Get(context.Background(), db, "players", options, &p)
	assert.Nil(t, err)
	assert.Equal(t, 1, p.ID)
	assert.Equal(t, "Ronaldinho Gaúcho", p.Name)
}

func TestSelect(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.Nil(t, err)
	defer db.Close()

	rows := sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "Ronaldinho Gaúcho").AddRow(2, "Ronaldo Fenômeno")
	mock.ExpectQuery(regexp.QuoteMeta(`SELECT * FROM players LIMIT $1 OFFSET $2 FOR UPDATE SKIP LOCKED`)).WillReturnRows(rows)

	options := NewFindAllOptions(PostgreSQLFlavor).WithLimit(10).WithOffset(0).WithForUpdate("SKIP LOCKED")
	p := []*player{}
	err = Select(context.Background(), db, "players", options, &p)
	assert.Nil(t, err)
	assert.Len(t, p, 2)
	assert.Equal(t, 1, p[0].ID)
	assert.Equal(t, "Ronaldinho Gaúcho", p[0].Name)
	assert.Equal(t, 2, p[1].ID)
	assert.Equal(t, "Ronaldo Fenômeno", p[1].Name)
}

func TestInsert(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.Nil(t, err)
	defer db.Close()

	//rows := sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "Ronaldinho Gaúcho")
	mock.ExpectExec(regexp.QuoteMeta(`INSERT INTO players`)).
		WithArgs(1, "Ronaldinho Gaúcho").
		WillReturnResult(sqlmock.NewResult(1, 1))

	p := player{
		ID:   1,
		Name: "Ronaldinho Gaúcho",
	}
	err = Insert(context.Background(), db, PostgreSQLFlavor, "", "players", &p)
	assert.Nil(t, err)
}

func TestUpdate(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.Nil(t, err)
	defer db.Close()

	mock.ExpectExec(regexp.QuoteMeta(`UPDATE players SET id = $1, name = $2 WHERE id = $3`)).
		WithArgs(1, "Ronaldinho Gaúcho", 1).
		WillReturnResult(sqlmock.NewResult(1, 1))

	p := player{
		ID:   1,
		Name: "Ronaldinho Gaúcho",
	}
	err = Update(context.Background(), db, PostgreSQLFlavor, "", "players", p.ID, &p)
	assert.Nil(t, err)
}

func TestDelete(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.Nil(t, err)
	defer db.Close()

	mock.ExpectExec(regexp.QuoteMeta(`DELETE FROM players WHERE id = $1`)).
		WithArgs(1).
		WillReturnResult(sqlmock.NewResult(1, 1))

	p := player{
		ID:   1,
		Name: "Ronaldinho Gaúcho",
	}
	err = Delete(context.Background(), db, PostgreSQLFlavor, "players", p.ID)
	assert.Nil(t, err)
}

func TestUpdateWithOptions(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.Nil(t, err)
	defer db.Close()

	mock.ExpectExec(regexp.QuoteMeta(`UPDATE players SET name = $1 WHERE id = $2`)).
		WithArgs("Ronaldinho Gaúcho", 1).
		WillReturnResult(sqlmock.NewResult(1, 1))

	options := NewUpdateOptions(PostgreSQLFlavor).WithAssignment("name", "Ronaldinho Gaúcho").WithFilter("id", 1)
	err = UpdateWithOptions(context.Background(), db, PostgreSQLFlavor, "players", options)
	assert.Nil(t, err)
}

func TestDeleteWithOptions(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.Nil(t, err)
	defer db.Close()

	mock.ExpectExec(regexp.QuoteMeta(`DELETE FROM players WHERE id = $1`)).
		WithArgs(1).
		WillReturnResult(sqlmock.NewResult(1, 1))

	options := NewDeleteOptions(PostgreSQLFlavor).WithFilter("id", 1)
	err = DeleteWithOptions(context.Background(), db, PostgreSQLFlavor, "players", options)
	assert.Nil(t, err)
}