fix --drop to work with sqlserver database snapshots

chucknorris · Jul 19, 2011 · c127fa4 · c127fa4
1 parent d1a25ec
commit c127fa4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/product/roundhouse.databases.sqlserver/SqlServerDatabase.cs b/product/roundhouse.databases.sqlserver/SqlServerDatabase.cs
@@ -190,9 +190,13 @@ public override string delete_database_script()
         {
             return string.Format(
                 @"USE master 
-                        IF EXISTS(SELECT * FROM sys.databases WHERE [name] = '{0}') 
+                        IF EXISTS(SELECT * FROM sys.databases WHERE [name] = '{0}' AND source_database_id is NULL) 
                         BEGIN 
                             ALTER DATABASE [{0}] SET SINGLE_USER WITH ROLLBACK IMMEDIATE
+                        END
+
+                        IF EXISTS(SELECT * FROM sys.databases WHERE [name] = '{0}') 
+                        BEGIN
                             EXEC msdb.dbo.sp_delete_database_backuphistory @database_name = '{0}' 
                             DROP DATABASE [{0}] 
                         END",