add tests to examples

Author: viewsharp

examples/booktest/mysql/db_test.go

@@ -150,6 +150,23 @@ func TestBooks(t *testing.T) {
 		t.Logf("Book %d author: %s\n", book.BookID, author.Name)
 	}
 
+	// retrieve first book
+	rows := dq.IterBooksByTitleYear(ctx, IterBooksByTitleYearParams{
+		Title: "my book title",
+		Yr:    2016,
+	})
+	for book := range rows.Iterate() {
+		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)
+	}
+	if err = rows.Err(); err != nil {
+		t.Fatal(err)
+	}
+
 	// find a book with either "cool" or "other" tag
 	t.Logf("---------\nTag search results:\n")
 	res, err := dq.BooksByTags(ctx, "cool")

examples/booktest/postgresql/db_test.go

@@ -139,6 +139,23 @@ func TestBooks(t *testing.T) {
 		t.Logf("Book %d author: %s\n", book.BookID, author.Name)
 	}
 
+	// retrieve first book
+	rows := dq.IterBooksByTitleYear(ctx, IterBooksByTitleYearParams{
+		Title: "my book title",
+		Year:  2016,
+	})
+	for book := range rows.Iterate() {
+		t.Logf("Book %d (%s): %s available: %s\n", book.BookID, book.BookType, book.Title, book.Available.Time.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)
+	}
+	if err = rows.Err(); err != nil {
+		t.Fatal(err)
+	}
+
 	// find a book with either "cool" or "other" tag
 	t.Logf("---------\nTag search results:\n")
 	res, err := dq.BooksByTags(ctx, []string{"cool", "other", "someother"})

examples/booktest/sqlite/db_test.go

@@ -138,6 +138,23 @@ func TestBooks(t *testing.T) {
 		t.Logf("Book %d author: %s\n", book.BookID, author.Name)
 	}
 
+	// retrieve first book
+	rows := dq.IterBooksByTitleYear(ctx, IterBooksByTitleYearParams{
+		Title: "my book title",
+		Yr:    2016,
+	})
+	for book := range rows.Iterate() {
+		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)
+	}
+	if err = rows.Err(); err != nil {
+		t.Fatal(err)
+	}
+
 	// 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"})

examples/ondeck/mysql/db_test.go

@@ -91,6 +91,20 @@ func runOnDeckQueries(t *testing.T, q *Queries) {
 		}
 	}
 
+	{
+		var actualCities []City
+		rows := q.IterCities(ctx)
+		for actualCity := range rows.Iterate() {
+			actualCities = append(actualCities, actualCity)
+		}
+		if err = rows.Err(); err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actualCities, []City{city}); diff != "" {
+			t.Errorf("iter city mismatch:\n%s", diff)
+		}
+	}
+
 	{
 		actual, err := q.ListVenues(ctx, city.Slug)
 		if err != nil {
@@ -101,6 +115,20 @@ func runOnDeckQueries(t *testing.T, q *Queries) {
 		}
 	}
 
+	{
+		var actualVenues []Venue
+		rows := q.IterVenues(ctx, city.Slug)
+		for actualVenue := range rows.Iterate() {
+			actualVenues = append(actualVenues, actualVenue)
+		}
+		if err = rows.Err(); err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actualVenues, []Venue{venue}); diff != "" {
+			t.Errorf("iter venue mismatch:\n%s", diff)
+		}
+	}
+
 	{
 		err := q.UpdateCityName(ctx, UpdateCityNameParams{
 			Slug: city.Slug,

examples/ondeck/postgresql/db_test.go

@@ -81,6 +81,20 @@ func runOnDeckQueries(t *testing.T, q *Queries) {
 		}
 	}
 
+	{
+		var actualCities []City
+		rows := q.IterCities(ctx)
+		for actualCity := range rows.Iterate() {
+			actualCities = append(actualCities, actualCity)
+		}
+		if err = rows.Err(); err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actualCities, []City{city}); diff != "" {
+			t.Errorf("iter city mismatch:\n%s", diff)
+		}
+	}
+
 	{
 		actual, err := q.ListVenues(ctx, city.Slug)
 		if err != nil {
@@ -91,6 +105,20 @@ func runOnDeckQueries(t *testing.T, q *Queries) {
 		}
 	}
 
+	{
+		var actualVenues []Venue
+		rows := q.IterVenues(ctx, city.Slug)
+		for actualVenue := range rows.Iterate() {
+			actualVenues = append(actualVenues, actualVenue)
+		}
+		if err = rows.Err(); err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actualVenues, []Venue{venue}); diff != "" {
+			t.Errorf("iter venue mismatch:\n%s", diff)
+		}
+	}
+
 	{
 		err := q.UpdateCityName(ctx, UpdateCityNameParams{
 			Slug: city.Slug,

examples/ondeck/sqlite/db_test.go

@@ -96,6 +96,20 @@ func runOnDeckQueries(t *testing.T, q *Queries) {
 		}
 	}
 
+	{
+		var actualCities []City
+		rows := q.IterCities(ctx)
+		for actualCity := range rows.Iterate() {
+			actualCities = append(actualCities, actualCity)
+		}
+		if err = rows.Err(); err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actualCities, []City{city}); diff != "" {
+			t.Errorf("iter city mismatch:\n%s", diff)
+		}
+	}
+
 	{
 		actual, err := q.ListVenues(ctx, city.Slug)
 		if err != nil {
@@ -106,6 +120,20 @@ func runOnDeckQueries(t *testing.T, q *Queries) {
 		}
 	}
 
+	{
+		var actualVenues []Venue
+		rows := q.IterVenues(ctx, city.Slug)
+		for actualVenue := range rows.Iterate() {
+			actualVenues = append(actualVenues, actualVenue)
+		}
+		if err = rows.Err(); err != nil {
+			t.Error(err)
+		}
+		if diff := cmp.Diff(actualVenues, []Venue{venue}); diff != "" {
+			t.Errorf("iter venue mismatch:\n%s", diff)
+		}
+	}
+
 	{
 		err := q.UpdateCityName(ctx, UpdateCityNameParams{
 			Slug: city.Slug,