diff options
Diffstat (limited to 'sqlr_test.go')
| -rw-r--r-- | sqlr_test.go | 42 |
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) } |
