Fixed constraints! Yay. All tests passing. But the build is still broken because trigger tests are not passing.

Author: lizbaron

Source/ApplyConstraint_Methods.sql

@@ -106,11 +106,15 @@ BEGIN
   DECLARE @AlterTableCmd NVARCHAR(MAX);
   DECLARE @CreateIndexCmd NVARCHAR(MAX);
   DECLARE @FinalCmd NVARCHAR(MAX);
+  DECLARE @NewNameOfOriginalConstraint NVARCHAR(MAX);
+  DECLARE @QuotedFullConstraintName NVARCHAR(MAX);
+
 
   SELECT @SchemaName = SchemaName,
          @OrgTableName = OrgTableName,
          @TableName = TableName,
-         @ConstraintName = OBJECT_NAME(@ConstraintObjectId)
+         @ConstraintName = OBJECT_NAME(@ConstraintObjectId),
+         @QuotedFullConstraintName = QUOTENAME(SchemaName)+'.'+QUOTENAME(OBJECT_NAME(@ConstraintObjectId))
     FROM tSQLt.Private_GetQuotedTableNameForConstraint(@ConstraintObjectId);
 
   SELECT @CreateFkCmd = cmd, @CreateIndexCmd = CreIdxCmd
@@ -119,9 +123,11 @@ BEGIN
                           ' ADD ' + @CreateFkCmd;
   SELECT @FinalCmd = @CreateIndexCmd + @AlterTableCmd;
 
-  EXEC tSQLt.Private_RenameObjectToUniqueName @SchemaName, @ConstraintName;
+  EXEC tSQLt.Private_RenameObjectToUniqueName @SchemaName, @ConstraintName, @NewName = @NewNameOfOriginalConstraint OUTPUT;
   EXEC (@FinalCmd);
 
+  EXEC tSQLt.Private_MarktSQLtTempObject @ObjectName = @QuotedFullConstraintName, @ObjectType = 'CONSTRAINT', @NewNameOfOriginalObject = @NewNameOfOriginalConstraint;
+
 END;
 GO
 

Source/tSQLt.ApplyTrigger.ssp.sql

@@ -7,6 +7,8 @@ CREATE PROCEDURE tSQLt.ApplyTrigger
 AS
 BEGIN
   DECLARE @OrgTableObjectId INT;
+  DECLARE @NewNameOfOriginalTrigger NVARCHAR(MAX);
+
   SELECT @OrgTableObjectId = OrgTableObjectId FROM tSQLt.Private_GetOriginalTableInfo(OBJECT_ID(@TableName)) orgTbl
   IF(@OrgTableObjectId IS NULL)
   BEGIN
@@ -28,9 +30,11 @@ BEGIN
     RAISERROR('%s is not a trigger on %s', 16, 10, @TriggerName, @TableName);
   END;
 
-  EXEC tSQLt.RemoveObject @FullTriggerName;
+  EXEC tSQLt.RemoveObject @ObjectName = @FullTriggerName, @NewName = @NewNameOfOriginalTrigger OUTPUT;
 
   EXEC(@TriggerCode);
+
+  EXEC tSQLt.Private_MarktSQLtTempObject @ObjectName = @FullTriggerName, @ObjectType = N'TRIGGER', @NewNameOfOriginalObject = @NewNameOfOriginalTrigger;
 END;
 ---Build-
 GO

Tests/ApplyTriggerTests.class.sql

@@ -120,3 +120,29 @@ BEGIN
 
 END;
 GO
+CREATE PROC ApplyTriggerTests.[test ApplyTrigger calls tSQLt.Private_MarktSQLtTempObject on new check constraints]
+AS
+BEGIN
+  CREATE TABLE ApplyTriggerTests.aSimpleTable ( Id INT);
+  EXEC('CREATE TRIGGER ApplyTriggerTests.aSimpleTrigger ON ApplyTriggerTests.aSimpleTable FOR INSERT AS RETURN;'); 
+  DECLARE @OriginalObjectId INT = OBJECT_ID('ApplyTriggerTests.aSimpleTrigger');
+
+  EXEC tSQLt.FakeTable @TableName = 'ApplyTriggerTests.aSimpleTable';
+
+  EXEC tSQLt.SpyProcedure @ProcedureName = 'tSQLt.Private_MarktSQLtTempObject';
+  TRUNCATE TABLE tSQLt.Private_MarktSQLtTempObject_SpyProcedureLog;--Quirkiness of testing the framework that you use to run the test
+
+  EXEC tSQLt.ApplyTrigger @TableName = 'ApplyTriggerTests.aSimpleTable', @TriggerName = 'aSimpleTrigger';
+
+  SELECT ObjectName, ObjectType, NewNameOfOriginalObject 
+    INTO #Actual 
+    FROM tSQLt.Private_MarktSQLtTempObject_SpyProcedureLog;
+  
+  SELECT TOP(0) A.* INTO #Expected FROM #Actual A RIGHT JOIN #Actual X ON 1=0;
+  INSERT INTO #Expected
+    VALUES('[ApplyTriggerTests].[aSimpleTrigger]', N'TRIGGER', OBJECT_NAME(@OriginalObjectId));
+
+  EXEC tSQLt.AssertEqualsTable '#Expected','#Actual';
+  
+END;
+GO

Tests/Private_MarktSQLtTempObjectTests.class.sql

@@ -80,3 +80,14 @@ BEGIN
     @ObjectType = N'CONSTRAINT';
 END;
 GO
+CREATE PROCEDURE Private_MarktSQLtTempObjectTests.[test can mark a trigger]
+AS
+BEGIN
+  CREATE TABLE Private_MarktSQLtTempObjectTests.TempTable1(i INT);
+  EXEC('CREATE TRIGGER Private_MarktSQLtTempObjectTests.TempTrigger ON Private_MarktSQLtTempObjectTests.TempTable1 FOR INSERT AS RETURN;');
+
+  EXEC Private_MarktSQLtTempObjectTests.[assert creates two extended properties on object]
+    @ObjectName = 'Private_MarktSQLtTempObjectTests.TempTrigger',
+    @ObjectType = N'TRIGGERX';
+END;
+GO

Tests/UndoTestDoublesTests.class.sql

@@ -329,7 +329,6 @@ BEGIN
   EXEC tSQLt.AssertEmptyTable @TableName = '#ShouldBeEmpty';
 END;
 GO
---tSQLt.Run UndoTestDoublesTests
 CREATE PROCEDURE UndoTestDoublesTests.[test synonyms are restored]
 AS
 BEGIN
@@ -340,8 +339,8 @@ BEGIN
     INTO #OriginalObjectIds
     FROM sys.objects O;
 
-  EXEC tSQLt.RemoveObject @ObjectName='UndoTestDoublesTests.aSimpleSynonym';
-  EXEC ('CREATE TABLE UndoTestDoublesTests.aSimpleSynonym(i INT);');
+  EXEC tSQLt.FakeTable @TableName = 'UndoTestDoublesTests.aSimpleSynonym';
+
   EXEC tSQLt.UndoTestDoubles;
 
   SELECT O.object_id,SCHEMA_NAME(O.schema_id) schema_name, O.name object_name, O.type_desc