Broke the build catastrophically.

Author: lizbaron

Source/tSQLt.FakeTable.ssp.sql

@@ -49,7 +49,7 @@ BEGIN
 
    EXEC tSQLt.Private_CreateFakeOfTable @SchemaName, @TableName, @OrigTableFullName, @Identity, @ComputedColumns, @Defaults;
 
-   EXEC tSQLt.Private_MarktSQLtTempObject @SchemaName, @TableName, @NewNameOfOriginalTable;
+   EXEC tSQLt.Private_MarktSQLtTempObject @OrigTableFullName, N'TABLE', @NewNameOfOriginalTable;
 END
 ---Build-
 GO

Source/tSQLt.Private_MarktSQLtTempObject.ssp.sql

@@ -2,25 +2,32 @@ IF OBJECT_ID('tSQLt.Private_MarktSQLtTempObject') IS NOT NULL DROP PROCEDURE tSQ
 GO
 ---Build+
 CREATE PROCEDURE tSQLt.Private_MarktSQLtTempObject
-  @SchemaName NVARCHAR(MAX),
-  @TableName NVARCHAR(MAX),
-  @NewNameOfOriginalTable NVARCHAR(4000)
+  @ObjectName NVARCHAR(MAX),
+  @ObjectType NVARCHAR(MAX),
+  --@ParentObjectName NVARCHAR(MAX) = NULL,
+  --@ParentObjectType NVARCHAR(MAX) = NULL,
+  @NewNameOfOriginalObject NVARCHAR(4000)
 AS
 BEGIN
-   DECLARE @UnquotedSchemaName NVARCHAR(MAX);SET @UnquotedSchemaName = OBJECT_SCHEMA_NAME(OBJECT_ID(@SchemaName+'.'+@TableName));
-   DECLARE @UnquotedTableName NVARCHAR(MAX);SET @UnquotedTableName = OBJECT_NAME(OBJECT_ID(@SchemaName+'.'+@TableName));
+   DECLARE @UnquotedSchemaName NVARCHAR(MAX);
+   DECLARE @UnquotedObjectName NVARCHAR(MAX);
+   SELECT 
+       @UnquotedSchemaName = SCHEMA_NAME(O.schema_id),
+       @UnquotedObjectName = O.name
+     FROM sys.objects O 
+    WHERE O.object_id = OBJECT_ID(@ObjectName);
 
    EXEC sys.sp_addextendedproperty 
       @name = N'tSQLt.IsTempObject',
       @value = 1, 
       @level0type = N'SCHEMA', @level0name = @UnquotedSchemaName, 
-      @level1type = N'TABLE',  @level1name = @UnquotedTableName;   
+      @level1type = @ObjectType,  @level1name = @UnquotedObjectName;   
 
    EXEC sys.sp_addextendedproperty 
       @name = N'tSQLt.Private_TestDouble_OrgObjectName', 
-      @value = @NewNameOfOriginalTable, 
+      @value = @NewNameOfOriginalObject, 
       @level0type = N'SCHEMA', @level0name = @UnquotedSchemaName, 
-      @level1type = N'TABLE',  @level1name = @UnquotedTableName;
+      @level1type = @ObjectType,  @level1name = @UnquotedObjectName;
 END;
 ---Build-
 GO

Source/tSQLt.SpyProcedure.ssp.sql

@@ -25,13 +25,16 @@ BEGIN
            @CreateProcedureStatement = @CreateProcedureStatement OUT,
            @CreateLogTableStatement = @CreateLogTableStatement OUT;
 
+    DECLARE @NewNameOfOriginalObject NVARCHAR(MAX);
 
-    EXEC tSQLt.Private_RenameObjectToUniqueNameUsingObjectId @ProcedureObjectId;
+    EXEC tSQLt.Private_RenameObjectToUniqueNameUsingObjectId @ProcedureObjectId, @NewName = @NewNameOfOriginalObject OUTPUT;
 
     EXEC(@CreateLogTableStatement);
 
     EXEC(@CreateProcedureStatement);
 
+    EXEC tSQLt.Private_MarktSQLtTempObject @ProcedureName, 'PROCEDURE', @NewNameOfOriginalObject;
+
     RETURN 0;
 END;
 ---Build-

Tests/SpyProcedureTests.class.sql

@@ -734,3 +734,26 @@ BEGIN
 
 END;
 GO
+CREATE PROC SpyProcedureTests.[test new Procedure Spy is marked as tSQLt.IsTempObject]
+AS
+BEGIN
+  EXEC('CREATE PROCEDURE SpyProcedureTests.TempProcedure1 AS RETURN;');
+  
+  EXEC tSQLt.SpyProcedure @ProcedureName = 'SpyProcedureTests.TempProcedure1';
+  
+  SELECT name, value 
+    INTO #Actual
+    FROM sys.extended_properties
+   WHERE class_desc = 'OBJECT_OR_COLUMN'
+     AND major_id = OBJECT_ID('SpyProcedureTests.TempProcedure1')
+     AND name = 'tSQLt.IsTempObject';
+
+  SELECT TOP(0) A.* INTO #Expected FROM #Actual A RIGHT JOIN #Actual X ON 1=0;
+  
+  INSERT INTO #Expected VALUES('tSQLt.IsTempObject',	1);
+
+  EXEC tSQLt.AssertEqualsTable '#Expected','#Actual';
+END;
+GO
+
+