src/VisualStudio/TestUtilities2/MockComponentModel.vb - main - githubmirror/roslyn

src/VisualStudio/TestUtilities2/MockComponentModel.vb
62 строки2 KB
Tomas Matousek
Update VS SDK to 17.12.39557-preview.2.1
12 ноя 2024, 23:24
12 ноя 2024, 23:24d1f0374
' Licensed to the .NET Foundation under one or more agreements.
' The .NET Foundation licenses this file to you under the MIT license.
' See the LICENSE file in the project root for more information.

Imports System.ComponentModel.Composition
Imports System.ComponentModel.Composition.Primitives
Imports Microsoft.VisualStudio.ComponentModelHost
Imports Microsoft.VisualStudio.Composition

Namespace Microsoft.VisualStudio.LanguageServices.UnitTests
    Friend NotInheritable Class MockComponentModel
        Implements IComponentModel

        Private ReadOnly _exportProvider As ExportProvider
        Private ReadOnly _providedServices As New Dictionary(Of Type, Object)

        Public Sub New(exportProvider As ExportProvider)
            _exportProvider = exportProvider
        End Sub

        <Obsolete>
        Public ReadOnly Property DefaultCatalog As ComposablePartCatalog Implements IComponentModel.DefaultCatalog
            Get
                Throw New NotImplementedException
            End Get
        End Property

        Public ReadOnly Property DefaultCompositionService As ICompositionService Implements IComponentModel.DefaultCompositionService
            Get
                Throw New NotImplementedException
            End Get
        End Property

        Public ReadOnly Property DefaultExportProvider As Hosting.ExportProvider Implements IComponentModel.DefaultExportProvider
            Get
                Return _exportProvider.AsExportProvider()
            End Get
        End Property

        <Obsolete>
        Public Function GetCatalog(catalogName As String) As ComposablePartCatalog Implements IComponentModel.GetCatalog
            Throw New NotImplementedException
        End Function

        Friend Sub ProvideService(Of T)(export As T)
            _providedServices.Add(GetType(T), export)
        End Sub

        Public Function GetExtensions(Of T As Class)() As IEnumerable(Of T) Implements IComponentModel.GetExtensions
            Return _exportProvider.GetExportedValues(Of T)()
        End Function

        Public Function GetService(Of T As Class)() As T Implements IComponentModel.GetService
            Dim possibleService As Object = Nothing
            If _providedServices.TryGetValue(GetType(T), possibleService) Then
                Return DirectCast(possibleService, T)
            End If

            Return _exportProvider.GetExportedValue(Of T)()
        End Function
    End Class
End Namespace