Object Explorer Service (#311)

* moving OE service from an old branch
2026-01-14 01:25:40 -05:00 · 2017-04-11 15:50:20 -07:00
parent 90861b7d9e
commit d903ba56a9
58 changed files with 15013 additions and 1057 deletions
--- a/test/Microsoft.SqlTools.ServiceLayer.UnitTests/ObjectExplorer/SmoQueryModelTests.cs
+++ b/test/Microsoft.SqlTools.ServiceLayer.UnitTests/ObjectExplorer/SmoQueryModelTests.cs
@@ -0,0 +1,65 @@
+//
+// 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 Microsoft.SqlServer.Management.Smo;
+using Microsoft.SqlTools.Extensibility;
+using Microsoft.SqlTools.ServiceLayer.ObjectExplorer.SmoModel;
+using Xunit;
+
+namespace Microsoft.SqlTools.ServiceLayer.UnitTests.ObjectExplorer
+{
+    public class SmoQueryModelTests
+    {
+
+        [Fact]
+        public void ShouldFindDatabaseQuerierFromRealPath()
+        {
+            // Given the extension type loader is set to find SmoCollectionQuerier objects
+            IMultiServiceProvider serviceProvider = ExtensionServiceProvider.CreateDefaultServiceProvider();
+            // When I request a database compatible querier
+            SmoQuerier querier = serviceProvider.GetService<SmoQuerier>(q => q.SupportedObjectTypes.Contains(typeof(Database)));
+            // Then I expect to get back the SqlDatabaseQuerier
+            Assert.NotNull(querier);
+            Assert.Equal(typeof(SqlDatabaseQuerier), querier.GetType());
+
+            // And I expect the service provider to have been set by the extension code
+            Assert.NotNull(querier.ServiceProvider);
+        }
+        
+        [Fact]
+        public void ShouldFindQuerierIfInExtensionList()
+        {
+            VerifyQuerierLookup(typeof(Table), typeof(SqlTableQuerier), expectExists: true);
+        }
+
+        [Fact]
+        public void ShouldNotFindQuerierIfNotInExtensionList()
+        {
+            VerifyQuerierLookup(typeof(Database), null, expectExists: false);
+        }
+
+        private static void VerifyQuerierLookup(Type smoType, Type querierType, bool expectExists)
+        {
+            ExtensionServiceProvider serviceProvider = ExtensionServiceProvider.Create(new Type[] {
+                typeof(SqlTableQuerier),
+                typeof(SqlLinkedServerQuerier)
+            });
+            SmoQuerier querier = serviceProvider.GetService<SmoQuerier>(q => q.SupportedObjectTypes.Contains(smoType));
+            if (expectExists)
+            {
+                Assert.NotNull(querier);
+                Assert.Equal(querierType, querier.GetType());
+                Assert.NotNull(querier.ServiceProvider);
+            }
+            else
+            {
+                Assert.Null(querier);
+            }
+        }
+
+    }
+}