32 lines
921 B
C#
32 lines
921 B
C#
// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using System.Reflection;
|
|
using System.Security.Principal;
|
|
using System.Threading;
|
|
using Xunit;
|
|
|
|
namespace Microsoft.TestCommon
|
|
{
|
|
public class RestoreThreadPrincipalAttribute : BeforeAfterTestAttribute
|
|
{
|
|
private IPrincipal _originalPrincipal;
|
|
|
|
public override void Before(MethodInfo methodUnderTest)
|
|
{
|
|
_originalPrincipal = Thread.CurrentPrincipal;
|
|
|
|
AppDomain.CurrentDomain.SetPrincipalPolicy(PrincipalPolicy.NoPrincipal);
|
|
|
|
Thread.CurrentPrincipal = null;
|
|
}
|
|
|
|
public override void After(MethodInfo methodUnderTest)
|
|
{
|
|
Thread.CurrentPrincipal = _originalPrincipal;
|
|
|
|
AppDomain.CurrentDomain.SetPrincipalPolicy(PrincipalPolicy.UnauthenticatedPrincipal);
|
|
}
|
|
}
|
|
}
|