1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135
|
// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information.
using System;
using System.Collections.Generic;
using System.Data.SqlClient;
using System.Net;
using System.Net.Mail;
using System.Reflection;
using System.Security;
using System.Security.Permissions;
using System.Web;
using Xunit;
using Xunit.Sdk;
namespace Microsoft.TestCommon
{
public class PartialTrustRunner : ITestClassCommand
{
private AppDomain sandbox;
// Delegate most of the work to the existing TestClassCommand class so that we
// can preserve any existing behavior (like supporting IUseFixture<T>).
private readonly TestClassCommand originalTestClassCommand = new TestClassCommand();
public int ChooseNextTest(ICollection<IMethodInfo> testsLeftToRun)
{
return this.originalTestClassCommand.ChooseNextTest(testsLeftToRun);
}
public Exception ClassFinish()
{
Exception result = this.originalTestClassCommand.ClassFinish();
if (this.sandbox != null)
{
AppDomain.Unload(this.sandbox);
this.sandbox = null;
}
return result;
}
public Exception ClassStart()
{
this.GuardTypeUnderTest();
Assembly xunitAssembly = typeof(FactAttribute).Assembly;
this.sandbox = CreatePartialTrustAppDomain();
return this.originalTestClassCommand.ClassStart();
}
private static AppDomain CreatePartialTrustAppDomain()
{
PermissionSet permissions = new PermissionSet(PermissionState.None);
permissions.AddPermission(new AspNetHostingPermission(AspNetHostingPermissionLevel.Medium));
permissions.AddPermission(new DnsPermission(PermissionState.Unrestricted));
permissions.AddPermission(new EnvironmentPermission(EnvironmentPermissionAccess.Read, "TEMP;TMP;USERNAME;OS;COMPUTERNAME"));
permissions.AddPermission(new FileIOPermission(FileIOPermissionAccess.AllAccess, AppDomain.CurrentDomain.BaseDirectory));
permissions.AddPermission(new IsolatedStorageFilePermission(PermissionState.None) { UsageAllowed = IsolatedStorageContainment.AssemblyIsolationByUser, UserQuota = Int64.MaxValue });
permissions.AddPermission(new SecurityPermission(SecurityPermissionFlag.Execution));
permissions.AddPermission(new SecurityPermission(SecurityPermissionFlag.ControlThread));
permissions.AddPermission(new SecurityPermission(SecurityPermissionFlag.ControlPrincipal));
permissions.AddPermission(new SecurityPermission(SecurityPermissionFlag.RemotingConfiguration));
permissions.AddPermission(new SmtpPermission(SmtpAccess.Connect));
permissions.AddPermission(new SqlClientPermission(PermissionState.Unrestricted));
permissions.AddPermission(new TypeDescriptorPermission(PermissionState.Unrestricted));
permissions.AddPermission(new WebPermission(PermissionState.Unrestricted));
permissions.AddPermission(new ReflectionPermission(ReflectionPermissionFlag.RestrictedMemberAccess));
AppDomainSetup setup = new AppDomainSetup() { ApplicationBase = AppDomain.CurrentDomain.BaseDirectory };
setup.PartialTrustVisibleAssemblies = new string[]
{
"System.Web, PublicKey=002400000480000094000000060200000024000052534131000400000100010007d1fa57c4aed9f0a32e84aa0faefd0de9e8fd6aec8f87fb03766c834c99921eb23be79ad9d5dcc1dd9ad236132102900b723cf980957fc4e177108fc607774f29e8320e92ea05ece4e821c0a5efe8f1645c4c0c93c1ab99285d622caa652c1dfad63d745d6f2de5f17e5eaf0fc4963d261c8a12436518206dc093344d5ad293",
"System.Web.Extensions, PublicKey=0024000004800000940000000602000000240000525341310004000001000100b5fc90e7027f67871e773a8fde8938c81dd402ba65b9201d60593e96c492651e889cc13f1415ebb53fac1131ae0bd333c5ee6021672d9718ea31a8aebd0da0072f25d87dba6fc90ffd598ed4da35e44c398c454307e8e33b8426143daec9f596836f97c8f74750e5975c64e2189f45def46b2a2b1247adc3652bf5c308055da9",
"System.Web.Abstractions, PublicKey=0024000004800000940000000602000000240000525341310004000001000100b5fc90e7027f67871e773a8fde8938c81dd402ba65b9201d60593e96c492651e889cc13f1415ebb53fac1131ae0bd333c5ee6021672d9718ea31a8aebd0da0072f25d87dba6fc90ffd598ed4da35e44c398c454307e8e33b8426143daec9f596836f97c8f74750e5975c64e2189f45def46b2a2b1247adc3652bf5c308055da9",
"System.Web.Routing, PublicKey=0024000004800000940000000602000000240000525341310004000001000100b5fc90e7027f67871e773a8fde8938c81dd402ba65b9201d60593e96c492651e889cc13f1415ebb53fac1131ae0bd333c5ee6021672d9718ea31a8aebd0da0072f25d87dba6fc90ffd598ed4da35e44c398c454307e8e33b8426143daec9f596836f97c8f74750e5975c64e2189f45def46b2a2b1247adc3652bf5c308055da9",
"System.ComponentModel.DataAnnotations, PublicKey=0024000004800000940000000602000000240000525341310004000001000100b5fc90e7027f67871e773a8fde8938c81dd402ba65b9201d60593e96c492651e889cc13f1415ebb53fac1131ae0bd333c5ee6021672d9718ea31a8aebd0da0072f25d87dba6fc90ffd598ed4da35e44c398c454307e8e33b8426143daec9f596836f97c8f74750e5975c64e2189f45def46b2a2b1247adc3652bf5c308055da9",
"System.Web.DynamicData, PublicKey=0024000004800000940000000602000000240000525341310004000001000100b5fc90e7027f67871e773a8fde8938c81dd402ba65b9201d60593e96c492651e889cc13f1415ebb53fac1131ae0bd333c5ee6021672d9718ea31a8aebd0da0072f25d87dba6fc90ffd598ed4da35e44c398c454307e8e33b8426143daec9f596836f97c8f74750e5975c64e2189f45def46b2a2b1247adc3652bf5c308055da9",
"System.Web.DataVisualization, PublicKey=0024000004800000940000000602000000240000525341310004000001000100b5fc90e7027f67871e773a8fde8938c81dd402ba65b9201d60593e96c492651e889cc13f1415ebb53fac1131ae0bd333c5ee6021672d9718ea31a8aebd0da0072f25d87dba6fc90ffd598ed4da35e44c398c454307e8e33b8426143daec9f596836f97c8f74750e5975c64e2189f45def46b2a2b1247adc3652bf5c308055da9",
"System.Web.ApplicationServices, PublicKey=0024000004800000940000000602000000240000525341310004000001000100b5fc90e7027f67871e773a8fde8938c81dd402ba65b9201d60593e96c492651e889cc13f1415ebb53fac1131ae0bd333c5ee6021672d9718ea31a8aebd0da0072f25d87dba6fc90ffd598ed4da35e44c398c454307e8e33b8426143daec9f596836f97c8f74750e5975c64e2189f45def46b2a2b1247adc3652bf5c308055da9"
};
return AppDomain.CreateDomain("Partial Trust Sandbox", null, setup, permissions);
}
public IEnumerable<ITestCommand> EnumerateTestCommands(IMethodInfo testMethod)
{
return this.originalTestClassCommand.EnumerateTestCommands(testMethod);
}
public IEnumerable<IMethodInfo> EnumerateTestMethods()
{
return this.originalTestClassCommand.EnumerateTestMethods();
}
public bool IsTestMethod(IMethodInfo testMethod)
{
return this.originalTestClassCommand.IsTestMethod(testMethod);
}
public object ObjectUnderTest
{
get
{
return sandbox.CreateInstanceAndUnwrap(this.TypeUnderTest.Type.Assembly.FullName, this.TypeUnderTest.Type.FullName);
}
}
public ITypeInfo TypeUnderTest
{
get
{
return this.originalTestClassCommand.TypeUnderTest;
}
set
{
if (!typeof(MarshalByRefObject).IsAssignableFrom(value.Type))
{
throw new InvalidOperationException("Test types to be run in PT must derive from MarshalByRefObject");
}
this.originalTestClassCommand.TypeUnderTest = value;
}
}
private void GuardTypeUnderTest()
{
if (TypeUnderTest == null)
{
throw new InvalidOperationException("Forgot to set TypeUnderTest before calling ObjectUnderTest");
}
}
}
}
|