From afabf2ccbc3080bb771a6856334a70c74a0187db Mon Sep 17 00:00:00 2001
From: Medeni Baykal <433724+Haplois@users.noreply.github.com>
Date: Wed, 25 Aug 2021 16:53:34 +0200
Subject: [PATCH] Fixed tests.

---
 .../Discovery/AssemblyEnumeratorTests.cs                    | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/test/UnitTests/MSTest.CoreAdapter.Unit.Tests/Discovery/AssemblyEnumeratorTests.cs b/test/UnitTests/MSTest.CoreAdapter.Unit.Tests/Discovery/AssemblyEnumeratorTests.cs
index b0d99d1e98..4564227b1b 100644
--- 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;
         }