| | | 1 | | // Licensed to the .NET Foundation under one or more agreements. |
| | | 2 | | // The .NET Foundation licenses this file to you under the MIT license. |
| | | 3 | | |
| | | 4 | | using System; |
| | | 5 | | using System.Globalization; |
| | | 6 | | using System.Reflection; |
| | | 7 | | using System.Threading; |
| | | 8 | | using Xunit.v3; |
| | | 9 | | |
| | | 10 | | /// <summary> |
| | | 11 | | /// Apply this attribute to your test method to replace the |
| | | 12 | | /// <see cref="Thread.CurrentThread" /> <see cref="CultureInfo.CurrentCulture" /> and |
| | | 13 | | /// <see cref="CultureInfo.CurrentUICulture" /> with another culture. |
| | | 14 | | /// </summary> |
| | | 15 | | [AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, AllowMultiple = false, Inherited = true)] |
| | | 16 | | public class UseCultureAttribute : BeforeAfterTestAttribute |
| | | 17 | | { |
| | | 18 | | readonly Lazy<CultureInfo> culture; |
| | | 19 | | readonly Lazy<CultureInfo> uiCulture; |
| | | 20 | | |
| | | 21 | | CultureInfo originalCulture; |
| | | 22 | | CultureInfo originalUICulture; |
| | | 23 | | |
| | | 24 | | /// <summary> |
| | | 25 | | /// Replaces the culture and UI culture of the current thread with |
| | | 26 | | /// <paramref name="culture" /> |
| | | 27 | | /// </summary> |
| | | 28 | | /// <param name="culture">The name of the culture.</param> |
| | | 29 | | /// <remarks> |
| | | 30 | | /// <para> |
| | | 31 | | /// This constructor overload uses <paramref name="culture" /> for both |
| | | 32 | | /// <see cref="Culture" /> and <see cref="UICulture" />. |
| | | 33 | | /// </para> |
| | | 34 | | /// </remarks> |
| | | 35 | | public UseCultureAttribute(string culture) |
| | 8 | 36 | | : this(culture, culture) { } |
| | | 37 | | |
| | | 38 | | /// <summary> |
| | | 39 | | /// Replaces the culture and UI culture of the current thread with |
| | | 40 | | /// <paramref name="culture" /> and <paramref name="uiCulture" /> |
| | | 41 | | /// </summary> |
| | | 42 | | /// <param name="culture">The name of the culture.</param> |
| | | 43 | | /// <param name="uiCulture">The name of the UI culture.</param> |
| | 4 | 44 | | public UseCultureAttribute(string culture, string uiCulture) |
| | | 45 | | { |
| | 7 | 46 | | this.culture = new Lazy<CultureInfo>(() => new CultureInfo(culture, false)); |
| | 7 | 47 | | this.uiCulture = new Lazy<CultureInfo>(() => new CultureInfo(uiCulture, false)); |
| | 4 | 48 | | } |
| | | 49 | | |
| | | 50 | | /// <summary> |
| | | 51 | | /// Gets the culture. |
| | | 52 | | /// </summary> |
| | 3 | 53 | | public CultureInfo Culture { get { return culture.Value; } } |
| | | 54 | | |
| | | 55 | | /// <summary> |
| | | 56 | | /// Gets the UI culture. |
| | | 57 | | /// </summary> |
| | 3 | 58 | | public CultureInfo UICulture { get { return uiCulture.Value; } } |
| | | 59 | | |
| | | 60 | | /// <summary> |
| | | 61 | | /// Stores the current <see cref="Thread.CurrentPrincipal" /> |
| | | 62 | | /// <see cref="CultureInfo.CurrentCulture" /> and <see cref="CultureInfo.CurrentUICulture" /> |
| | | 63 | | /// and replaces them with the new cultures defined in the constructor. |
| | | 64 | | /// </summary> |
| | | 65 | | /// <param name="methodUnderTest">The method under test</param> |
| | | 66 | | /// <param name="test">The test being run</param> |
| | | 67 | | public override void Before(MethodInfo methodUnderTest, IXunitTest test) |
| | | 68 | | { |
| | 3 | 69 | | originalCulture = Thread.CurrentThread.CurrentCulture; |
| | 3 | 70 | | originalUICulture = Thread.CurrentThread.CurrentUICulture; |
| | | 71 | | |
| | 3 | 72 | | Thread.CurrentThread.CurrentCulture = Culture; |
| | 3 | 73 | | Thread.CurrentThread.CurrentUICulture = UICulture; |
| | | 74 | | |
| | 3 | 75 | | CultureInfo.CurrentCulture.ClearCachedData(); |
| | 3 | 76 | | CultureInfo.CurrentUICulture.ClearCachedData(); |
| | 3 | 77 | | } |
| | | 78 | | |
| | | 79 | | /// <summary> |
| | | 80 | | /// Restores the original <see cref="CultureInfo.CurrentCulture" /> and |
| | | 81 | | /// <see cref="CultureInfo.CurrentUICulture" /> to <see cref="Thread.CurrentPrincipal" /> |
| | | 82 | | /// </summary> |
| | | 83 | | /// <param name="methodUnderTest">The method under test</param> |
| | | 84 | | /// <param name="test">The test being run</param> |
| | | 85 | | public override void After(MethodInfo methodUnderTest, IXunitTest test) |
| | | 86 | | { |
| | 3 | 87 | | Thread.CurrentThread.CurrentCulture = originalCulture; |
| | 3 | 88 | | Thread.CurrentThread.CurrentUICulture = originalUICulture; |
| | | 89 | | |
| | 3 | 90 | | CultureInfo.CurrentCulture.ClearCachedData(); |
| | 3 | 91 | | CultureInfo.CurrentUICulture.ClearCachedData(); |
| | 3 | 92 | | } |
| | | 93 | | } |