AspNetTest.cs

using StackExchange.Profiling.Storage;
using System;
using System.Threading;
using System.Threading.Tasks;
using Xunit;

namespace StackExchange.Profiling.Tests
{
    public abstract class AspNetTest : BaseTest
    {
        /// <summary>
        /// URL that <see cref="GetRequest"/> and <see cref="GetProfiler"/> will hit.
        /// </summary>
        public const string DefaultRequestUrl = "http://localhost/Test.aspx";

        protected AspNetTest(ITestOutputHelper output) : base(output)
        {
            ThreadPool.SetMinThreads(50, 50);
            // Instance per class, so multiple tests swapping the provider don't cause issues here
            // It's not a threading issue of the profiler, but rather tests swapping providers
            MiniProfiler.Configure(Options = new MiniProfilerOptions()
            {
                StopwatchProvider = () => new UnitTestStopwatch(),
                Storage = new MemoryCacheStorage(TimeSpan.FromDays(1))
            });
        }

        /// <summary>
        /// Returns a simulated HTTP request to <paramref name="url"/>.
        /// </summary>
        /// <param name="url">The URL.</param>
        /// <param name="startAndStopProfiler">The start And Stop Profiler.</param>
        /// <returns>the request</returns>
        public IDisposable GetRequest(string url = DefaultRequestUrl, bool startAndStopProfiler = true)
        {
            var result = new Subtext.TestLibrary.HttpSimulator();

            result.SimulateRequest(new Uri(url));

            if (startAndStopProfiler)
            {
                var mp = Options.StartProfiler();
                result.OnBeforeDispose += () => mp?.Stop();
            }

            return result;
        }

        /// <summary>
        /// Returns a profiler for <paramref name="url"/>. Only child steps will take any time,
        /// e.g. when <paramref name="childDepth"/> is 0, the resulting <see cref="MiniProfiler.DurationMilliseconds"/> will be zero.
        /// </summary>
        /// <param name="url">The URL of the request.</param>
        /// <param name="childDepth">number of levels of child steps underneath result's <see cref="MiniProfiler.Root"/>.</param>
        /// <param name="stepMs">Amount of time each step will "do work for" in each step.</param>
        /// <returns>The generated <see cref="MiniProfiler"/>.</returns>
        public MiniProfiler GetProfiler(string url = DefaultRequestUrl, int childDepth = 0, int stepMs = StepTimeMilliseconds)
        {
            using (GetRequest(url, startAndStopProfiler: false))
            {
                var result = Options.StartProfiler(url);
                Assert.NotNull(result);
                AddRecursiveChildren(result, childDepth, stepMs);
                result.Stop();
                return result;
            }
        }

        /// <summary>
        /// Returns a profiler for <paramref name="url"/>. Only child steps will take any time,
        /// e.g. when <paramref name="childDepth"/> is 0, the resulting <see cref="MiniProfiler.DurationMilliseconds"/> will be zero.
        /// </summary>
        /// <param name="url">The URL of the request.</param>
        /// <param name="childDepth">number of levels of child steps underneath result's <see cref="MiniProfiler.Root"/></param>
        /// <param name="stepMs">Amount of time each step will "do work for" in each step</param>
        /// <returns>The generated <see cref="MiniProfiler"/>.</returns>
        public async Task<MiniProfiler> GetProfilerAsync(string url = DefaultRequestUrl, int childDepth = 0, int stepMs = StepTimeMilliseconds)
        {
            using (GetRequest(url, startAndStopProfiler: false))
            {
                var result = Options.StartProfiler();
                Assert.NotNull(result);
                AddRecursiveChildren(result, childDepth, stepMs);
                await result.StopAsync().ConfigureAwait(false);
                return result;
            }
        }
    }
}