From 8484badeb463efb62bfec7bca01b79bf78d233e5 Mon Sep 17 00:00:00 2001
From: Yuki Takizawa <y.takizawa64@gmail.com>
Date: Sat, 2 Mar 2024 00:02:49 +0900
Subject: [PATCH] Call say_hello function for test

---
 examples/booktest/postgresql/db_test.go   | 13 ++++++++++++-
 examples/booktest/postgresql/query.sql    |  3 +++
 examples/booktest/postgresql/query.sql.go | 11 +++++++++++
 examples/booktest/postgresql/schema.sql   |  4 ++--
 4 files changed, 28 insertions(+), 3 deletions(-)

diff --git a/examples/booktest/postgresql/db_test.go b/examples/booktest/postgresql/db_test.go
index e33ee1b602..2198a4bf14 100644
--- a/examples/booktest/postgresql/db_test.go
+++ b/examples/booktest/postgresql/db_test.go
@@ -150,7 +150,18 @@ func TestBooks(t *testing.T) {
 		t.Logf("Book %d: '%s', Author: '%s', ISBN: '%s' Tags: '%v'\n", ab.BookID, ab.Title, ab.Name.String, ab.Isbn, ab.Tags)
 	}
 
-	// TODO: call say_hello(varchar)
+	// call function
+	pgText, err := dq.SayHello(ctx, "world")
+	if err != nil {
+		t.Fatal(err)
+	}
+	str, err := pgText.Value()
+	if err != nil {
+		t.Fatal(err)
+	}
+	if str != "hello world" {
+		t.Fatal("expected function result to be \"hello world\". actual:", str)
+	}
 
 	// get book 4 and delete
 	b5, err := dq.GetBook(ctx, b3.BookID)
diff --git a/examples/booktest/postgresql/query.sql b/examples/booktest/postgresql/query.sql
index 194897a4bb..75b8cd3b6d 100644
--- a/examples/booktest/postgresql/query.sql
+++ b/examples/booktest/postgresql/query.sql
@@ -58,3 +58,6 @@ WHERE book_id = $3;
 UPDATE books
 SET title = $1, tags = $2, isbn = $4
 WHERE book_id = $3;
+
+-- name: SayHello :one
+select * from say_hello($1);
diff --git a/examples/booktest/postgresql/query.sql.go b/examples/booktest/postgresql/query.sql.go
index 998e0241e0..c39710403e 100644
--- a/examples/booktest/postgresql/query.sql.go
+++ b/examples/booktest/postgresql/query.sql.go
@@ -206,6 +206,17 @@ func (q *Queries) GetBook(ctx context.Context, bookID int32) (Book, error) {
 	return i, err
 }
 
+const sayHello = `-- name: SayHello :one
+select say_hello from say_hello($1)
+`
+
+func (q *Queries) SayHello(ctx context.Context, s string) (pgtype.Text, error) {
+	row := q.db.QueryRow(ctx, sayHello, s)
+	var say_hello pgtype.Text
+	err := row.Scan(&say_hello)
+	return say_hello, err
+}
+
 const updateBook = `-- name: UpdateBook :exec
 UPDATE books
 SET title = $1, tags = $2
diff --git a/examples/booktest/postgresql/schema.sql b/examples/booktest/postgresql/schema.sql
index 2beecaba1a..3cdd97ca5f 100644
--- a/examples/booktest/postgresql/schema.sql
+++ b/examples/booktest/postgresql/schema.sql
@@ -23,9 +23,9 @@ CREATE TABLE books (
 
 CREATE INDEX books_title_idx ON books(title, year);
 
-CREATE FUNCTION say_hello(text) RETURNS text AS $$
+CREATE FUNCTION say_hello(s text) RETURNS text AS $$
 BEGIN
-          RETURN CONCAT('hello ', $1);
+          RETURN CONCAT('hello ', s);
 END;
 $$ LANGUAGE plpgsql;