Fixed tests.

microsoft · Aug 25, 2021 · afabf2c · afabf2c
1 parent 9a621be
commit afabf2c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/UnitTests/MSTest.CoreAdapter.Unit.Tests/Discovery/AssemblyEnumeratorTests.cs b/test/UnitTests/MSTest.CoreAdapter.Unit.Tests/Discovery/AssemblyEnumeratorTests.cs
@@ -424,6 +424,12 @@ private static Mock<TestableAssembly> CreateMockTestableAssembly()
                     true))
                 .Returns(new Attribute[0]);
 
+            mockAssembly
+                .Setup(a => a.GetCustomAttributes(
+                    typeof(FrameworkV2::Microsoft.VisualStudio.TestTools.UnitTesting.TestDataSourceDiscoveryAttribute),
+                    true))
+                .Returns(new Attribute[0]);
+
             return mockAssembly;
         }