Atata.AspNetCore

builder.Sessions.AddWebApplication(x => x
    .Use<Program>());

builder.Sessions.AddWebApplication(x => x
    .Use(() => new WebApplicationFactory<Program>()));

var containerSession = await Context.Sessions.AddWebApplication()
    .Use<Program>()
    .BuildAsync();

var containerSession = await Context.Sessions.AddWebApplication()
    .Use(() => new WebApplicationFactory<Program>())
    .BuildAsync();

public sealed class CustomWebApplicationFactory : WebApplicationFactory<Program>
{
    protected override void ConfigureWebHost(IWebHostBuilder builder)
    {
        base.ConfigureWebHost(builder);

        builder.UseSetting("SomeKey", "SomeValue");

        builder.ConfigureTestServices(services =>
        {
            // Custom service configuration can be done here.
        });
    }
}

builder.Sessions.AddWebApplication(x => x
    .Use(new CustomWebApplicationFactory()));

public class CustomWebApplicationSession : WebApplicationSession
{
    protected override void ConfigureWebHost(IWebHostBuilder builder)
    {
        base.ConfigureWebHost(builder);

        builder.ConfigureTestServices(services =>
        {
            // Custom service configuration can be done here.
        });
    }
}

builder.Sessions.AddWebApplication<CustomWebApplicationSession>(x => x
    .Use<Program>());

var session = Context.Sessions.Get<WebApplicationSession>();

public static class WebApplicationSessionAtataSessionsBuilderExtensions
{
    // Adds a new instance of WebApplicationSessionBuilder builder.
    public static AtataContextBuilder AddWebApplication(
        this AtataSessionsBuilder builder,
        Action<WebApplicationSessionBuilder>? configure = null);

    // Adds a new instance of WebApplicationSessionBuilder<TSession> builder.
    public static AtataContextBuilder AddWebApplication<TSession>(
        this AtataSessionsBuilder builder,
        Action<WebApplicationSessionBuilder<TSession>>? configure = null)
        where TSession : WebApplicationSession, new();

    // Configures existing nameless WebApplicationSessionBuilder session builder.
    public static AtataContextBuilder ConfigureWebApplication(
        this AtataSessionsBuilder builder,
        Action<WebApplicationSessionBuilder> configure,
        ConfigurationMode mode = default);

    // Configures existing nameless WebApplicationSessionBuilder<TSession> session builder.
    public static AtataContextBuilder ConfigureWebApplication<TSession>(
        this AtataSessionsBuilder builder,
        Action<WebApplicationSessionBuilder<TSession>> configure,
        ConfigurationMode mode = default)
        where TSession : WebApplicationSession, new();

    // Configures existing WebApplicationSessionBuilder session builder that has the specified name.
    public static AtataContextBuilder ConfigureWebApplication(
        this AtataSessionsBuilder builder,
        string? name,
        Action<WebApplicationSessionBuilder> configure,
        ConfigurationMode mode = default);

    // Configures existing WebApplicationSessionBuilder<TSession> session builder that has the specified name.
    public static AtataContextBuilder ConfigureWebApplication<TSession>(
        this AtataSessionsBuilder builder,
        string? name,
        Action<WebApplicationSessionBuilder<TSession>> configure,
        ConfigurationMode mode = default)
        where TSession : WebApplicationSession, new();
}

public static class WebApplicationSessionAtataSessionCollectionExtensions
{
    // Creates a new WebApplicationSessionBuilder and adds it to the collection.
    public static WebApplicationSessionBuilder AddWebApplication(
        this AtataSessionCollection collection,
        Action<WebApplicationSessionBuilder>? configure = null);

    // Creates a new WebApplicationSessionBuilder<TSession> and adds it to the collection.
    public static WebApplicationSessionBuilder<TSession> AddWebApplication<TSession>(
        this AtataSessionCollection collection,
        Action<WebApplicationSessionBuilder<TSession>>? configure = null)
        where TSession : WebApplicationSession, new();
}

public abstract class WebApplicationSessionBuilder<TSession, TBuilder> : AtataSessionBuilder<TSession, TBuilder>
    where TSession : WebApplicationSession, new()
    where TBuilder : WebApplicationSessionBuilder<TSession, TBuilder>
{
    // Gets or sets a value indicating whether to dispose the WebApplicationFactory<TEntryPoint>
    // when AtataSession.DisposeAsync method is invoked.
    // The default value is true.
    public bool DisposeWebApplicationFactory { get; set; }

    // Gets or sets a value indicating whether to collect application logs.
    // The default value is true.
    public bool CollectApplicationLogs { get; set; }

    // Gets or sets a value indicating whether to transmit application logs to Atata.
    // The default value is true.
    // When true, adds an AtataLoggerProvider instance, with SourceNameForAtataLog as Atata log source
    // and MinimumLogLevelForAtataLog as minimum log level, to the application builder's ILoggingBuilder.
    public bool TransmitApplicationLogsToAtata { get; set; }

    // Gets or sets a value indicating whether to clear predefined application logging providers.
    // The default value is true.
    // When true, calls ClearProviders method for application builder's ILoggingBuilder.
    public bool ClearPredefinedApplicationLoggingProviders { get; set; }

    // Gets or sets the source name for Atata log.
    // The default value is "App".
    public string SourceNameForAtataLog { get; set; }

    // Gets or sets the default application log level.
    // The default value is null, which doesn't change the predefined application configuration.
    // When a value is set, adds "Logging:LogLevel:Default" to the application configuration.
    public MSLogLevel? DefaultApplicationLogLevel { get; set; }

    // Gets or sets the minimum application log level.
    // The default value is null, which doesn't change the predefined application configuration.
    // When a value is set, calls SetMinimumLevel method for application builder's ILoggingBuilder.
    public MSLogLevel? MinimumApplicationLogLevel { get; set; }

    // Gets or sets the minimum log level for Atata log.
    // The default value is MSLogLevel.Trace.
    public MSLogLevel MinimumLogLevelForAtataLog { get; set; }

    // Sets the WebApplicationFactory<TEntryPoint> to use for the session.
    public TBuilder Use<TEntryPoint>()
        where TEntryPoint : class;

    // Sets the specified WebApplicationFactory<TEntryPoint> instance to use for the session.
    public TBuilder Use<TEntryPoint>(WebApplicationFactory<TEntryPoint> webApplicationFactory)
        where TEntryPoint : class;

    // Configures and sets a new WebApplicationFactory<TEntryPoint> instance to use for the session.
    public TBuilder Use<TEntryPoint>(Action<WebApplicationFactory<TEntryPoint>> configure)
        where TEntryPoint : class;

    // Sets the WebApplicationFactory<TEntryPoint> to use for the session using a factory method.
    public TBuilder Use<TEntryPoint>(Func<WebApplicationFactory<TEntryPoint>> webApplicationFactoryCreator)
        where TEntryPoint : class;

    // Adds a configuration action to be applied to the IWebHostBuilder.
    public TBuilder UseConfiguration(Action<IWebHostBuilder> configure);

    // Sets a value indicating whether to use Kestrel as the server.
    // The default value is false.
    public TBuilder UseKestrel(bool useKestrel = true);

    // Configures the builder to use Kestrel as the server on the specified port.
    public TBuilder UseKestrel(int port);

    // Configures the builder to use Kestrel as the server.
    public TBuilder UseKestrel(Action<KestrelServerOptions> configureKestrelOptions);

    // Configures the builder to start only when the Kestrel server port is available.
    public TBuilder UseStartWhenKestrelPortIsAvailable();

    // Sets a value indicating whether to dispose the WebApplicationFactory<TEntryPoint>
    // when AtataSession.DisposeAsync method is invoked.
    // The default value is true.
    public TBuilder UseDisposeWebApplicationFactory(bool disposeFactory);

    // Sets a value indicating whether to collect application logs.
    // The default value is true.
    public TBuilder UseCollectApplicationLogs(bool collectLogs);

    // Sets a value indicating whether to transmit application logs to Atata.
    // The default value is true.
    // When true, adds an AtataLoggerProvider instance, with SourceNameForAtataLog as Atata log source
    // and MinimumLogLevelForAtataLog as minimum log level, to the application builder's ILoggingBuilder.
    public TBuilder UseTransmitApplicationLogsToAtata(bool transmitLogs);

    // Sets a value indicating whether to clear predefined application logging providers.
    // The default value is true.
    // When true, calls ClearProviders method for application builder's ILoggingBuilder.
    public TBuilder UseClearPredefinedApplicationLoggingProviders(bool clearProviders);

    // Sets the source name for Atata log.
    // The default value is "App".
    public TBuilder UseSourceNameForAtataLog(string sourceName);

    // Sets the default application log level.
    // The default value is null, which doesn't change the predefined application configuration.
    // When a value is set, adds "Logging:LogLevel:Default" to the application configuration.
    public TBuilder UseDefaultApplicationLogLevel(MSLogLevel? logLevel);

    // Sets the minimum application log level.
    // The default value is null, which doesn't change the predefined application configuration.
    // When a value is set, calls SetMinimumLevel method for application builder's ILoggingBuilder.
    public TBuilder UseMinimumApplicationLogLevel(MSLogLevel? logLevel);

    /// Sets the minimum log level for Atata log.
    public TBuilder UseMinimumLogLevelForAtataLog(MSLogLevel logLevel);
}

public class WebApplicationSessionBuilder<TSession> : WebApplicationSessionBuilder<TSession, WebApplicationSessionBuilder<TSession>>
    where TSession : WebApplicationSession, new()
{
}

public class WebApplicationSessionBuilder : WebApplicationSessionBuilder<WebApplicationSession, WebApplicationSessionBuilder>
{
}

public class WebApplicationSession : AtataSession
{
    public WebApplicationSession();

    // Gets the Uri of the started web application.
    public Uri Uri { get; }

    // Gets the URL string of the started web application.
    public string Url { get; }

    // Gets the Microsoft.AspNetCore.TestHost.TestServer instance associated with the session.
    // In case of Kestrel server, the property will be null.
    public TestServer? TestServer { get; }

    // Gets the IServiceProvider for resolving application services.
    public IServiceProvider Services { get; }

    // Gets the client options used for HTTP client creation.
    public WebApplicationFactoryClientOptions ClientOptions { get; }

    // Gets the Microsoft.Extensions.Logging.Testing.FakeLogCollector of application.
    public FakeLogCollector FakeLogCollector { get; }

    // Creates WebApplicationSessionBuilder instance for WebApplicationSession configuration.
    public static WebApplicationSessionBuilder CreateBuilder();

    public HttpClient CreateClient();

    public HttpClient CreateClient(WebApplicationFactoryClientOptions options);

    public HttpClient CreateDefaultClient(params DelegatingHandler[] handlers);

    public HttpClient CreateDefaultClient(Uri baseAddress, params DelegatingHandler[] handlers);

    // Configures the web host builder for the session.
    // Override this method to customize the IWebHostBuilder used to create the application.
    protected virtual void ConfigureWebHost(IWebHostBuilder builder);
}