sqltoolsservice/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/Scripting/AsyncScriptAsScriptingOperationTests.cs

//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//


using System.Collections.Generic;
using System.Threading.Tasks;
using Microsoft.SqlTools.ServiceLayer.Test.Common;
using Microsoft.SqlTools.SqlCore.Scripting;
using Microsoft.SqlTools.SqlCore.Scripting.Contracts;
using NUnit.Framework;

namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Scripting
{
    public class AsyncScriptAsScriptingOperationTests
    {
        public static IEnumerable<TestCaseData> ScriptAsTestCases
        {
            get
            {
                yield return new TestCaseData(
                    "CREATE TABLE testTable1 (c1 int)",
                    new ScriptingParams()
                    {
                        ScriptDestination = "ToEditor",
                        ScriptingObjects = new List<ScriptingObject>()
                        {
                            new ScriptingObject()
                            {
                                Name = "testTable1",
                                Schema = "dbo",
                                Type = "Table",

                            }
                        },
                        Operation = ScriptingOperationType.Select,
                        ScriptOptions = new ScriptOptions()
                        {
                            ScriptCreateDrop = "ScriptSelect"
                        }
                    },
                    new List<string>() { "SELECT TOP (1000) [c1]" });

                yield return new TestCaseData(
                    "CREATE TABLE testTable1 (c1 int)",
                    new ScriptingParams()
                    {
                        ScriptDestination = "ToEditor",
                        ScriptingObjects = new List<ScriptingObject>()
                        {
                            new ScriptingObject()
                            {
                                Name = "testTable1",
                                Schema = "dbo",
                                Type = "Table"
                            }
                        },
                        Operation = ScriptingOperationType.Delete,
                        ScriptOptions = new ScriptOptions()
                        {
                            ScriptCreateDrop = "ScriptDrop"
                        }
                    },
                    new List<string> { "DROP TABLE [dbo].[testTable1]" }
                    );

                yield return new TestCaseData(
                            @"CREATE TABLE testTable1 (c1 int)
                            GO
                            CREATE CLUSTERED INDEX [ClusteredIndex-1] ON [dbo].[testTable1]
                            (
	                            [c1] ASC
                            )
                            GO
                            ",
                            new ScriptingParams()
                            {
                                ScriptDestination = "ToEditor",
                                ScriptingObjects = new List<ScriptingObject>()
                                {
                                    new ScriptingObject()
                                    {
                                        Name = "testTable1",
                                        Schema = "dbo",
                                        Type = "Table"
                                    }
                                },
                                Operation = ScriptingOperationType.Create,
                                ScriptOptions = new ScriptOptions()
                                {
                                    ScriptCreateDrop = "ScriptCreate"
                                }
                            },
                            new List<string> { "CREATE TABLE [dbo].[testTable1]", "CREATE CLUSTERED INDEX [ClusteredIndex-1] ON [dbo].[testTable1]" }
                    );
            }
        }

        [Test]
        [TestCaseSource("ScriptAsTestCases")]
        public async Task TestCommonScenarios(
            string query, ScriptingParams scriptingParams, List<string> expectedScriptContents)
        {
            var testDb = await SqlTestDb.CreateNewAsync(TestServerType.OnPrem, false, null, query, "ScriptingTests");
            scriptingParams.ConnectionString = testDb.ConnectionString;

            var actualScript = await AsyncScriptAsScriptingOperation.GetScriptAsScript(scriptingParams, null);

            foreach(var expectedStr in expectedScriptContents)
            {
                Assert.That(actualScript, Does.Contain(expectedStr));
            }

            await testDb.CleanupAsync();
        }
    }
}