Fix a few tests

eventflow · Nov 11, 2022 · f82e189 · f82e189
1 parent 974e82f
commit f82e189
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Source/EventFlow.Tests/UnitTests/ReadStores/BaseReadModelTests.cs b/Source/EventFlow.Tests/UnitTests/ReadStores/BaseReadModelTests.cs
@@ -75,7 +75,7 @@ public void SetUp()
 
             _eventStoreMock
                 .Setup(s => s.LoadAllEventsAsync(It.IsAny<GlobalPosition>(), It.IsAny<int>(), It.IsAny<IEventUpgradeContext>(), It.IsAny<CancellationToken>()))
-                .Returns<GlobalPosition, int, CancellationToken>((s, p, c) => Task.FromResult(GetEvents(s, p)));
+                .Returns<GlobalPosition, int, IEventUpgradeContext, CancellationToken>((s, p, uc, c) => Task.FromResult(GetEvents(s, p)));
             _readStoreManagerMock
                 .Setup(m => m.ReadModelType)
                 .Returns(typeof(TReadModel));

diff --git a/Source/EventFlow.Tests/UnitTests/ReadStores/ReadModelPopulatorTests.cs b/Source/EventFlow.Tests/UnitTests/ReadStores/ReadModelPopulatorTests.cs
@@ -21,8 +21,12 @@
 // IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 // CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
+using EventFlow.TestHelpers;
+using NUnit.Framework;
+
 namespace EventFlow.Tests.UnitTests.ReadStores
 {
+    [Category(Categories.Unit)]
     public class ReadModelPopulatorTests : BaseReadModelTests<TestReadModel>
     {
     }