Handle empty or null schema for CreateView with replace option DAT-16446

liquibase-integration-tests/src/test/groovy/liquibase/command/core/GenerateChangeLogMSSQLIntegrationTest.groovy

@@ -1,11 +1,15 @@
 package liquibase.command.core
 
 import liquibase.Scope
+import liquibase.command.CommandScope
+import liquibase.command.core.helpers.DbUrlConnectionArgumentsCommandStep
 import liquibase.command.util.CommandUtil
 import liquibase.extension.testing.testsystem.DatabaseTestSystem
 import liquibase.extension.testing.testsystem.TestSystemFactory
 import liquibase.extension.testing.testsystem.spock.LiquibaseIntegrationTest
+import liquibase.resource.SearchPathResourceAccessor
 import liquibase.util.FileUtil
+import liquibase.util.StringUtil
 import spock.lang.Shared
 import spock.lang.Specification
 
@@ -71,4 +75,75 @@ class GenerateChangeLogMSSQLIntegrationTest extends Specification {
         where:
         fileType << ['xml', 'json', 'yml']
     }
+
+    def "Should include schema when generating changelog with view and using 'use-or-replace-option'"() {
+        when:
+        String changelogFile = "target/test-classes/changelogs/" + StringUtil.randomIdentifer(10) + "/formatted.mssql.sql"
+        String updateChangelogFile = "target/test-classes/changelogs/" + StringUtil.randomIdentifer(10) + "/formatted.sql"
+
+        def contents =
+                """
+-- liquibase formatted sql
+
+-- changeset liquibase:1 label:basic
+CREATE TABLE Employees (
+    EmployeeID INT,
+    FirstName NVARCHAR(50)
+);
+-- rollback drop table Employees
+
+--changeset wesley:1575473414720-9 splitStatements:false
+CREATE VIEW employees_view AS SELECT FirstName FROM [dbo].Employees;
+
+"""
+        File f = new File(updateChangelogFile)
+        f.getParentFile().mkdirs()
+        f.write(contents + "\n")
+
+        CommandScope updateCommandScope = new CommandScope(UpdateCommandStep.COMMAND_NAME)
+        updateCommandScope.addArgumentValue(UpdateCommandStep.CHANGELOG_FILE_ARG, updateChangelogFile)
+        updateCommandScope.addArgumentValue(DbUrlConnectionArgumentsCommandStep.USERNAME_ARG, mssql.getUsername())
+        updateCommandScope.addArgumentValue(DbUrlConnectionArgumentsCommandStep.PASSWORD_ARG, mssql.getPassword())
+        updateCommandScope.addArgumentValue(DbUrlConnectionArgumentsCommandStep.URL_ARG, mssql.getConnectionUrl())
+
+        CommandScope commandScope = new CommandScope(GenerateChangelogCommandStep.COMMAND_NAME)
+        commandScope.addArgumentValue(GenerateChangelogCommandStep.CHANGELOG_FILE_ARG, changelogFile)
+        commandScope.addArgumentValue(DbUrlConnectionArgumentsCommandStep.USERNAME_ARG, mssql.getUsername())
+        commandScope.addArgumentValue(DbUrlConnectionArgumentsCommandStep.PASSWORD_ARG, mssql.getPassword())
+        commandScope.addArgumentValue(DbUrlConnectionArgumentsCommandStep.URL_ARG, mssql.getConnectionUrl())
+        commandScope.addArgumentValue(GenerateChangelogCommandStep.USE_OR_REPLACE_OPTION, true);
+        OutputStream outputStream = new ByteArrayOutputStream()
+        commandScope.setOutput(outputStream)
+        Map<String, Object> scopeValues = new HashMap<>()
+        outputStream.flush()
+
+        scopeValues.put("liquibase.pro.sql.inline", true)
+        scopeValues.put(Scope.Attr.resourceAccessor.name(), new SearchPathResourceAccessor("."))
+        Scope.child(scopeValues, new Scope.ScopedRunner() {
+            @Override
+            void run() throws Exception {
+                updateCommandScope.execute()
+                commandScope.execute()
+            }
+        })
+        def generatedChangelog = new File(changelogFile)
+        def generatedChangelogContents = FileUtil.getContents(generatedChangelog)
+
+        then:
+        noExceptionThrown()
+        generatedChangelogContents.contains("N'CREATE VIEW [DBO].[employees_view] AS SELECT '")
+
+        cleanup:
+        try {
+            generatedChangelog.delete()
+        } catch (Exception ignored) {
+
+        }
+
+        CommandUtil.runDropAll(mssql)
+
+        if (mssql.getConnection()) {
+            mssql.getConnection().close()
+        }
+    }
 }

liquibase-standard/src/main/java/liquibase/command/core/GenerateChangelogCommandStep.java

@@ -10,6 +10,7 @@
 import liquibase.configuration.ConfigurationValueObfuscator;
 import liquibase.database.Database;
 import liquibase.database.ObjectQuotingStrategy;
+import liquibase.database.core.MSSQLDatabase;
 import liquibase.diff.DiffResult;
 import liquibase.diff.output.DiffOutputControl;
 import liquibase.diff.output.changelog.DiffToChangeLog;
@@ -126,8 +127,15 @@ public void run(CommandResultsBuilder resultsBuilder) throws Exception {
         diffOutputControl.setDataDir(commandScope.getArgumentValue(DATA_OUTPUT_DIR_ARG));
         referenceDatabase.setOutputDefaultSchema(diffOutputControl.getIncludeSchema());
 
-        if(commandScope.getArgumentValue(GenerateChangelogCommandStep.USE_OR_REPLACE_OPTION).booleanValue()) {
+        //
+        // For MSSQL, if we have the --use-or-replace-option, then we need to also include the schema
+        // because the generator code might need it for CreateView
+        //
+        if (commandScope.getArgumentValue(GenerateChangelogCommandStep.USE_OR_REPLACE_OPTION).booleanValue()) {
             diffOutputControl.setReplaceIfExistsSet(true);
+            if (referenceDatabase instanceof MSSQLDatabase) {
+                diffOutputControl.setIncludeSchema(true);
+            }
         }
 
         InternalSnapshotCommandStep.logUnsupportedDatabase(referenceDatabase, this.getClass());