snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

examples/ondeck/postgresql/db_test.go

@@ -0,0 +1,154 @@
+//go:build examples
+
+package ondeck
+
+import (
+	"context"
+	"database/sql"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	_ "github.com/lib/pq"
+
+	"github.com/sqlc-dev/sqlc/internal/sqltest/local"
+)
+
+func runOnDeckQueries(t *testing.T, q *Queries) {
+	ctx := context.Background()
+
+	city, err := q.CreateCity(ctx, CreateCityParams{
+		Slug: "san-francisco",
+		Name: "San Francisco",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	venueID, err := q.CreateVenue(ctx, CreateVenueParams{
+		Slug:            "the-fillmore",
+		Name:            "The Fillmore",
+		City:            city.Slug,
+		SpotifyPlaylist: "spotify:uri",
+		Status:          StatusOpen,
+		Statuses:        []Status{StatusOpen, StatusClosed},
+		Tags:            []string{"rock", "punk"},
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	venue, err := q.GetVenue(ctx, GetVenueParams{
+		Slug: "the-fillmore",
+		City: city.Slug,
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if diff := cmp.Diff(venue.ID, venueID); diff != "" {
+		t.Errorf("venue ID mismatch:\n%s", diff)
+	}
+
+	{
+		actual, err := q.GetCity(ctx, city.Slug)
+		if err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actual, city); diff != "" {
+			t.Errorf("get city mismatch:\n%s", diff)
+		}
+	}
+
+	{
+		actual, err := q.VenueCountByCity(ctx)
+		if err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actual, []VenueCountByCityRow{
+			{city.Slug, 1},
+		}); diff != "" {
+			t.Errorf("venue county mismatch:\n%s", diff)
+		}
+	}
+
+	{
+		actual, err := q.ListCities(ctx)
+		if err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actual, []City{city}); diff != "" {
+			t.Errorf("list city mismatch:\n%s", diff)
+		}
+	}
+
+	{
+		actual, err := q.ListVenues(ctx, city.Slug)
+		if err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actual, []Venue{venue}); diff != "" {
+			t.Errorf("list venue mismatch:\n%s", diff)
+		}
+	}
+
+	{
+		err := q.UpdateCityName(ctx, UpdateCityNameParams{
+			Slug: city.Slug,
+			Name: "SF",
+		})
+		if err != nil {
+			t.Error(err)
+		}
+	}
+
+	{
+		id, err := q.UpdateVenueName(ctx, UpdateVenueNameParams{
+			Slug: venue.Slug,
+			Name: "Fillmore",
+		})
+		if err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(id, venue.ID); diff != "" {
+			t.Errorf("update venue mismatch:\n%s", diff)
+		}
+	}
+
+	{
+		err := q.DeleteVenue(ctx, venue.Slug)
+		if err != nil {
+			t.Error(err)
+		}
+	}
+}
+
+func TestPrepared(t *testing.T) {
+	t.Parallel()
+
+	uri := local.PostgreSQL(t, []string{"schema"})
+	db, err := sql.Open("postgres", uri)
+	if err != nil {
+		t.Fatalf("%s: %s", uri, err)
+	}
+	defer db.Close()
+
+	q, err := Prepare(context.Background(), db)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	runOnDeckQueries(t, q)
+}
+
+func TestQueries(t *testing.T) {
+	t.Parallel()
+
+	uri := local.PostgreSQL(t, []string{"schema"})
+	db, err := sql.Open("postgres", uri)
+	if err != nil {
+		t.Fatalf("%s: %s", uri, err)
+	}
+	defer db.Close()
+
+	runOnDeckQueries(t, New(db))
+}