sqltoolsservice/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/ObjectExplorer/StatelessObjectExplorerServiceTests.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;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.SqlTools.ServiceLayer.Test.Common;
using Microsoft.SqlTools.ServiceLayer.Test.Common.Extensions;
using Microsoft.SqlTools.SqlCore.ObjectExplorer;
using NUnit.Framework;

namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.ObjectExplorer
{
    public class StatelessObjectExplorerServiceTests
    {

        string databaseName = "tempdb";

        ObjectExplorerServerInfo serverInfo = new ObjectExplorerServerInfo()
        {
            DatabaseName = "tempdb",
            ServerName = "testserver",
            UserName = "testuser",
            IsCloud = true,
            isDefaultOrSystemDatabase = false
        };

        ObjectExplorerOptions options = new ObjectExplorerOptions()
        {
            GroupBySchemaFlagGetter = () => true,
            OperationTimeoutSeconds = 10000,
        };

        [Test]
        [TestCase("", "dbo")]
        [TestCase("testserver/{0}/dbo", "Tables")]
        [TestCase("testserver/{0}/dbo/Tables", "dbo.t1")]
        [TestCase("testserver/{0}/dbo/Tables", "dbo.t2")]
        public async Task ExpandingPathShouldReturnCorrectNodes(string oePath, string childLabel)
        {
            var query = @"Create table t1 (c1 int)
                            GO
                            Create table t2 (c1 int)
                            GO";
            await RunTest(databaseName, query, "testdb", async (testdbName, connectionString) =>
            {
                serverInfo.DatabaseName = testdbName;
                var pathWithDb = string.Format(oePath, testdbName);
                var nodes = await StatelessObjectExplorer.Expand(connectionString, null, pathWithDb, serverInfo, options);

                Assert.True(nodes.Any(node => node.Label == childLabel), $"Expansion result for {pathWithDb} does not contain node {childLabel}");
            });
        }

        [Test]
        public async Task ProvidingNodeShouldSkipExpandingFromTop()
        {
            var query = @"Create table t1 (c1 int)
                            GO
                            Create table t2 (c1 int)
                            GO";
            await RunTest(databaseName, query, "testdb", async (testdbName, connectionString) =>
            {
                serverInfo.DatabaseName = testdbName;
                var oePath = "";
                var pathWithDb = string.Format(oePath, testdbName);

                var nodes = await StatelessObjectExplorer.Expand(connectionString, null, pathWithDb, serverInfo, options);
                Assert.True(nodes.Any(node => node.Label == "dbo"), $"Expansion result for {pathWithDb} does not contain node dbo");

                nodes = await StatelessObjectExplorer.ExpandTreeNode(nodes[0], options, null, null);
                Assert.True(nodes.Any(node => node.Label == "Tables"), $"Expansion result for {pathWithDb} does not contain node t1");

                nodes = await StatelessObjectExplorer.ExpandTreeNode(nodes.First(node => node.Label == "Tables"), options, null, null);
                Assert.True(nodes.Any(node => node.Label == "dbo.t1"), $"Expansion result for {pathWithDb} does not contain node t1");

                nodes = await StatelessObjectExplorer.ExpandTreeNode(nodes.First(node => node.Label == "dbo.t1"), options, null, null);
                Assert.True(nodes.Any(node => node.Label == "Columns"), $"Expansion result for {pathWithDb} does not contain node Columns");

                nodes = await StatelessObjectExplorer.ExpandTreeNode(nodes.First(node => node.Label == "Columns"), options, null, null);
                Assert.True(nodes.Any(node => node.Label == "c1 (int, null)"), $"Expansion result for {pathWithDb} does not contain node c1");
            });
        }

        private async Task RunTest(string databaseName, string query, string testDbPrefix, Func<string, string, Task> test)
        {
            SqlTestDb? testDb = null;
            try
            {
                testDb = await SqlTestDb.CreateNewAsync(TestServerType.OnPrem, false, null, query, testDbPrefix);
                if (databaseName == "#testDb#")
                {
                    databaseName = testDb.DatabaseName;
                }
                await test(testDb.DatabaseName, testDb.ConnectionString);
            }
            catch (Exception ex)
            {
                string msg = ex.BuildRecursiveErrorMessage();
                throw new Exception($"Failed to run OE test. error:{msg} {ex.StackTrace}");
            }
            finally
            {
                if (testDb != null)
                {
                    await testDb.CleanupAsync();
                }
            }
        }
    }
}