// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information. using System.Net.Http; using System.Web.Http.Controllers; using System.Web.Http.Dispatcher; using Moq; using Xunit; using Assert = Microsoft.TestCommon.AssertEx; namespace System.Web.Http.Tracing.Tracers { public class HttpControllerSelectorTracerTest { private static readonly IHttpController _controller = new Mock().Object; private static readonly HttpControllerContext _controllerContext = ContextUtil.CreateControllerContext(instance: _controller); private static readonly HttpRequestMessage _request = _controllerContext.Request; [Fact] public void SelectController_Invokes_Inner_And_Traces() { // Arrange Mock mockControllerDescriptor = new Mock(_controllerContext.Configuration, "AnyController", _controller.GetType()); Mock mockSelector = new Mock(); mockSelector.Setup(b => b.SelectController(It.IsAny())).Returns(mockControllerDescriptor.Object); TestTraceWriter traceWriter = new TestTraceWriter(); HttpControllerSelectorTracer tracer = new HttpControllerSelectorTracer(mockSelector.Object, traceWriter); TraceRecord[] expectedTraces = new TraceRecord[] { new TraceRecord(_request, TraceCategories.ControllersCategory, TraceLevel.Info) { Kind = TraceKind.Begin, Operation = "SelectController" }, new TraceRecord(_request, TraceCategories.ControllersCategory, TraceLevel.Info) { Kind = TraceKind.End, Operation = "SelectController" } }; // Act HttpControllerDescriptor controllerDescriptor = ((IHttpControllerSelector)tracer).SelectController(_request); // Assert Assert.Equal(expectedTraces, traceWriter.Traces, new TraceRecordComparer()); Assert.IsAssignableFrom(controllerDescriptor); } [Fact] public void SelectController_Throws_And_Traces_When_Inner_Throws() { // Arrange Mock mockSelector = new Mock(); InvalidOperationException exception = new InvalidOperationException("test"); mockSelector.Setup(b => b.SelectController(It.IsAny())).Throws(exception); TestTraceWriter traceWriter = new TestTraceWriter(); HttpControllerSelectorTracer tracer = new HttpControllerSelectorTracer(mockSelector.Object, traceWriter); TraceRecord[] expectedTraces = new TraceRecord[] { new TraceRecord(_request, TraceCategories.ControllersCategory, TraceLevel.Info) { Kind = TraceKind.Begin, Operation = "SelectController" }, new TraceRecord(_request, TraceCategories.ControllersCategory, TraceLevel.Error) { Kind = TraceKind.End, Operation = "SelectController" } }; // Act Exception thrown = Assert.Throws(() => ((IHttpControllerSelector)tracer).SelectController(_request)); // Assert Assert.Equal(expectedTraces, traceWriter.Traces, new TraceRecordComparer()); Assert.Same(exception, thrown); Assert.Same(exception, traceWriter.Traces[1].Exception); } } }