Eli Ribble
e7ece52c8c
It's...way too many rows, now that I have the actual data. It needs to be reworked.
111 lines
2.7 KiB
Go
111 lines
2.7 KiB
Go
// Code generated by BobGen psql v0.41.1. DO NOT EDIT.
|
|
// This file is meant to be re-generated in place and/or deleted at any time.
|
|
|
|
package sql
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/stephenafamo/bob"
|
|
"github.com/stephenafamo/bob/dialect/psql"
|
|
testutils "github.com/stephenafamo/bob/test/utils"
|
|
)
|
|
|
|
func TestTrapCountByLocationID(t *testing.T) {
|
|
t.Run("Base", func(t *testing.T) {
|
|
var sb strings.Builder
|
|
|
|
query := TrapCountByLocationID(random_int32(nil), []string{random_string(nil)})
|
|
|
|
if _, err := query.WriteQuery(t.Context(), &sb, 1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if diff := cmp.Diff(trapCountByLocationIDSQL, sb.String()); diff != "" {
|
|
t.Fatalf("unexpected result (-got +want):\n%s", diff)
|
|
}
|
|
})
|
|
|
|
t.Run("Mod", func(t *testing.T) {
|
|
var sb strings.Builder
|
|
|
|
query := TrapCountByLocationID(random_int32(nil), []string{random_string(nil)})
|
|
|
|
if _, err := psql.Select(query).WriteQuery(t.Context(), &sb, 1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
queryDiff, err := testutils.QueryDiff(trapCountByLocationIDSQL, sb.String(), formatQuery)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if queryDiff != "" {
|
|
fmt.Println(sb.String())
|
|
t.Fatalf("unexpected result (-got +want):\n%s", queryDiff)
|
|
}
|
|
})
|
|
|
|
t.Run("Scanning", func(t *testing.T) {
|
|
if testDB == nil {
|
|
t.Skip("skipping test, no DSN provided")
|
|
}
|
|
|
|
ctxTx, cancel := context.WithCancel(t.Context())
|
|
defer cancel()
|
|
|
|
tx, err := testDB.Begin(ctxTx)
|
|
if err != nil {
|
|
t.Fatalf("Error starting transaction: %v", err)
|
|
}
|
|
|
|
defer func() {
|
|
if err := tx.Rollback(ctxTx); err != nil {
|
|
t.Fatalf("Error rolling back transaction: %v", err)
|
|
}
|
|
}()
|
|
|
|
query, args, err := bob.Build(ctxTx, psql.Select(TrapCountByLocationID(random_int32(nil), []string{random_string(nil)})))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
rows, err := tx.QueryContext(ctxTx, query, args...)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
columns, err := rows.Columns()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(columns) != 5 {
|
|
t.Fatalf("expected %d columns, got %d", 5, len(columns))
|
|
}
|
|
|
|
if columns[0] != "trapdata_globalid" {
|
|
t.Fatalf("expected column %d to be %s, got %s", 0, "trapdata_globalid", columns[0])
|
|
}
|
|
|
|
if columns[1] != "trapdata_enddate" {
|
|
t.Fatalf("expected column %d to be %s, got %s", 1, "trapdata_enddate", columns[1])
|
|
}
|
|
|
|
if columns[2] != "total_females" {
|
|
t.Fatalf("expected column %d to be %s, got %s", 2, "total_females", columns[2])
|
|
}
|
|
|
|
if columns[3] != "total_males" {
|
|
t.Fatalf("expected column %d to be %s, got %s", 3, "total_males", columns[3])
|
|
}
|
|
|
|
if columns[4] != "total" {
|
|
t.Fatalf("expected column %d to be %s, got %s", 4, "total", columns[4])
|
|
}
|
|
})
|
|
}
|