59 lines
2.3 KiB
C#
59 lines
2.3 KiB
C#
// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information.
|
|
|
|
namespace System.Data.Entity
|
|
{
|
|
using System.Collections.Generic;
|
|
using System.Data.Entity.TestHelpers;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using Xunit;
|
|
using Xunit.Sdk;
|
|
|
|
/// <summary>
|
|
/// Attribute that is applied to a method to indicate that it is a fact that
|
|
/// should be run by the test runner in partial trust.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// The class containing the method must derive from <see cref="MarshalByRefObject" />.
|
|
///
|
|
/// If the test class is decorated using <see cref="IUseFixture<>" />, the fixture data is
|
|
/// instantiated in the main
|
|
/// <see cref="AppDomain" />
|
|
/// and then serialized to the partial trust
|
|
/// sandbox.
|
|
/// </remarks>
|
|
public class PartialTrustFactAttribute : FactAttribute
|
|
{
|
|
protected override IEnumerable<ITestCommand> EnumerateTestCommands(IMethodInfo method)
|
|
{
|
|
var fixtures = GetFixtures(method.Class);
|
|
|
|
return base.EnumerateTestCommands(method).Select(tc => new PartialTrustCommand(tc, fixtures));
|
|
}
|
|
|
|
private static IDictionary<MethodInfo, object> GetFixtures(ITypeInfo typeUnderTest)
|
|
{
|
|
var fixtures = new Dictionary<MethodInfo, object>();
|
|
|
|
foreach (var @interface in typeUnderTest.Type.GetInterfaces())
|
|
{
|
|
if (@interface.IsGenericType)
|
|
{
|
|
var genericDefinition = @interface.GetGenericTypeDefinition();
|
|
|
|
if (genericDefinition == typeof(IUseFixture<>))
|
|
{
|
|
var dataType = @interface.GetGenericArguments()[0];
|
|
var fixtureData = Activator.CreateInstance(dataType);
|
|
var method = @interface.GetMethod("SetFixture", new[] { dataType });
|
|
|
|
fixtures[method] = fixtureData;
|
|
}
|
|
}
|
|
}
|
|
|
|
return fixtures;
|
|
}
|
|
}
|
|
}
|