pull/2945/head
Mark Monteiro
5 years ago
parent
51b610b999
commit
307754a0e0
@ -0,0 +1,124 @@
|
||||
using System;
|
||||
using System.Collections.Concurrent;
|
||||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Threading.Tasks;
|
||||
using Emby.Server.Implementations;
|
||||
using Emby.Server.Implementations.IO;
|
||||
using Emby.Server.Implementations.Networking;
|
||||
using Jellyfin.Drawing.Skia;
|
||||
using Jellyfin.Server;
|
||||
using MediaBrowser.Common;
|
||||
using Microsoft.AspNetCore.Hosting;
|
||||
using Microsoft.AspNetCore.Mvc.Testing;
|
||||
using Microsoft.AspNetCore.TestHost;
|
||||
using Microsoft.Extensions.DependencyInjection;
|
||||
using Microsoft.Extensions.Hosting;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Serilog;
|
||||
using Serilog.Extensions.Logging;
|
||||
|
||||
namespace MediaBrowser.Api.Tests
|
||||
{
|
||||
/// <summary>
|
||||
/// Factory for bootstrapping the Jellyfin application in memory for functional end to end tests.
|
||||
/// </summary>
|
||||
public class JellyfinApplicationFactory : WebApplicationFactory<Startup>
|
||||
{
|
||||
private static readonly string _testPathRoot = Path.Combine(Path.GetTempPath(), "jellyfin-test-data");
|
||||
private static readonly ConcurrentBag<ApplicationHost> _appHosts = new ConcurrentBag<ApplicationHost>();
|
||||
|
||||
/// <summary>
|
||||
/// Initializes a new instance of <see cref="JellyfinApplicationFactory"/>.
|
||||
/// </summary>
|
||||
public JellyfinApplicationFactory()
|
||||
{
|
||||
// Perform static initialization that only needs to happen once per test-run
|
||||
Log.Logger = new LoggerConfiguration().WriteTo.Console().CreateLogger();
|
||||
Program.PerformStaticInitialization();
|
||||
}
|
||||
|
||||
/// <inheritdoc/>
|
||||
protected override IWebHostBuilder CreateWebHostBuilder()
|
||||
{
|
||||
return new WebHostBuilder();
|
||||
}
|
||||
|
||||
/// <inheritdoc/>
|
||||
protected override void ConfigureWebHost(IWebHostBuilder builder)
|
||||
{
|
||||
// Specify the startup command line options
|
||||
var commandLineOpts = new StartupOptions
|
||||
{
|
||||
NoWebClient = true,
|
||||
NoAutoRunWebApp = true
|
||||
};
|
||||
|
||||
// Use a temporary directory for the application paths
|
||||
var webHostPathRoot = Path.Combine(_testPathRoot, "test-host-" + Path.GetFileNameWithoutExtension(Path.GetRandomFileName()));
|
||||
Directory.CreateDirectory(Path.Combine(webHostPathRoot, "logs"));
|
||||
Directory.CreateDirectory(Path.Combine(webHostPathRoot, "config"));
|
||||
Directory.CreateDirectory(Path.Combine(webHostPathRoot, "cache"));
|
||||
Directory.CreateDirectory(Path.Combine(webHostPathRoot, "jellyfin-web"));
|
||||
var appPaths = new ServerApplicationPaths(
|
||||
webHostPathRoot,
|
||||
Path.Combine(webHostPathRoot, "logs"),
|
||||
Path.Combine(webHostPathRoot, "config"),
|
||||
Path.Combine(webHostPathRoot, "cache"),
|
||||
Path.Combine(webHostPathRoot, "jellyfin-web"));
|
||||
|
||||
// Create the logging config file
|
||||
// TODO: We shouldn't need to do this since we are only logging to console
|
||||
Program.InitLoggingConfigFile(appPaths).Wait();
|
||||
|
||||
// Create a copy of the application configuration to use for startup
|
||||
var startupConfig = Program.CreateAppConfiguration(commandLineOpts, appPaths);
|
||||
|
||||
// Create the app host and initialize it
|
||||
ILoggerFactory loggerFactory = new SerilogLoggerFactory();
|
||||
var appHost = new CoreAppHost(
|
||||
appPaths,
|
||||
loggerFactory,
|
||||
commandLineOpts,
|
||||
new ManagedFileSystem(loggerFactory.CreateLogger<ManagedFileSystem>(), appPaths),
|
||||
new SkiaEncoder(loggerFactory.CreateLogger<SkiaEncoder>(), appPaths),
|
||||
new NetworkManager(loggerFactory.CreateLogger<NetworkManager>()));
|
||||
_appHosts.Add(appHost);
|
||||
var serviceCollection = new ServiceCollection();
|
||||
appHost.InitAsync(serviceCollection, startupConfig).Wait();
|
||||
|
||||
// Configure the web host builder
|
||||
Program.ConfigureWebHostBuilder(builder, appHost, serviceCollection, commandLineOpts, startupConfig, appPaths);
|
||||
}
|
||||
|
||||
/// <inheritdoc/>
|
||||
protected override TestServer CreateServer(IWebHostBuilder builder)
|
||||
{
|
||||
// Create the test server using the base implementation
|
||||
var testServer = base.CreateServer(builder);
|
||||
|
||||
// Finish initializing the app host
|
||||
var appHost = (CoreAppHost)testServer.Services.GetRequiredService<IApplicationHost>();
|
||||
appHost.ServiceProvider = testServer.Services;
|
||||
appHost.InitializeServices();
|
||||
appHost.FindParts();
|
||||
appHost.RunStartupTasksAsync().Wait();
|
||||
|
||||
return testServer;
|
||||
}
|
||||
|
||||
/// <inheritdoc/>
|
||||
protected override void Dispose(bool disposing)
|
||||
{
|
||||
foreach (var host in _appHosts)
|
||||
{
|
||||
host.Dispose();
|
||||
}
|
||||
|
||||
_appHosts.Clear();
|
||||
|
||||
base.Dispose(disposing);
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue