src/Compilers/Test/Core/Mocks/TestSourceReferenceResolver.cs - main - githubmirror/roslyn

src/Compilers/Test/Core/Mocks/TestSourceReferenceResolver.cs
64 строки2 KB
Tomas Matousek
Move
23 окт 2020, 21:06
23 окт 2020, 21:06d49cd2b
// 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.

#nullable disable

using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Runtime.CompilerServices;
using System.Text;
using Microsoft.CodeAnalysis;

namespace Roslyn.Test.Utilities
{
    public sealed class TestSourceReferenceResolver : SourceReferenceResolver
    {
        public static readonly SourceReferenceResolver Default = new TestSourceReferenceResolver(sources: null);

        public static SourceReferenceResolver Create(params KeyValuePair<string, string>[] sources)
        {
            return new TestSourceReferenceResolver(sources.ToDictionary(p => p.Key, p => (object)p.Value));
        }

        public static SourceReferenceResolver Create(params KeyValuePair<string, object>[] sources)
        {
            return new TestSourceReferenceResolver(sources.ToDictionary(p => p.Key, p => p.Value));
        }

        public static SourceReferenceResolver Create(Dictionary<string, string> sources = null)
        {
            return new TestSourceReferenceResolver(sources.ToDictionary(p => p.Key, p => (object)p.Value));
        }

        private readonly Dictionary<string, object> _sources;

        private TestSourceReferenceResolver(Dictionary<string, object> sources)
        {
            _sources = sources;
        }

        public override string NormalizePath(string path, string baseFilePath) => path;

        public override string ResolveReference(string path, string baseFilePath) =>
            _sources?.ContainsKey(path) == true ? path : null;

        public override Stream OpenRead(string resolvedPath)
        {
            if (_sources != null && resolvedPath != null)
            {
                var data = _sources[resolvedPath];
                return new MemoryStream((data is string) ? Encoding.UTF8.GetBytes((string)data) : (byte[])data);
            }
            else
            {
                throw new IOException();
            }
        }

        public override bool Equals(object other) => ReferenceEquals(this, other);

        public override int GetHashCode() => RuntimeHelpers.GetHashCode(this);
    }
}