100 lines
3.5 KiB
C#
100 lines
3.5 KiB
C#
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using Microsoft.AspNetCore.Mvc.ApplicationModels;
|
|
using Microsoft.AspNetCore.Mvc.Internal;
|
|
using Microsoft.AspNetCore.Mvc.ModelBinding;
|
|
using Microsoft.Extensions.Options;
|
|
using Xunit;
|
|
|
|
namespace Microsoft.AspNetCore.Mvc.ViewFeatures.Internal
|
|
{
|
|
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; }
|
|
}
|
|
}
|
|
}
|