ImageSharp/tests/ImageSharp.Tests/TestFileSystem.cs

// Copyright (c) Six Labors and contributors.
// Licensed under the Apache License, Version 2.0.

using System;
using System.Collections.Generic;
using System.IO;

namespace SixLabors.ImageSharp.Tests
{
    /// <summary>
    /// A test image file.
    /// </summary>
    public class TestFileSystem : ImageSharp.IO.IFileSystem
    {

        public static TestFileSystem Global { get; } = new TestFileSystem();

        public static void RegisterGlobalTestFormat()
        {
            Configuration.Default.FileSystem = Global;
        }

        Dictionary<string, Stream> fileSystem = new Dictionary<string, Stream>(StringComparer.OrdinalIgnoreCase);

        public void AddFile(string path, Stream data)
        {
            fileSystem.Add(path, data);
        }

        public Stream Create(string path)
        {
            // if we have injected a fake file use it instead
            lock (fileSystem)
            {
                if (fileSystem.ContainsKey(path))
                {
                    Stream stream = fileSystem[path];
                    stream.Position = 0;
                    return stream;
                }
            }

            return File.Create(path);
        }


        public Stream OpenRead(string path)
        {
            // if we have injected a fake file use it instead
            lock (fileSystem)
            {
                if (fileSystem.ContainsKey(path))
                {
                    Stream stream = fileSystem[path];
                    stream.Position = 0;
                    return stream;
                }
            }

            return File.OpenRead(path);
        }
    }
}