summaryrefslogtreecommitdiffstats
path: root/sqlr_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'sqlr_test.go')
-rw-r--r--sqlr_test.go42
1 files changed, 4 insertions, 38 deletions
diff --git a/sqlr_test.go b/sqlr_test.go
index f2ebfbf..34ed8d5 100644
--- a/sqlr_test.go
+++ b/sqlr_test.go
@@ -4,6 +4,7 @@ import (
"context"
"database/sql"
"fmt"
+ "log"
"reflect"
"testing"
@@ -178,53 +179,18 @@ func TestScan_Errors(t *testing.T) {
}
}
-func TestQuery(t *testing.T) {
- ctx := context.Background()
- db, err := sql.Open("sqlite3", ":memory:")
- if err != nil {
- t.Fatalf("Failed to create in-memory database: %s", err)
- }
- defer db.Close()
- if _, err := db.ExecContext(ctx, `
- CREATE TABLE Tbl (Name, HighScore);
- INSERT INTO Tbl VALUES ("rose", 100)
- `); err != nil {
- t.Fatalf("Failed to initialize test database: %s", err)
- }
-
- type row struct {
- Name string
- HighScore int
- }
- const query = "SELECT Name, HighScore FROM Tbl"
- var got []row
- for row, err := range sqlr.Query[row](ctx, db, query) {
- if err != nil {
- t.Fatalf("Query(%q) failed: %s", query, err)
- }
- got = append(got, row)
- }
- want := []row{{
- Name: "rose",
- HighScore: 100,
- }}
- if diff := cmp.Diff(want, got); diff != "" {
- t.Errorf("Query(%q) returned unexpected diff (-want +got):\n%s", query, diff)
- }
-}
-
func ExampleQuery() {
ctx := context.Background()
db, err := sql.Open("sqlite3", ":memory:")
if err != nil {
- panic(err)
+ log.Fatalf("Failed to create in-memory database: %s", err)
}
defer db.Close()
if _, err := db.ExecContext(ctx, `
CREATE TABLE Tbl (Name, HighScore);
INSERT INTO Tbl VALUES ("rose", 100)
`); err != nil {
- panic(err)
+ log.Fatalf("Failed to initialize database: %s", err)
}
type row struct {
@@ -233,7 +199,7 @@ func ExampleQuery() {
}
for row, err := range sqlr.Query[row](ctx, db, "SELECT Name, HighScore FROM Tbl") {
if err != nil {
- panic(err)
+ log.Fatal(err)
}
fmt.Printf("%s %d\n", row.Name, row.HighScore)
}