@@ -200,32 +200,35 @@ BEGIN;
200200DECLARE @SqlMajorVersion TINYINT;
201201DECLARE @Verbose BIT = 0 ;
202202DECLARE @DatabaseName SYSNAME = ' tSQLt' ;
203+ DECLARE @Sql NVARCHAR(MAX);
203204DECLARE @FailMessage NVARCHAR(MAX) = N' Did not find test sensitivity classifications in output.' ;
204-
205205DECLARE @Expected NVARCHAR(1000 ) = N' %Label: Highly Confidential <br /> Type: Financial <br /> Rank: CRITICAL <br />%' ;
206206
207207SET @SqlMajorVersion = CAST(SERVERPROPERTY(' ProductMajorVersion' ) AS TINYINT);
208208
209209IF (@SqlMajorVersion >= 15 )
210+ BEGIN
211+ -- Setup
212+ IF OBJECT_ID(' tempdb..#result' ) IS NOT NULL
213+ BEGIN
214+ DROP TABLE # result;
215+ END
216+ CREATE TABLE # result ([markdown] NVARCHAR(MAX));
217+
218+ SET @Sql = N' ADD SENSITIVITY CLASSIFICATION TO [tSQLt].[CaptureOutputLog].[OutputText]
219+ WITH (LABEL=' ' Highly Confidential' ' , INFORMATION_TYPE=' ' Financial' ' , RANK=CRITICAL)' ;
220+ EXEC sp_executesql @Sql;
221+
222+ -- Get results
223+ INSERT INTO # result
224+ EXEC sp_doc @DatabaseName = @DatabaseName, @Verbose = @Verbose;
225+
226+ -- Assert
227+ IF NOT EXISTS (SELECT 1 FROM # result WHERE [markdown] LIKE @Expected)
210228 BEGIN
211- -- Setup
212- IF OBJECT_ID(' tempdb..#result' ) IS NOT NULL
213- BEGIN
214- DROP TABLE # result;
215- END
216- CREATE TABLE # result ([markdown] NVARCHAR(MAX));
217-
218- ADD SENSITIVITY CLASSIFICATION TO [tSQLt].[CaptureOutputLog].[OutputText]
219- WITH (LABEL= ' Highly Confidential' , INFORMATION_TYPE= ' Financial' , RANK= CRITICAL)
220-
221- -- Get results
222- INSERT INTO # result
223- EXEC sp_doc @DatabaseName = @DatabaseName, @Verbose = @Verbose;
224-
225- -- Assert
226- IF NOT EXISTS (SELECT 1 FROM # result WHERE [markdown] LIKE @Expected)
227- EXEC tSQLt .Fail @FailMessage;
229+ EXEC tSQLt .Fail @FailMessage;
228230 END;
231+ END;
229232
230233-- Succeed if version < 15
231234EXEC tSQLt .ExpectNoException ;
0 commit comments