From 73e95c56c18b45d0c513282a549e953f93b5f9c9 Mon Sep 17 00:00:00 2001
From: John Luo <johluo@microsoft.com>
Date: Wed, 29 May 2019 16:22:17 -0700
Subject: [PATCH]  Allow flaky attribute on class and assembly
 (dotnet/extensions#1742)

* Allow flaky attribute on class and assembly\n\nCommit migrated from https://github.com/dotnet/extensions/commit/23ba460e278c3335f70595ebc5b19fe42e80da51
---
 src/ObjectPool/test/DefaultObjectPoolTest.cs | 2 +-
 src/Testing/src/xunit/FlakyAttribute.cs      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/ObjectPool/test/DefaultObjectPoolTest.cs b/src/ObjectPool/test/DefaultObjectPoolTest.cs
index b44aa7e1c7..9bf84dca03 100644
--- a/src/ObjectPool/test/DefaultObjectPoolTest.cs
+++ b/src/ObjectPool/test/DefaultObjectPoolTest.cs
@@ -45,7 +45,7 @@ namespace Microsoft.Extensions.ObjectPool
         {
             // Arrange
             var pool = new DefaultObjectPool<List<int>>(new ListPolicy());
-            
+
             // Act
             var list = pool.Get();
 
diff --git a/src/Testing/src/xunit/FlakyAttribute.cs b/src/Testing/src/xunit/FlakyAttribute.cs
index f58026c7ca..ab4450e685 100644
--- a/src/Testing/src/xunit/FlakyAttribute.cs
+++ b/src/Testing/src/xunit/FlakyAttribute.cs
@@ -50,7 +50,7 @@ namespace Microsoft.AspNetCore.Testing.xunit
     /// </para>
     /// </example>
     [TraitDiscoverer("Microsoft.AspNetCore.Testing.xunit.FlakyTestDiscoverer", "Microsoft.AspNetCore.Testing")]
-    [AttributeUsage(AttributeTargets.Method)]
+    [AttributeUsage(AttributeTargets.Method | AttributeTargets.Class | AttributeTargets.Assembly)]
     public sealed class FlakyAttribute : Attribute, ITraitAttribute
     {
         /// <summary>