Avalonia/tests/Avalonia.UnitTests/UnitTestApplication.cs

// Copyright (c) The Avalonia Project. All rights reserved.
// Licensed under the MIT license. See licence.md file in the project root for full license information.

using System;
using Avalonia.Input;
using Avalonia.Layout;
using Avalonia.Platform;
using Avalonia.Styling;
using Avalonia.Controls;
using Avalonia.Rendering;
using Avalonia.Threading;
using System.Reactive.Disposables;
using System.Reactive.Concurrency;

namespace Avalonia.UnitTests
{
    public class UnitTestApplication : Application
    {
        private readonly TestServices _services;

        public UnitTestApplication(TestServices services)
        {
            _services = services ?? new TestServices();
            RegisterServices();
        }

        public static new UnitTestApplication Current => (UnitTestApplication)Application.Current;

        public TestServices Services => _services;

        public static IDisposable Start(TestServices services = null)
        {
            var scope = AvaloniaLocator.EnterScope();
            var app = new UnitTestApplication(services);
            AvaloniaLocator.CurrentMutable.BindToSelf<Application>(app);
            Dispatcher.UIThread.UpdateServices();
            return Disposable.Create(() =>
            {
                scope.Dispose();
                Dispatcher.UIThread.UpdateServices();
            });
        }

        public override void RegisterServices()
        {
            AvaloniaLocator.CurrentMutable
                .Bind<IAssetLoader>().ToConstant(Services.AssetLoader)
                .Bind<IFocusManager>().ToConstant(Services.FocusManager)
                .BindToSelf<IGlobalStyles>(this)
                .Bind<IInputManager>().ToConstant(Services.InputManager)
                .Bind<IKeyboardDevice>().ToConstant(Services.KeyboardDevice?.Invoke())
                .Bind<IKeyboardNavigationHandler>().ToConstant(Services.KeyboardNavigation)
                .Bind<ILayoutManager>().ToConstant(Services.LayoutManager)
                .Bind<IRuntimePlatform>().ToConstant(Services.Platform)
                .Bind<IRendererFactory>().ToConstant(new RendererFactory(Services.Renderer))
                .Bind<IPlatformRenderInterface>().ToConstant(Services.RenderInterface)
                .Bind<IRenderLoop>().ToConstant(Services.RenderLoop)
                .Bind<IPlatformThreadingInterface>().ToConstant(Services.ThreadingInterface)
                .Bind<IScheduler>().ToConstant(Services.Scheduler)
                .Bind<IStandardCursorFactory>().ToConstant(Services.StandardCursorFactory)
                .Bind<IStyler>().ToConstant(Services.Styler)
                .Bind<IWindowingPlatform>().ToConstant(Services.WindowingPlatform)
                .Bind<IApplicationLifecycle>().ToConstant(this);
            var styles = Services.Theme?.Invoke();

            if (styles != null)
            {
                Styles.AddRange(styles);
            }
        }

        private class RendererFactory : IRendererFactory
        {
            Func<IRenderRoot, IRenderLoop, IRenderer> _func;

            public RendererFactory(Func<IRenderRoot, IRenderLoop, IRenderer> func)
            {
                _func = func;
            }

            public IRenderer CreateRenderer(IRenderRoot root, IRenderLoop renderLoop)
            {
                return _func?.Invoke(root, renderLoop);
            }
        }
    }
}