@@ -221,19 +221,38 @@ public void Process_VariableReplacement_Normal(string eol, string eof)
221221
222222 [ Test ]
223223 [ TestCaseSource ( nameof ( EolEofCases ) ) ]
224- public void Process_VariableReplacement_WithComments ( string eol , string eof )
224+ public void Process_VariableReplacement_WithComments_Default ( string eol , string eof )
225225 {
226226 var preprocessor = new SqlCmdPreprocessor
227227 {
228228 Variables = { [ "foo" ] = "bar" }
229229 } ;
230230
231231 var batches = preprocessor . Process (
232- Lines ( eol , eof , "a $(FOO) b /* $(FOO) */ $(FOO) -- $(FOO)" )
232+ Lines ( eol , eof , "-- $(FOO)" , " a $(FOO) b /* $(FOO) */ $(FOO) -- $(FOO)")
233233 ) ;
234234
235235 batches . Should ( ) . Equal (
236- Lines ( eol , eof , "a bar b /* $(FOO) */ bar -- $(FOO)" )
236+ Lines ( eol , eof , "-- $(FOO)" , "a bar b /* $(FOO) */ bar -- $(FOO)" )
237+ ) ;
238+ }
239+
240+ [ Test ]
241+ [ TestCaseSource ( nameof ( EolEofCases ) ) ]
242+ public void Process_VariableReplacement_WithComments_ReplacementEnabled ( string eol , string eof )
243+ {
244+ var preprocessor = new SqlCmdPreprocessor
245+ {
246+ Variables = { [ "foo" ] = "bar" } ,
247+ EnableVariableReplacementInComments = true
248+ } ;
249+
250+ var batches = preprocessor . Process (
251+ Lines ( eol , eof , "-- $(FOO)" , "a $(FOO) b /* $(FOO) */ $(FOO) -- $(FOO)" )
252+ ) ;
253+
254+ batches . Should ( ) . Equal (
255+ Lines ( eol , eof , "-- bar" , "a bar b /* bar */ bar -- bar" )
237256 ) ;
238257 }
239258
0 commit comments