aboutsummaryrefslogtreecommitdiff
path: root/select_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'select_test.go')
-rw-r--r--select_test.go88
1 files changed, 48 insertions, 40 deletions
diff --git a/select_test.go b/select_test.go
index f38fbdd..d4b1970 100644
--- a/select_test.go
+++ b/select_test.go
@@ -16,6 +16,7 @@ import (
)
func TestSelectReturningSomeRecords(t *testing.T) {
+ ctx := t.Context()
md := mock.NewDriver()
db := sql.OpenDB(md)
@@ -35,7 +36,7 @@ func TestSelectReturningSomeRecords(t *testing.T) {
AndReturnColumns("name", "id").
WithRow("foo", 1).
WithRow("bar", 2)
- records := must.Return(store.Select(db, `SELECT * FROM basic_records WHERE id < ?`, 3))(t)
+ records := must.Return(store.Select(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3))(t)
assert.SliceEqual(t, records,
basicRecord{1, "foo"},
basicRecord{2, "bar"},
@@ -48,7 +49,7 @@ func TestSelectReturningSomeRecords(t *testing.T) {
AndReturnColumns("id", "name").
WithRow(1, "ffoo").
WithRow(2, "bbar")
- records := must.Return(store.SelectWhere(db, `id < ?`, 3))(t)
+ records := must.Return(store.SelectWhere(ctx, db, `id < ?`, 3))(t)
assert.SliceEqual(t, records,
basicRecord{1, "ffoo"},
basicRecord{2, "bbar"},
@@ -62,7 +63,7 @@ func TestSelectReturningSomeRecords(t *testing.T) {
WithRow(1, "fffoo").
WithRow(2, "bbbar")
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- records := must.Return(query.Select(db, 3))(t)
+ records := must.Return(query.Select(ctx, db, 3))(t)
assert.SliceEqual(t, records,
basicRecord{1, "fffoo"},
basicRecord{2, "bbbar"},
@@ -75,7 +76,7 @@ func TestSelectReturningSomeRecords(t *testing.T) {
AndReturnColumns("name", "id").
WithRow("ffffoo", 1).
WithRow("bbbbar", 2)
- record := must.Return(store.SelectOne(db, `SELECT * FROM basic_records WHERE id < ?`, 3))(t)
+ record := must.Return(store.SelectOne(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3))(t)
assert.Equal(t, record, basicRecord{1, "ffffoo"})
})
@@ -85,7 +86,7 @@ func TestSelectReturningSomeRecords(t *testing.T) {
AndReturnColumns("id", "name").
WithRow(1, "fffffoo").
WithRow(2, "bbbbbar")
- record := must.Return(store.SelectOneWhere(db, `id < ?`, 3))(t)
+ record := must.Return(store.SelectOneWhere(ctx, db, `id < ?`, 3))(t)
assert.Equal(t, record, basicRecord{1, "fffffoo"})
})
@@ -96,12 +97,13 @@ func TestSelectReturningSomeRecords(t *testing.T) {
WithRow(1, "ffffffoo").
WithRow(2, "bbbbbbar")
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- record := must.Return(query.SelectOne(db, 3))(t)
+ record := must.Return(query.SelectOne(ctx, db, 3))(t)
assert.Equal(t, record, basicRecord{1, "ffffffoo"})
})
}
func TestSelectReturningNoRecords(t *testing.T) {
+ ctx := t.Context()
md := mock.NewDriver()
db := sql.OpenDB(md)
@@ -119,7 +121,7 @@ func TestSelectReturningNoRecords(t *testing.T) {
md.ForQuery(`SELECT * FROM basic_records WHERE id < ?`).
ExpectQueryWithArgs(3).
AndReturnColumns("name", "id")
- records := must.Return(store.Select(db, `SELECT * FROM basic_records WHERE id < ?`, 3))(t)
+ records := must.Return(store.Select(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3))(t)
assert.SliceEqual(t, records, nil...)
})
@@ -127,7 +129,7 @@ func TestSelectReturningNoRecords(t *testing.T) {
md.ForQuery(`SELECT "id", "name" FROM "basic_records" WHERE id < ?`).
ExpectQueryWithArgs(3).
AndReturnColumns("id", "name")
- records := must.Return(store.SelectWhere(db, `id < ?`, 3))(t)
+ records := must.Return(store.SelectWhere(ctx, db, `id < ?`, 3))(t)
assert.SliceEqual(t, records, nil...)
})
@@ -136,7 +138,7 @@ func TestSelectReturningNoRecords(t *testing.T) {
ExpectQueryWithArgs(3).
AndReturnColumns("id", "name")
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- records := must.Return(query.Select(db, 3))(t)
+ records := must.Return(query.Select(ctx, db, 3))(t)
assert.SliceEqual(t, records, nil...)
})
@@ -144,7 +146,7 @@ func TestSelectReturningNoRecords(t *testing.T) {
md.ForQuery(`SELECT * FROM basic_records WHERE id < ?`).
ExpectQueryWithArgs(3).
AndReturnColumns("name", "id")
- _, err := store.SelectOne(db, `SELECT * FROM basic_records WHERE id < ?`, 3)
+ _, err := store.SelectOne(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3)
assert.ErrEqual(t, err, sql.ErrNoRows.Error())
})
@@ -152,7 +154,7 @@ func TestSelectReturningNoRecords(t *testing.T) {
md.ForQuery(`SELECT "id", "name" FROM "basic_records" WHERE id < ?`).
ExpectQueryWithArgs(3).
AndReturnColumns("id", "name")
- _, err := store.SelectOneWhere(db, `id < ?`, 3)
+ _, err := store.SelectOneWhere(ctx, db, `id < ?`, 3)
assert.ErrEqual(t, err, sql.ErrNoRows.Error())
})
@@ -161,12 +163,13 @@ func TestSelectReturningNoRecords(t *testing.T) {
ExpectQueryWithArgs(3).
AndReturnColumns("id", "name")
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- _, err := query.SelectOne(db, 3)
+ _, err := query.SelectOne(ctx, db, 3)
assert.ErrEqual(t, err, sql.ErrNoRows.Error())
})
}
func TestSelectIntoUnexpectedField(t *testing.T) {
+ ctx := t.Context()
md := mock.NewDriver()
db := sql.OpenDB(md)
@@ -193,18 +196,19 @@ func TestSelectIntoUnexpectedField(t *testing.T) {
t.Run("using Store.Select", func(t *testing.T) {
commonSetup()
- _, err := store.Select(db, `SELECT * FROM basic_records WHERE id < ?`, 3)
+ _, err := store.Select(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3)
assert.ErrEqual(t, err, expectedError)
})
t.Run("using Store.SelectOne", func(t *testing.T) {
commonSetup()
- _, err := store.SelectOne(db, `SELECT * FROM basic_records WHERE id < ?`, 3)
+ _, err := store.SelectOne(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3)
assert.ErrEqual(t, err, expectedError)
})
}
func TestSelectWithScanError(t *testing.T) {
+ ctx := t.Context()
md := mock.NewDriver()
db := sql.OpenDB(md)
@@ -229,44 +233,45 @@ func TestSelectWithScanError(t *testing.T) {
t.Run("using Store.Select", func(t *testing.T) {
commonSetup(`SELECT * FROM basic_records WHERE id < ?`)
- _, err := store.Select(db, `SELECT * FROM basic_records WHERE id < ?`, 3)
+ _, err := store.Select(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3)
assert.ErrEqual(t, err, expectedError)
})
t.Run("using Store.SelectWhere", func(t *testing.T) {
commonSetup(`SELECT "id", "created_at" FROM "basic_records" WHERE id < ?`)
- _, err := store.SelectWhere(db, `id < ?`, 3)
+ _, err := store.SelectWhere(ctx, db, `id < ?`, 3)
assert.ErrEqual(t, err, expectedError)
})
t.Run("using PreparedSelectQuery.Select", func(t *testing.T) {
commonSetup(`SELECT "id", "created_at" FROM "basic_records" WHERE id < ?`)
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- _, err := query.Select(db, 3)
+ _, err := query.Select(ctx, db, 3)
assert.ErrEqual(t, err, expectedError)
})
t.Run("using Store.SelectOne", func(t *testing.T) {
commonSetup(`SELECT * FROM basic_records WHERE id < ?`)
- _, err := store.SelectOne(db, `SELECT * FROM basic_records WHERE id < ?`, 3)
+ _, err := store.SelectOne(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3)
assert.ErrEqual(t, err, expectedError)
})
t.Run("using Store.SelectOneWhere", func(t *testing.T) {
commonSetup(`SELECT "id", "created_at" FROM "basic_records" WHERE id < ?`)
- _, err := store.SelectOneWhere(db, `id < ?`, 3)
+ _, err := store.SelectOneWhere(ctx, db, `id < ?`, 3)
assert.ErrEqual(t, err, expectedError)
})
t.Run("using PreparedSelectQuery.SelectOne", func(t *testing.T) {
commonSetup(`SELECT "id", "created_at" FROM "basic_records" WHERE id < ?`)
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- _, err := query.SelectOne(db, 3)
+ _, err := query.SelectOne(ctx, db, 3)
assert.ErrEqual(t, err, expectedError)
})
}
func TestSelectIntoEmbeddedTypes(t *testing.T) {
+ ctx := t.Context()
md := mock.NewDriver()
db := sql.OpenDB(md)
@@ -299,7 +304,7 @@ func TestSelectIntoEmbeddedTypes(t *testing.T) {
t.Run("using Store.Select", func(t *testing.T) {
commonSetup(`SELECT * FROM composite_records`)
- records := must.Return(store.Select(db, `SELECT * FROM composite_records`))(t)
+ records := must.Return(store.Select(ctx, db, `SELECT * FROM composite_records`))(t)
assert.SliceDeepEqual(t, records,
compositeRecord{1, HasCreatedAt{time.Unix(1, 0)}, &HasUpdatedAt{new(time.Unix(3, 0))}},
compositeRecord{2, HasCreatedAt{time.Unix(2, 0)}, &HasUpdatedAt{nil}},
@@ -308,7 +313,7 @@ func TestSelectIntoEmbeddedTypes(t *testing.T) {
t.Run("using Store.SelectWhere", func(t *testing.T) {
commonSetup(`SELECT "id", "created_at", "updated_at" FROM "composite_records" WHERE TRUE`)
- records := must.Return(store.SelectWhere(db, `TRUE`))(t)
+ records := must.Return(store.SelectWhere(ctx, db, `TRUE`))(t)
assert.SliceDeepEqual(t, records,
compositeRecord{1, HasCreatedAt{time.Unix(1, 0)}, &HasUpdatedAt{new(time.Unix(3, 0))}},
compositeRecord{2, HasCreatedAt{time.Unix(2, 0)}, &HasUpdatedAt{nil}},
@@ -318,7 +323,7 @@ func TestSelectIntoEmbeddedTypes(t *testing.T) {
t.Run("using PreparedSelectQuery.Select", func(t *testing.T) {
commonSetup(`SELECT "id", "created_at", "updated_at" FROM "composite_records" WHERE TRUE`)
query := store.MustPrepareSelectQueryWhere(`TRUE`)
- records := must.Return(query.Select(db))(t)
+ records := must.Return(query.Select(ctx, db))(t)
assert.SliceDeepEqual(t, records,
compositeRecord{1, HasCreatedAt{time.Unix(1, 0)}, &HasUpdatedAt{new(time.Unix(3, 0))}},
compositeRecord{2, HasCreatedAt{time.Unix(2, 0)}, &HasUpdatedAt{nil}},
@@ -327,7 +332,7 @@ func TestSelectIntoEmbeddedTypes(t *testing.T) {
t.Run("using Store.SelectOne", func(t *testing.T) {
commonSetup(`SELECT * FROM composite_records`)
- record := must.Return(store.SelectOne(db, `SELECT * FROM composite_records`))(t)
+ record := must.Return(store.SelectOne(ctx, db, `SELECT * FROM composite_records`))(t)
assert.DeepEqual(t, record,
compositeRecord{1, HasCreatedAt{time.Unix(1, 0)}, &HasUpdatedAt{new(time.Unix(3, 0))}},
)
@@ -335,7 +340,7 @@ func TestSelectIntoEmbeddedTypes(t *testing.T) {
t.Run("using Store.SelectOneWhere", func(t *testing.T) {
commonSetup(`SELECT "id", "created_at", "updated_at" FROM "composite_records" WHERE TRUE`)
- record := must.Return(store.SelectOneWhere(db, `TRUE`))(t)
+ record := must.Return(store.SelectOneWhere(ctx, db, `TRUE`))(t)
assert.DeepEqual(t, record,
compositeRecord{1, HasCreatedAt{time.Unix(1, 0)}, &HasUpdatedAt{new(time.Unix(3, 0))}},
)
@@ -344,7 +349,7 @@ func TestSelectIntoEmbeddedTypes(t *testing.T) {
t.Run("using PreparedSelectQuery.SelectOne", func(t *testing.T) {
commonSetup(`SELECT "id", "created_at", "updated_at" FROM "composite_records" WHERE TRUE`)
query := store.MustPrepareSelectQueryWhere(`TRUE`)
- record := must.Return(query.SelectOne(db))(t)
+ record := must.Return(query.SelectOne(ctx, db))(t)
assert.DeepEqual(t, record,
compositeRecord{1, HasCreatedAt{time.Unix(1, 0)}, &HasUpdatedAt{new(time.Unix(3, 0))}},
)
@@ -352,6 +357,7 @@ func TestSelectIntoEmbeddedTypes(t *testing.T) {
}
func TestSelectCapturingQueryError(t *testing.T) {
+ ctx := t.Context()
md := mock.NewDriver()
db := sql.OpenDB(md)
@@ -366,39 +372,40 @@ func TestSelectCapturingQueryError(t *testing.T) {
)
t.Run("using Store.Select", func(t *testing.T) {
- _, err := store.Select(db, `SELECT * FROM basic_records WHERE id < ?`, 3)
+ _, err := store.Select(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3)
assert.ErrEqual(t, err, "during Query(): unexpected query: SELECT * FROM basic_records WHERE id < ?")
})
t.Run("using Store.SelectWhere", func(t *testing.T) {
- _, err := store.SelectWhere(db, `id < ?`, 3)
+ _, err := store.SelectWhere(ctx, db, `id < ?`, 3)
assert.ErrEqual(t, err, `during Query(): unexpected query: SELECT "id", "name" FROM "basic_records" WHERE id < ?`)
})
t.Run("using PreparedSelectQuery.Select", func(t *testing.T) {
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- _, err := query.Select(db, 3)
+ _, err := query.Select(ctx, db, 3)
assert.ErrEqual(t, err, `during Query(): unexpected query: SELECT "id", "name" FROM "basic_records" WHERE id < ?`)
})
t.Run("using Store.SelectOne", func(t *testing.T) {
- _, err := store.SelectOne(db, `SELECT * FROM basic_records WHERE id < ?`, 3)
+ _, err := store.SelectOne(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3)
assert.ErrEqual(t, err, "during Query(): unexpected query: SELECT * FROM basic_records WHERE id < ?")
})
t.Run("using Store.SelectOneWhere", func(t *testing.T) {
- _, err := store.SelectOneWhere(db, `id < ?`, 3)
+ _, err := store.SelectOneWhere(ctx, db, `id < ?`, 3)
assert.ErrEqual(t, err, `unexpected query: SELECT "id", "name" FROM "basic_records" WHERE id < ?`)
})
t.Run("using PreparedSelectQuery.SelectOne", func(t *testing.T) {
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- _, err := query.SelectOne(db, 3)
+ _, err := query.SelectOne(ctx, db, 3)
assert.ErrEqual(t, err, `unexpected query: SELECT "id", "name" FROM "basic_records" WHERE id < ?`)
})
}
func TestSelectCapturingCloseError(t *testing.T) {
+ ctx := t.Context()
md := mock.NewDriver()
db := sql.OpenDB(md)
@@ -423,44 +430,45 @@ func TestSelectCapturingCloseError(t *testing.T) {
t.Run("using Store.Select", func(t *testing.T) {
commonSetup(`SELECT * FROM basic_records WHERE id < ?`)
- _, err := store.Select(db, `SELECT * FROM basic_records WHERE id < ?`, 3)
+ _, err := store.Select(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3)
assert.ErrEqual(t, err, "during Rows.Err(): datacenter on fire")
})
t.Run("using Store.SelectWhere", func(t *testing.T) {
commonSetup(`SELECT "id", "name" FROM "basic_records" WHERE id < ?`)
- _, err := store.SelectWhere(db, `id < ?`, 3)
+ _, err := store.SelectWhere(ctx, db, `id < ?`, 3)
assert.ErrEqual(t, err, "during Rows.Err(): datacenter on fire")
})
t.Run("using PreparedSelectQuery.Select", func(t *testing.T) {
commonSetup(`SELECT "id", "name" FROM "basic_records" WHERE id < ?`)
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- _, err := query.Select(db, 3)
+ _, err := query.Select(ctx, db, 3)
assert.ErrEqual(t, err, "during Rows.Err(): datacenter on fire")
})
t.Run("using Store.SelectOne", func(t *testing.T) {
commonSetup(`SELECT * FROM basic_records WHERE id < ?`)
- _, err := store.SelectOne(db, `SELECT * FROM basic_records WHERE id < ?`, 3)
+ _, err := store.SelectOne(ctx, db, `SELECT * FROM basic_records WHERE id < ?`, 3)
assert.ErrEqual(t, err, "during Rows.Err(): datacenter on fire")
})
t.Run("using Store.SelectOneWhere", func(t *testing.T) {
commonSetup(`SELECT "id", "name" FROM "basic_records" WHERE id < ?`)
- _, err := store.SelectOneWhere(db, `id < ?`, 3)
+ _, err := store.SelectOneWhere(ctx, db, `id < ?`, 3)
assert.ErrEqual(t, err, "datacenter on fire")
})
t.Run("using PreparedSelectQuery.SelectOne", func(t *testing.T) {
commonSetup(`SELECT "id", "name" FROM "basic_records" WHERE id < ?`)
query := store.MustPrepareSelectQueryWhere(`id < ?`)
- _, err := query.SelectOne(db, 3)
+ _, err := query.SelectOne(ctx, db, 3)
assert.ErrEqual(t, err, "datacenter on fire")
})
}
func TestSelectNotPossibleWithoutTableName(t *testing.T) {
+ ctx := t.Context()
md := mock.NewDriver()
db := sql.OpenDB(md)
@@ -471,12 +479,12 @@ func TestSelectNotPossibleWithoutTableName(t *testing.T) {
store := oblast.MustNewStore[basicRecord](oblast.SqliteDialect())
t.Run("using Store.SelectWhere", func(t *testing.T) {
- _, err := store.SelectWhere(db, `id < ?`, 3)
+ _, err := store.SelectWhere(ctx, db, `id < ?`, 3)
assert.ErrEqual(t, err, "cannot execute SelectWhere() because query could not be autogenerated")
})
t.Run("using Store.SelectOneWhere", func(t *testing.T) {
- _, err := store.SelectOneWhere(db, `id < ?`, 3)
+ _, err := store.SelectOneWhere(ctx, db, `id < ?`, 3)
assert.ErrEqual(t, err, "cannot execute SelectOneWhere() because query could not be autogenerated")
})
generated by cgit v1.2.3 (git 2.25.1) at 2026-06-08 13:42:28 +0000