File: Filters\ViewDataAttributeApplicationModelProviderTest.cs
Web Access
Project: src\src\Mvc\Mvc.ViewFeatures\test\Microsoft.AspNetCore.Mvc.ViewFeatures.Test.csproj (Microsoft.AspNetCore.Mvc.ViewFeatures.Test)
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Reflection;
using Microsoft.AspNetCore.Mvc.ApplicationModels;
using Microsoft.AspNetCore.Mvc.ModelBinding;
using Microsoft.Extensions.Options;
 
namespace Microsoft.AspNetCore.Mvc.ViewFeatures.Filters;
 
public class ViewDataAttributeApplicationModelProviderTest
{
    [Fact]
    public void OnProvidersExecuting_DoesNotAddFilter_IfTypeHasNoViewDataProperties()
    {
        // Arrange
        var type = typeof(TestController_NoViewDataProperties);
        var provider = new ViewDataAttributeApplicationModelProvider();
        var context = GetContext(type);
 
        // Act
        provider.OnProvidersExecuting(context);
 
        // Assert
        var controller = Assert.Single(context.Result.Controllers);
        Assert.Empty(controller.Filters);
    }
 
    [Fact]
    public void AddsViewDataPropertyFilter_ForViewDataAttributeProperties()
    {
        // Arrange
        var type = typeof(TestController_NullableNonPrimitiveViewDataProperty);
        var provider = new ViewDataAttributeApplicationModelProvider();
        var context = GetContext(type);
 
        // Act
        provider.OnProvidersExecuting(context);
 
        // Assert
        var controller = Assert.Single(context.Result.Controllers);
        Assert.IsType<ControllerViewDataAttributeFilterFactory>(Assert.Single(controller.Filters));
    }
 
    [Fact]
    public void InitializeFilterFactory_WithExpectedPropertyHelpers_ForViewDataAttributeProperties()
    {
        // Arrange
        var expected = typeof(TestController_OneViewDataProperty).GetProperty(nameof(TestController_OneViewDataProperty.Test2));
        var provider = new ViewDataAttributeApplicationModelProvider();
        var context = GetContext(typeof(TestController_OneViewDataProperty));
 
        // Act
        provider.OnProvidersExecuting(context);
        var controller = context.Result.Controllers.SingleOrDefault();
        var filter = Assert.IsType<ControllerViewDataAttributeFilterFactory>(Assert.Single(controller.Filters));
 
        // Assert
        Assert.NotNull(filter);
        var property = Assert.Single(filter.Properties);
        Assert.Same(expected, property.PropertyInfo);
        Assert.Equal("Test2", property.Key);
    }
 
    private static ApplicationModelProviderContext GetContext(Type type)
    {
        var defaultProvider = new DefaultApplicationModelProvider(
            Options.Create(new MvcOptions()),
            new EmptyModelMetadataProvider());
 
        var context = new ApplicationModelProviderContext(new[] { type.GetTypeInfo() });
        defaultProvider.OnProvidersExecuting(context);
        return context;
    }
 
    public class TestController_NoViewDataProperties
    {
        public DateTime? DateTime { get; set; }
    }
 
    public class TestController_NullableNonPrimitiveViewDataProperty
    {
        [ViewData]
        public DateTime? DateTime { get; set; }
    }
 
    public class TestController_OneViewDataProperty
    {
        public string Test { get; set; }
 
        [ViewData]
        public string Test2 { get; set; }
    }
}