small fixes

sillsdev · Dec 4, 2024 · 5b1d67d · 5b1d67d
1 parent a6650e5
commit 5b1d67d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/DataAccess/src/SIL.DataAccess/MongoRepository.cs b/src/DataAccess/src/SIL.DataAccess/MongoRepository.cs
@@ -162,7 +162,7 @@ public async Task<int> UpdateAllAsync<TFilter>(
         {
             ArrayFilters = [new JsonArrayFilterDefinition<TFilter>(jsonArrayFilterDefinition)]
         };
-        return await UpdateAllAsync(filter, update, updateOptions, cancellationToken);
+        return await UpdateAllAsync(filter, update, updateOptions, cancellationToken).ConfigureAwait(false);
     }
 
     public async Task<int> UpdateAllAsync(

diff --git a/src/Serval/test/Serval.DataFiles.Tests/Services/CorpusServiceTests.cs b/src/Serval/test/Serval.DataFiles.Tests/Services/CorpusServiceTests.cs
@@ -47,9 +47,16 @@ private class TestEnvironment
         public TestEnvironment()
         {
             Corpora = new MemoryRepository<Corpus>();
+            DataAccessContext = Substitute.For<IDataAccessContext>();
+            DataAccessContext
+                .WithTransactionAsync(Arg.Any<Func<CancellationToken, Task<Corpus>>>(), Arg.Any<CancellationToken>())
+                .Returns(x =>
+                {
+                    return ((Func<CancellationToken, Task<Corpus>>)x[0])((CancellationToken)x[1]);
+                });
             Service = new CorpusService(
                 Corpora,
-                Substitute.For<IDataAccessContext>(),
+                DataAccessContext,
                 Substitute.For<IDataFileService>(),
                 Substitute.For<IScopedMediator>()
             );
@@ -58,5 +65,7 @@ public TestEnvironment()
         public MemoryRepository<Corpus> Corpora { get; }
 
         public CorpusService Service { get; }
+
+        public IDataAccessContext DataAccessContext { get; }
     }
 }