Merge pull request #3442 from gdoddsy/issue_3429

Update sp_DatabaseRestore.sql

Author: BrentOzar

sp_DatabaseRestore.sql

@@ -39,7 +39,8 @@ ALTER PROCEDURE [dbo].[sp_DatabaseRestore]
     @Version     VARCHAR(30) = NULL OUTPUT,
     @VersionDate DATETIME = NULL OUTPUT,
     @VersionCheckMode BIT = 0,
-    @FileNamePrefix NVARCHAR(260) = NULL
+    @FileNamePrefix NVARCHAR(260) = NULL,
+	@RunStoredProcAfterRestore NVARCHAR(260) = NULL
 AS
 SET NOCOUNT ON;
 SET STATISTICS XML OFF;
@@ -1637,6 +1638,29 @@ END;'
 -- If test restore then blow the database away (be careful)
 IF @TestRestore = 1
 	BEGIN
+
+		IF @RunStoredProcAfterRestore IS NOT NULL AND LEN(LTRIM(@RunStoredProcAfterRestore)) > 0
+		BEGIN
+			PRINT 'Attempting to run ' + @RunStoredProcAfterRestore
+			SET @sql = N'EXEC ' + @RestoreDatabaseName + '.' + @RunStoredProcAfterRestore
+
+			IF @Debug = 1 OR @Execute = 'N'
+			BEGIN
+				IF @sql IS NULL PRINT '@sql is NULL when building for @RunStoredProcAfterRestore'
+				PRINT @sql
+			END
+			
+			IF @RunRecovery = 0
+			BEGIN
+				PRINT 'Unable to run Run Stored Procedure After Restore as database is not recovered. Run command again with @RunRecovery = 1'
+			END
+			ELSE
+            BEGIN
+				IF @Debug IN (0, 1) AND @Execute = 'Y'
+					EXEC sp_executesql  @sql
+			END
+		END
+
 		SET @sql = N'DROP DATABASE ' + @RestoreDatabaseName + NCHAR(13);
 
 			IF @Debug = 1 OR @Execute = 'N'