add linter & tests

wroge · Dec 6, 2024 · 246a644 · 246a644
1 parent f3cf4de
commit 246a644
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 1 deletion.
diff --git a/.github/workflows/go.yml b/.github/workflows/go.yml
@@ -23,7 +23,7 @@ jobs:
     - name: golangci-lint
       uses: golangci/golangci-lint-action@v6
       with:
-        version: v1.60
+        version: v1.62
 
     - name: Test
       run: go test -coverprofile=coverage.txt ./...

diff --git a/sqlt_test.go b/sqlt_test.go
@@ -2,6 +2,7 @@ package sqlt_test
 
 import (
 	"context"
+	"errors"
 	"testing"
 
 	"github.com/DATA-DOG/go-sqlmock"
@@ -29,6 +30,11 @@ func TestOne(t *testing.T) {
 	}
 
 	stmt := sqlt.QueryStmt[Param, Book](
+		sqlt.End(func(err error, runner *sqlt.Runner) {
+			if runner.SQL.String() != "SELECT id, title FROM books WHERE title = ?" {
+				t.Fail()
+			}
+		}),
 		sqlt.Parse(`
 			SELECT
 				{{ ScanInt64 Dest.ID "id" }}
@@ -47,6 +53,43 @@ func TestOne(t *testing.T) {
 	}
 }
 
+func TestOneError(t *testing.T) {
+	db, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	mock.ExpectQuery("SELECT id, title FROM books WHERE title = ?").WithArgs("TEST").WillReturnError(errors.New("ERROR"))
+
+	type Param struct {
+		Title string
+	}
+
+	type Book struct {
+		ID    int64
+		Title string
+	}
+
+	stmt := sqlt.QueryStmt[Param, Book](
+		sqlt.End(func(err error, runner *sqlt.Runner) {
+			if err == nil || err.Error() != "ERROR" {
+				t.Fail()
+			}
+		}),
+		sqlt.Parse(`
+			SELECT
+				{{ ScanInt64 Dest.ID "id" }}
+				{{- ScanString Dest.Title ", title" }}
+			FROM books WHERE title = {{ .Title }}
+		`),
+	)
+
+	_, err = stmt.One(context.Background(), db, Param{Title: "TEST"})
+	if err == nil || err.Error() != "ERROR" {
+		t.Fail()
+	}
+}
+
 func TestFirst(t *testing.T) {
 	db, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))
 	if err != nil {
@@ -69,6 +112,11 @@ func TestFirst(t *testing.T) {
 	}
 
 	stmt := sqlt.QueryStmt[Param, Book](
+		sqlt.End(func(err error, runner *sqlt.Runner) {
+			if runner.SQL.String() != "SELECT id, title FROM books WHERE title = ?" {
+				t.Fail()
+			}
+		}),
 		sqlt.Parse(`
 			SELECT
 				{{ ScanInt64 Dest.ID "id" }}
@@ -109,6 +157,11 @@ func TestAll(t *testing.T) {
 	}
 
 	stmt := sqlt.QueryStmt[Param, Book](
+		sqlt.End(func(err error, runner *sqlt.Runner) {
+			if runner.SQL.String() != "SELECT id, title FROM books WHERE title = ?" {
+				t.Fail()
+			}
+		}),
 		sqlt.Parse(`
 			SELECT
 				{{ ScanInt64 Dest.ID "id" }}