|
4 | 4 | // </copyright> |
5 | 5 | //------------------------------------------------------------------------------ |
6 | 6 |
|
| 7 | +using System; |
7 | 8 | using Microsoft.SqlServer.TransactSql.ScriptDom; |
8 | 9 | using Microsoft.VisualStudio.TestTools.UnitTesting; |
9 | 10 | using SqlStudio.Tests.AssemblyTools.TestCategory; |
@@ -113,5 +114,48 @@ public void TestSqlServerlessScriptGenerator() |
113 | 114 | var scriptGenerator = new SqlServerlessScriptGenerator(options); |
114 | 115 | Assert.AreEqual(SqlVersion.Sql160, scriptGenerator.Options.SqlVersion); |
115 | 116 | } |
| 117 | + |
| 118 | + [TestMethod] |
| 119 | + [Priority(0)] |
| 120 | + [SqlStudioTestCategory(Category.UnitTest)] |
| 121 | + public void TestNewlinesBetweenStatementsGeneratorOption() { |
| 122 | + var tableName = new SchemaObjectName(); |
| 123 | + tableName.Identifiers.Add(new Identifier { Value = "TableName" }); |
| 124 | + |
| 125 | + var tableStatement = new CreateTableStatement |
| 126 | + { |
| 127 | + SchemaObjectName = tableName |
| 128 | + }; |
| 129 | + var tableStatementString = "CREATE TABLE TableName;"; |
| 130 | + |
| 131 | + var statements = new StatementList(); |
| 132 | + statements.Statements.Add(tableStatement); |
| 133 | + statements.Statements.Add(tableStatement); |
| 134 | + |
| 135 | + var generatorOptions = new SqlScriptGeneratorOptions { |
| 136 | + KeywordCasing = KeywordCasing.Uppercase, |
| 137 | + IncludeSemicolons = true, |
| 138 | + NumNewlinesAfterStatement = 0 |
| 139 | + }; |
| 140 | + |
| 141 | + var generator = new Sql80ScriptGenerator(generatorOptions); |
| 142 | + |
| 143 | + generator.GenerateScript(statements, out var sql); |
| 144 | + |
| 145 | + Assert.AreEqual(tableStatementString + tableStatementString, sql); |
| 146 | + |
| 147 | + generatorOptions.NumNewlinesAfterStatement = 1; |
| 148 | + generator = new Sql80ScriptGenerator(generatorOptions); |
| 149 | + |
| 150 | + generator.GenerateScript(statements, out sql); |
| 151 | + |
| 152 | + Assert.AreEqual(tableStatementString + Environment.NewLine + tableStatementString, sql); |
| 153 | + |
| 154 | + generatorOptions.NumNewlinesAfterStatement = 2; |
| 155 | + generator = new Sql80ScriptGenerator(generatorOptions); |
| 156 | + |
| 157 | + generator.GenerateScript(statements, out sql); |
| 158 | + Assert.AreEqual(tableStatementString + Environment.NewLine + Environment.NewLine + tableStatementString, sql); |
| 159 | + } |
116 | 160 | } |
117 | 161 | } |
0 commit comments