Support Object Explorer FindNodes request (#589)

2026-01-14 01:25:40 -05:00 · 2018-03-15 10:47:52 -07:00
parent d36efb578c
commit 365fe2282e
8 changed files with 584 additions and 33 deletions
--- a/test/Microsoft.SqlTools.ServiceLayer.UnitTests/ObjectExplorer/NodePathGeneratorTests.cs
+++ b/test/Microsoft.SqlTools.ServiceLayer.UnitTests/ObjectExplorer/NodePathGeneratorTests.cs
@@ -0,0 +1,164 @@
+//
+// 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.Collections.Generic;
+using System.Data.SqlClient;
+using System.Linq;
+using System.Threading.Tasks;
+using Microsoft.SqlTools.Hosting.Protocol;
+using Microsoft.SqlTools.ServiceLayer.Connection;
+using Microsoft.SqlTools.ServiceLayer.Connection.Contracts;
+using Microsoft.SqlTools.ServiceLayer.ObjectExplorer;
+using Microsoft.SqlTools.ServiceLayer.ObjectExplorer.Contracts;
+using Microsoft.SqlTools.ServiceLayer.ObjectExplorer.Nodes;
+using Microsoft.SqlTools.ServiceLayer.ObjectExplorer.SmoModel;
+using Microsoft.SqlTools.ServiceLayer.UnitTests.Utility;
+using Moq;
+using Xunit;
+using Microsoft.SqlTools.ServiceLayer.LanguageServices;
+using Microsoft.SqlServer.Management.Common;
+using Microsoft.SqlTools.ServiceLayer.Test.Common.RequestContextMocking;
+
+namespace Microsoft.SqlTools.ServiceLayer.UnitTests.ObjectExplorer
+{
+
+    public class NodePathGeneratorTests
+    {
+        private ObjectExplorerService.ObjectExplorerSession serverSession;
+        private ObjectExplorerService.ObjectExplorerSession databaseSession;
+        private const string serverName = "testServer";
+        private const string databaseName = "testDatabase";
+
+        public NodePathGeneratorTests()
+        {
+            var serverRoot = new TreeNode
+            {
+                NodeType = "Server",
+                NodeValue = serverName
+            };
+
+            serverSession = new ObjectExplorerService.ObjectExplorerSession("serverUri", serverRoot, null, null);
+
+            var databaseRoot = new TreeNode
+            {
+                NodeType = "Database",
+                NodeValue = databaseName,
+                Parent = serverRoot
+            };
+
+            databaseSession = new ObjectExplorerService.ObjectExplorerSession("databaseUri", databaseRoot, null, null);
+        }
+
+        [Fact]
+        public void FindCorrectPathsForTableWithServerRoot()
+        {
+            var paths = NodePathGenerator.FindNodePaths(serverSession, "Table", "testSchema", "testTable", databaseName);
+            var expectedPaths = new List<string>
+            {
+                "testServer/Databases/testDatabase/Tables/testSchema.testTable",
+                "testServer/Databases/System Databases/testDatabase/Tables/testSchema.testTable",
+                "testServer/Databases/testDatabase/Tables/System Tables/testSchema.testTable",
+                "testServer/Databases/System Databases/testDatabase/Tables/System Tables/testSchema.testTable"
+            };
+
+            Assert.Equal(expectedPaths.Count, paths.Count);
+            foreach (var expectedPath in expectedPaths)
+            {
+                Assert.True(paths.Contains(expectedPath));
+            }
+        }
+
+        [Fact]
+        public void FindCorrectPathsForTableWithDatabaseRoot()
+        {
+            var paths = NodePathGenerator.FindNodePaths(databaseSession, "Table", "testSchema", "testTable", null);
+            var expectedPaths = new List<string>
+            {
+                "testServer/testDatabase/Tables/testSchema.testTable",
+                "testServer/testDatabase/Tables/System Tables/testSchema.testTable"
+            };
+
+            Assert.Equal(expectedPaths.Count, paths.Count);
+            foreach (var expectedPath in expectedPaths)
+            {
+                Assert.True(paths.Contains(expectedPath));
+            }
+        }
+
+        [Fact]
+        public void FindCorrectPathsForColumnWithServerRoot()
+        {
+            var paths = NodePathGenerator.FindNodePaths(serverSession, "Column", null, "testColumn", databaseName, new List<string> { "testSchema.testTable" });
+            var expectedPaths = new List<string>
+            {
+                "testServer/Databases/testDatabase/Tables/testSchema.testTable/Columns/testColumn",
+                "testServer/Databases/System Databases/testDatabase/Tables/testSchema.testTable/Columns/testColumn",
+                "testServer/Databases/testDatabase/Tables/System Tables/testSchema.testTable/Columns/testColumn",
+                "testServer/Databases/System Databases/testDatabase/Tables/System Tables/testSchema.testTable/Columns/testColumn",
+                "testServer/Databases/testDatabase/Views/testSchema.testTable/Columns/testColumn",
+                "testServer/Databases/System Databases/testDatabase/Views/testSchema.testTable/Columns/testColumn",
+                "testServer/Databases/testDatabase/Views/System Views/testSchema.testTable/Columns/testColumn",
+                "testServer/Databases/System Databases/testDatabase/Views/System Views/testSchema.testTable/Columns/testColumn"
+            };
+
+            Assert.Equal(expectedPaths.Count, paths.Count);
+            foreach (var expectedPath in expectedPaths)
+            {
+                Assert.True(paths.Contains(expectedPath));
+            }
+        }
+
+        [Fact]
+        public void FindCorrectPathsForColumnWithDatabaseRoot()
+        {
+            var paths = NodePathGenerator.FindNodePaths(databaseSession, "Column", null, "testColumn", databaseName, new List<string> { "testSchema.testTable" });
+            var expectedPaths = new List<string>
+            {
+                "testServer/testDatabase/Tables/testSchema.testTable/Columns/testColumn",
+                "testServer/testDatabase/Tables/System Tables/testSchema.testTable/Columns/testColumn",
+                "testServer/testDatabase/Views/testSchema.testTable/Columns/testColumn",
+                "testServer/testDatabase/Views/System Views/testSchema.testTable/Columns/testColumn"
+            };
+
+            Assert.Equal(expectedPaths.Count, paths.Count);
+            foreach (var expectedPath in expectedPaths)
+            {
+                Assert.True(paths.Contains(expectedPath));
+            }
+        }
+
+        [Fact]
+        public void FindCorrectPathsForDatabase()
+        {
+            var paths = NodePathGenerator.FindNodePaths(serverSession, "Database", null, databaseName, null);
+            var expectedPaths = new List<string>
+            {
+                "testServer/Databases/testDatabase",
+                "testServer/Databases/System Databases/testDatabase"
+            };
+
+            Assert.Equal(expectedPaths.Count, paths.Count);
+            foreach (var expectedPath in expectedPaths)
+            {
+                Assert.True(paths.Contains(expectedPath));
+            }
+        }
+
+        [Fact]
+        public void FindPathForInvalidTypeReturnsEmpty()
+        {
+            var serverPaths = NodePathGenerator.FindNodePaths(serverSession, "WrongType", "testSchema", "testTable", databaseName);
+            Assert.Equal(0, serverPaths.Count);
+        }
+
+        [Fact]
+        public void FindPathMissingParentReturnsEmpty()
+        {
+            var serverPaths = NodePathGenerator.FindNodePaths(serverSession, "Column", "testSchema", "testColumn", databaseName);
+            Assert.Equal(0, serverPaths.Count);
+        }
+    }
+}