TestLanguageServices.cs

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System;
using System.Collections.Generic;
using System.Linq;
using Microsoft.CodeAnalysis.Razor;

namespace Microsoft.CodeAnalysis.Host
{
    internal class TestLanguageServices : HostLanguageServices
    {
        private readonly HostWorkspaceServices _workspaceServices;
        private readonly IEnumerable<ILanguageService> _languageServices;

        public TestLanguageServices(HostWorkspaceServices workspaceServices, IEnumerable<ILanguageService> languageServices)
        {
            if (workspaceServices == null)
            {
                throw new ArgumentNullException(nameof(workspaceServices));
            }

            if (languageServices == null)
            {
                throw new ArgumentNullException(nameof(languageServices));
            }

            _workspaceServices = workspaceServices;
            _languageServices = languageServices;
        }

        public override HostWorkspaceServices WorkspaceServices => _workspaceServices;

        public override string Language => RazorLanguage.Name;

        public override TLanguageService GetService<TLanguageService>()
        {
            var service = _languageServices.OfType<TLanguageService>().FirstOrDefault();

            if (service == null)
            {
                throw new InvalidOperationException($"Test Razor language services not configured properly, missing language service '{typeof(TLanguageService).FullName}'.");
            }

            return service;
        }
    }
}