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

examples/booktest/sqlite/db_test.go

@@ -0,0 +1,163 @@
+//go:build examples
+
+package booktest
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/sqlc-dev/sqlc/internal/sqltest"
+)
+
+// TODO: Enum is not yet supported
+const (
+	BooksBookTypeFICTION    string = "FICTION"
+	BooksBookTypeNONFICTION string = "NONFICTION"
+)
+
+func TestBooks(t *testing.T) {
+	db, cleanup := sqltest.SQLite(t, []string{"schema.sql"})
+	defer db.Close()
+	defer cleanup()
+
+	ctx := context.Background()
+	dq := New(db)
+
+	// create an author
+	a, err := dq.CreateAuthor(ctx, "Unknown Master")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// create transaction
+	tx, err := db.Begin()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	tq := dq.WithTx(tx)
+
+	// save first book
+	now := time.Now()
+	_, err = tq.CreateBook(ctx, CreateBookParams{
+		AuthorID:  a.AuthorID,
+		Isbn:      "1",
+		Title:     "my book title",
+		BookType:  BooksBookTypeFICTION,
+		Yr:        2016,
+		Available: now,
+		Tag:       "",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// save second book
+	b1, err := tq.CreateBook(ctx, CreateBookParams{
+		AuthorID:  a.AuthorID,
+		Isbn:      "2",
+		Title:     "the second book",
+		BookType:  BooksBookTypeFICTION,
+		Yr:        2016,
+		Available: now,
+		Tag:       "unique",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// update the title and tags
+	err = tq.UpdateBook(ctx, UpdateBookParams{
+		BookID: b1.BookID,
+		Title:  "changed second title",
+		Tag:    "disastor",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// save third book
+	_, err = tq.CreateBook(ctx, CreateBookParams{
+		AuthorID:  a.AuthorID,
+		Isbn:      "3",
+		Title:     "the third book",
+		BookType:  BooksBookTypeFICTION,
+		Yr:        2001,
+		Available: now,
+		Tag:       "cool",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// save fourth book
+	b3, err := tq.CreateBook(ctx, CreateBookParams{
+		AuthorID:  a.AuthorID,
+		Isbn:      "4",
+		Title:     "4th place finisher",
+		BookType:  BooksBookTypeFICTION,
+		Yr:        2011,
+		Available: now,
+		Tag:       "other",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// tx commit
+	err = tx.Commit()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// upsert, changing ISBN and title
+	err = dq.UpdateBookISBN(ctx, UpdateBookISBNParams{
+		BookID: b3.BookID,
+		Isbn:   "NEW ISBN",
+		Title:  "never ever gonna finish, a quatrain",
+		Tag:    "someother",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// retrieve first book
+	books0, err := dq.BooksByTitleYear(ctx, BooksByTitleYearParams{
+		Title: "my book title",
+		Yr:    2016,
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	for _, book := range books0 {
+		t.Logf("Book %d (%s): %s available: %s\n", book.BookID, book.BookType, book.Title, book.Available.Format(time.RFC822Z))
+		author, err := dq.GetAuthor(ctx, book.AuthorID)
+		if err != nil {
+			t.Fatal(err)
+		}
+		t.Logf("Book %d author: %s\n", book.BookID, author.Name)
+	}
+
+	// find a book with either "cool" or "other" or "someother" tag
+	t.Logf("---------\nTag search results:\n")
+	res, err := dq.BooksByTags(ctx, []string{"cool", "other", "someother"})
+	if err != nil {
+		t.Fatal(err)
+	}
+	for _, ab := range res {
+		t.Logf("Book %d: '%s', Author: '%s', ISBN: '%s' Tag: '%v'\n", ab.BookID, ab.Title, ab.Name.String, ab.Isbn, ab.Tag)
+	}
+
+	// TODO: call say_hello(varchar)
+
+	// get book 4 and delete
+	b5, err := dq.GetBook(ctx, b3.BookID)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := dq.DeleteBook(ctx, b5.BookID); err != nil {
+		t.Fatal(err)
+	}
+
+}