Fix and test for rowcount

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;

sqlparser/parser.go

@@ -417,7 +417,7 @@ func (d *Document) parseCreate(s *Scanner, createCountInBatch int) (result Creat
 	// point we copy the rest until the batch ends; *but* track dependencies
 	// + some other details mentioned below
 
-	firstAs := true
+	//firstAs := true // See comment below on rowcount
 
 tailloop:
 	for {
@@ -473,18 +473,41 @@ tailloop:
 		case tt == ReservedWordToken && s.Token() == "as":
 			CopyToken(s, &result.Body)
 			NextTokenCopyingWhitespace(s, &result.Body)
-			if firstAs {
-				// Add the `RoutineName` token as a convenience, so that we can refer to the procedure/function name
-				// from inside the procedure (for example, when logging)
-				if result.CreateType == "procedure" {
-					procNameToken := Unparsed{
-						Type:     OtherToken,
-						RawValue: fmt.Sprintf(templateRoutineName, strings.Trim(result.QuotedName.Value, "[]")),
+			/*
+					    TODO: Fix and re-enable
+						This code add RoutineName for convenience.  So:
+
+						create procedure [code@5420c0269aaf].Test as
+						begin
+							select 1
+						end
+						go
+
+						becomes:
+
+						create procedure [code@5420c0269aaf].Test as
+						declare @RoutineName nvarchar(128)
+						set @RoutineName = 'Test'
+						begin
+							select 1
+						end
+						go
+
+						However, for some very strange reason, @@rowcount is 1 with the first version,
+						and it is 2 with the second version.
+				if firstAs {
+					// Add the `RoutineName` token as a convenience, so that we can refer to the procedure/function name
+					// from inside the procedure (for example, when logging)
+					if result.CreateType == "procedure" {
+						procNameToken := Unparsed{
+							Type:     OtherToken,
+							RawValue: fmt.Sprintf(templateRoutineName, strings.Trim(result.QuotedName.Value, "[]")),
+						}
+						result.Body = append(result.Body, procNameToken)
 					}
-					result.Body = append(result.Body, procNameToken)
+					firstAs = false
 				}
-				firstAs = false
-			}
+			*/
 
 		default:
 			CopyToken(s, &result.Body)

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,26 @@
+package sqltest
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"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(`[code].Test`)
+
+	res, err := fixture.DB.ExecContext(ctx, patched)
+	require.NoError(t, err)
+	rowsAffected, err := res.RowsAffected()
+	require.NoError(t, err)
+	assert.Equal(t, int64(1), rowsAffected)
+}

sqltest/test.sql

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