Test for rowcount

Author: dfava

migrations/0001.sqlcode.sql

@@ -132,7 +132,4 @@ grant execute on sqlcode.CreateCodeSchema to [sqlcode-deploy-role];
 grant execute on sqlcode.DropCodeSchema to [sqlcode-deploy-role];
 grant create procedure to [sqlcode-deploy-role];
 grant create function to [sqlcode-deploy-role];
-grant create type to [sqlcode-deploy-role];
-
-
-alter user
+grant create type to [sqlcode-deploy-role];

migrations/0002.sqlcode.sql

@@ -1,5 +1,5 @@
 -- We want to re-create the procedure DropCodeSchema.
--- Frist, everything related to this procedure must be dropped
+-- First, everything related to this procedure must be dropped
 -- before it is re-created in the end.
 
 drop procedure sqlcode.DropCodeSchema;

sqltest/sql.go

@@ -0,0 +1,15 @@
+package sqltest
+
+import (
+	"embed"
+
+	"github.com/vippsas/sqlcode"
+)
+
+//go:embed *.sql
+var sqlfs embed.FS
+
+var SQL = sqlcode.MustInclude(
+	sqlcode.Options{},
+	sqlfs,
+)

sqltest/sqlcode_test.go

@@ -0,0 +1,23 @@
+package sqltest
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func Test_RowsAffected(t *testing.T) {
+	fixture := NewFixture()
+	defer fixture.Teardown()
+	fixture.RunMigrationFile("../migrations/0001.sqlcode.sql")
+
+	ctx := context.Background()
+
+	require.NoError(t, SQL.EnsureUploaded(ctx, fixture.DB))
+	patched := SQL.Patch(`select [code].Test`)
+
+	res, err := fixture.DB.ExecContext(ctx, patched)
+	require.NoError(t, err)
+	println(res.RowsAffected())
+}

sqltest/test.sql

@@ -0,0 +1,4 @@
+create procedure [code].Test as
+begin
+select 1
+end