11package author
22
33import (
4- "context"
54 "errors"
5+ "testing"
6+
67 "github.com/jschaf/pggen/internal/ptrs"
78 "github.com/stretchr/testify/require"
8- "testing"
99
1010 "github.com/jackc/pgx/v4"
1111 "github.com/jschaf/pggen/internal/pgtest"
@@ -21,7 +21,7 @@ func TestNewQuerier_FindAuthorByID(t *testing.T) {
2121 insertAuthor (t , q , "george" , "washington" )
2222
2323 t .Run ("FindAuthorByID" , func (t * testing.T ) {
24- authorByID , err := q .FindAuthorByID (context . Background (), adamsID )
24+ authorByID , err := q .FindAuthorByID (t . Context (), adamsID )
2525 require .NoError (t , err )
2626 assert .Equal (t , FindAuthorByIDRow {
2727 AuthorID : adamsID ,
@@ -32,7 +32,7 @@ func TestNewQuerier_FindAuthorByID(t *testing.T) {
3232 })
3333
3434 t .Run ("FindAuthorByID - none-exists" , func (t * testing.T ) {
35- missingAuthorByID , err := q .FindAuthorByID (context . Background (), 888 )
35+ missingAuthorByID , err := q .FindAuthorByID (t . Context (), 888 )
3636 require .Error (t , err , "expected error when finding author ID that doesn't match" )
3737 assert .Zero (t , missingAuthorByID , "expected zero value when error" )
3838 if ! errors .Is (err , pgx .ErrNoRows ) {
@@ -50,7 +50,7 @@ func TestNewQuerier_FindAuthors(t *testing.T) {
5050 carverID := insertAuthor (t , q , "george" , "carver" )
5151
5252 t .Run ("FindAuthors - 1 row - john" , func (t * testing.T ) {
53- authors , err := q .FindAuthors (context . Background (), "john" )
53+ authors , err := q .FindAuthors (t . Context (), "john" )
5454 require .NoError (t , err )
5555 want := []FindAuthorsRow {
5656 {
@@ -64,7 +64,7 @@ func TestNewQuerier_FindAuthors(t *testing.T) {
6464 })
6565
6666 t .Run ("FindAuthors - 2 rows - george" , func (t * testing.T ) {
67- authors , err := q .FindAuthors (context . Background (), "george" )
67+ authors , err := q .FindAuthors (t . Context (), "george" )
6868 require .NoError (t , err )
6969 want := []FindAuthorsRow {
7070 {AuthorID : washingtonID , FirstName : "george" , LastName : "washington" , Suffix : nil },
@@ -74,7 +74,7 @@ func TestNewQuerier_FindAuthors(t *testing.T) {
7474 })
7575
7676 t .Run ("FindAuthors - 0 rows - joe" , func (t * testing.T ) {
77- authors , err := q .FindAuthors (context . Background (), "joe" )
77+ authors , err := q .FindAuthors (t . Context (), "joe" )
7878 require .NoError (t , err )
7979 assert .Equal (t , []FindAuthorsRow {}, authors )
8080 })
@@ -89,7 +89,7 @@ func TestNewQuerier_FindFirstNames(t *testing.T) {
8989 insertAuthor (t , q , "george" , "washington" )
9090
9191 t .Run ("FindAuthorByID" , func (t * testing.T ) {
92- firstNames , err := q .FindFirstNames (context . Background (), adamsID )
92+ firstNames , err := q .FindFirstNames (t . Context (), adamsID )
9393 require .NoError (t , err )
9494 assert .Equal (t , []* string {ptrs .String ("george" ), ptrs .String ("john" )}, firstNames )
9595 })
@@ -101,7 +101,7 @@ func TestNewQuerier_InsertAuthorSuffix(t *testing.T) {
101101 q := NewQuerier (conn )
102102
103103 t .Run ("InsertAuthorSuffix" , func (t * testing.T ) {
104- author , err := q .InsertAuthorSuffix (context . Background (), InsertAuthorSuffixParams {
104+ author , err := q .InsertAuthorSuffix (t . Context (), InsertAuthorSuffixParams {
105105 FirstName : "john" ,
106106 LastName : "adams" ,
107107 Suffix : "Jr." ,
@@ -127,12 +127,12 @@ func TestNewQuerier_DeleteAuthorsByFirstName(t *testing.T) {
127127 insertAuthor (t , q , "george" , "carver" )
128128
129129 t .Run ("DeleteAuthorsByFirstName" , func (t * testing.T ) {
130- tag , err := q .DeleteAuthorsByFirstName (context . Background (), "george" )
130+ tag , err := q .DeleteAuthorsByFirstName (t . Context (), "george" )
131131 require .NoError (t , err )
132132 assert .Truef (t , tag .Delete (), "expected delete tag; got %s" , tag .String ())
133133 assert .Equal (t , int64 (2 ), tag .RowsAffected ())
134134
135- authors , err := q .FindAuthors (context . Background (), "george" )
135+ authors , err := q .FindAuthors (t . Context (), "george" )
136136 require .NoError (t , err )
137137 assert .Empty (t , authors , "no authors should remain with first name of george" )
138138 })
@@ -143,15 +143,15 @@ func TestNewQuerier_DeleteAuthorsByFullName(t *testing.T) {
143143 defer cleanup ()
144144 q := NewQuerier (conn )
145145 washingtonID := insertAuthor (t , q , "george" , "washington" )
146- _ , err := q .InsertAuthorSuffix (context . Background (), InsertAuthorSuffixParams {
146+ _ , err := q .InsertAuthorSuffix (t . Context (), InsertAuthorSuffixParams {
147147 FirstName : "george" ,
148148 LastName : "washington" ,
149149 Suffix : "Jr." ,
150150 })
151151 require .NoError (t , err )
152152
153153 t .Run ("DeleteAuthorsByFullName" , func (t * testing.T ) {
154- tag , err := q .DeleteAuthorsByFullName (context . Background (), DeleteAuthorsByFullNameParams {
154+ tag , err := q .DeleteAuthorsByFullName (t . Context (), DeleteAuthorsByFullNameParams {
155155 FirstName : "george" ,
156156 LastName : "washington" ,
157157 Suffix : "Jr." ,
@@ -160,7 +160,7 @@ func TestNewQuerier_DeleteAuthorsByFullName(t *testing.T) {
160160 assert .Truef (t , tag .Delete (), "expected delete tag; got %s" , tag .String ())
161161 assert .Equal (t , int64 (1 ), tag .RowsAffected ())
162162
163- authors , err := q .FindAuthors (context . Background (), "george" )
163+ authors , err := q .FindAuthors (t . Context (), "george" )
164164 require .NoError (t , err )
165165 want := []FindAuthorsRow {
166166 {
@@ -179,21 +179,21 @@ func TestNewQuerier_StringAggFirstName(t *testing.T) {
179179 defer cleanup ()
180180 q := NewQuerier (conn )
181181 washingtonID := insertAuthor (t , q , "george" , "washington" )
182- _ , err := q .InsertAuthorSuffix (context . Background (), InsertAuthorSuffixParams {
182+ _ , err := q .InsertAuthorSuffix (t . Context (), InsertAuthorSuffixParams {
183183 FirstName : "george" ,
184184 LastName : "washington" ,
185185 Suffix : "Jr." ,
186186 })
187187 require .NoError (t , err )
188188
189189 t .Run ("StringAggFirstName - null" , func (t * testing.T ) {
190- firstNames , err := q .StringAggFirstName (context . Background (), 999 )
190+ firstNames , err := q .StringAggFirstName (t . Context (), 999 )
191191 require .NoError (t , err )
192192 require .Nil (t , firstNames )
193193 })
194194
195195 t .Run ("StringAggFirstName - one" , func (t * testing.T ) {
196- firstNames , err := q .StringAggFirstName (context . Background (), washingtonID )
196+ firstNames , err := q .StringAggFirstName (t . Context (), washingtonID )
197197 require .NoError (t , err )
198198 assert .Equal (t , "george" , * firstNames )
199199 })
@@ -204,29 +204,29 @@ func TestNewQuerier_ArrayAggFirstName(t *testing.T) {
204204 defer cleanup ()
205205 q := NewQuerier (conn )
206206 washingtonID := insertAuthor (t , q , "george" , "washington" )
207- _ , err := q .InsertAuthorSuffix (context . Background (), InsertAuthorSuffixParams {
207+ _ , err := q .InsertAuthorSuffix (t . Context (), InsertAuthorSuffixParams {
208208 FirstName : "george" ,
209209 LastName : "washington" ,
210210 Suffix : "Jr." ,
211211 })
212212 require .NoError (t , err )
213213
214214 t .Run ("ArrayAggFirstName - null" , func (t * testing.T ) {
215- firstNames , err := q .ArrayAggFirstName (context . Background (), 999 )
215+ firstNames , err := q .ArrayAggFirstName (t . Context (), 999 )
216216 require .NoError (t , err )
217217 require .Nil (t , firstNames )
218218 })
219219
220220 t .Run ("ArrayAggFirstName - one" , func (t * testing.T ) {
221- firstNames , err := q .ArrayAggFirstName (context . Background (), washingtonID )
221+ firstNames , err := q .ArrayAggFirstName (t . Context (), washingtonID )
222222 require .NoError (t , err )
223223 assert .Equal (t , []string {"george" }, firstNames )
224224 })
225225}
226226
227227func insertAuthor (t * testing.T , q * DBQuerier , first , last string ) int32 {
228228 t .Helper ()
229- authorID , err := q .InsertAuthor (context . Background (), first , last )
229+ authorID , err := q .InsertAuthor (t . Context (), first , last )
230230 require .NoError (t , err , "insert author" )
231231 return authorID
232232}
0 commit comments