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
|
// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information.
using System.Threading;
using Xunit;
namespace Microsoft.TestCommon
{
/// <summary>
/// Preserves the current <see cref="SynchronizationContext"/>. Use this attribute on
/// tests which modify the current <see cref="SynchronizationContext"/>.
/// </summary>
public class PreserveSyncContextAttribute : BeforeAfterTestAttribute
{
private SynchronizationContext _syncContext;
public override void Before(System.Reflection.MethodInfo methodUnderTest)
{
_syncContext = SynchronizationContext.Current;
}
public override void After(System.Reflection.MethodInfo methodUnderTest)
{
SynchronizationContext.SetSynchronizationContext(_syncContext);
}
}
}
|
