tSQLt-org · NReilingh · Nov 27, 2016 · Nov 28, 2016 · Nov 28, 2016 · Dec 6, 2016
@@ -1,5 +1,6 @@
 tSQLt._Header.sql
 tSQLt.DropClass.ssp.sql
+tSQLt.DropAllClasses.ssp.sql
 tSQLt.Private_Bin2Hex.sfn.sql
 tSQLt.Private_NewTestClassList.tbl.sql
 tSQLt.Private_ResetNewTestClassList.ssp.sql

@@ -139,6 +139,12 @@
          <AssociatedConnUserName />
          <FullPath>tSQLt.CaptureOutputLog.tbl.sql</FullPath>
        </FileNode>
+        <FileNode Name="tSQLt.DropAllClasses.ssp.sql">
+          <AssociatedConnectionMoniker />
+          <AssociatedConnSrvName />
+          <AssociatedConnUserName />
+          <FullPath>tSQLt.DropAllClasses.ssp.sql</FullPath>
+        </FileNode>
        <FileNode Name="tSQLt.DropClass.ssp.sql">
          <AssociatedConnectionMoniker>8c91a03d-f9b4-46c0-a305-b5dcc79ff907:Dev_tSQLt:True</AssociatedConnectionMoniker>
          <AssociatedConnSrvName>Dev_tSQLt</AssociatedConnSrvName>

@@ -0,0 +1,26 @@
+IF OBJECT_ID('tSQLt.DropAllClasses') IS NOT NULL DROP PROCEDURE tSQLt.DropAllClasses;
+GO
+---Build+
+CREATE PROCEDURE tSQLt.DropAllClasses
+AS
+BEGIN
+  DECLARE TestClass CURSOR LOCAL FAST_FORWARD
+  FOR
+  SELECT Name FROM tSQLt.TestClasses;
+
+  DECLARE @schema sysname;
+  OPEN TestClass;
+
+  WHILE 1 = 1
+  BEGIN
+    FETCH NEXT FROM TestClass INTO @schema;
+    IF @@FETCH_STATUS <> 0
+      BREAK;
+    EXEC tSQLt.DropClass @ClassName = @schema;
+  END;
+
+  CLOSE TestClass;
+  DEALLOCATE TestClass;
+END;
+---Build-
+GO
@@ -541,6 +541,8 @@ GO
 CREATE PROCEDURE tSQLt.Uninstall
 AS
 BEGIN
+  EXEC tSQLt.DropAllClasses;
+
  DROP TYPE tSQLt.Private;

  EXEC tSQLt.DropClass 'tSQLt';  

@@ -0,0 +1,50 @@
+EXEC tSQLt.NewTestClass 'DropAllClassesTests';
+GO
+
+CREATE PROC DropAllClassesTests.[test DropClass not called if no test classes exist]
+AS
+BEGIN
+    EXEC tSQLt.FakeTable 'tSQLt.TestClasses';
+    EXEC tSQLt.SpyProcedure 'tSQLt.DropClass';
+
+    EXEC tSQLt.DropAllClasses;
+
+    EXEC tSQLt.AssertEmptyTable 'tSQLt.DropClass_SpyProcedureLog';
+END;
+GO
+CREATE PROC DropAllClassesTests.[test that one test class results in one call to DropClass]
+AS
+BEGIN
+    EXEC tSQLt.FakeTable 'tSQLt.TestClasses';
+    EXEC tSQLt.SpyProcedure 'tSQLt.DropClass';
+    EXEC('INSERT INTO tSQLt.TestClasses (Name)
+          VALUES  (''MyTestClass'');');
+
+    SELECT ClassName = Name INTO #expected FROM tSQLt.TestClasses;
+
+    EXEC tSQLt.DropAllClasses;
+
+    SELECT ClassName INTO #actual FROM tSQLt.DropClass_SpyProcedureLog;
+
+    EXEC tSQLt.AssertEqualsTable '#expected', '#actual';
+END;
+GO
+CREATE PROC DropAllClassesTests.[test that multiple test classes are all sent to DropClass]
+AS
+BEGIN
+    EXEC tSQLt.FakeTable 'tSQLt.TestClasses';
+    EXEC tSQLt.SpyProcedure 'tSQLt.DropClass';
+    EXEC('INSERT INTO tSQLt.TestClasses (Name)
+          VALUES  (''MyTestClass1''),
+                  (''MyTestClass2''),
+                  (''MyTestClass3'');');
+
+    SELECT ClassName = Name INTO #expected FROM tSQLt.TestClasses;
+
+    EXEC tSQLt.DropAllClasses;
+
+    SELECT ClassName INTO #actual FROM tSQLt.DropClass_SpyProcedureLog;
+
+    EXEC tSQLt.AssertEqualsTable '#expected', '#actual';
+END;
+GO
@@ -91,6 +91,12 @@
          <AssociatedConnUserName />
          <FullPath>BootStrapTest.sql</FullPath>
        </FileNode>
+        <FileNode Name="DropAllClassesTests.class.sql">
+          <AssociatedConnectionMoniker />
+          <AssociatedConnSrvName />
+          <AssociatedConnUserName />
+          <FullPath>DropAllClassesTests.class.sql</FullPath>
+        </FileNode>
        <FileNode Name="DropClassTests.class.sql">
          <AssociatedConnectionMoniker>8c91a03d-f9b4-46c0-a305-b5dcc79ff907:Dev_tSQLt:True</AssociatedConnectionMoniker>
          <AssociatedConnSrvName>Dev_tSQLt</AssociatedConnSrvName>

@@ -1606,6 +1606,28 @@ BEGIN
 END;
 GO

+CREATE PROCEDURE tSQLt_test.[test Uninstall removes test classes]
+AS
+BEGIN
+  EXEC tSQLt.SpyProcedure 'tSQLt.DropAllClasses';
+  DECLARE @call INT;
+  BEGIN TRAN;
+  DECLARE @TranName CHAR(32); EXEC tSQLt.GetNewTranName @TranName OUT;
+  SAVE TRAN @TranName;
+
+  EXEC tSQLt.Uninstall;
+  SET @call = (SELECT 1 FROM tSQLt.DropAllClasses_SpyProcedureLog);
+
+  ROLLBACK TRAN @TranName;
+  COMMIT TRAN;
+
+  IF @id IS NULL
+  BEGIN
+    EXEC tSQLt.Fail 'DropAllClasses not called';
+  END;
+END;
+GO
+
 CREATE PROCEDURE tSQLt_test.[test Uninstall removes schema tSQLt]
 AS
 BEGIN
@@ -1671,4 +1693,4 @@ END;
 GO

 --ROLLBACK
--tSQLt_test
+--tSQLt_test