Fix test: Add ability to specify CorpusService mock

sillsdev · Oct 16, 2024 · 5a61291 · 5a61291
1 parent 31a31c3
commit 5a61291
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/Machine/src/Serval.Machine.Shared/Services/PreprocessBuildJob.cs b/src/Machine/src/Serval.Machine.Shared/Services/PreprocessBuildJob.cs
@@ -105,7 +105,7 @@ CancellationToken cancellationToken
         int trainCount = 0;
         int pretranslateCount = 0;
         pretranslateWriter.WriteStartArray();
-        ParallelCorpusPreprocessor.PreprocessCorpora(
+        new ParallelCorpusPreprocessor() { CorpusService = CorpusService }.Preprocess(
             corpora,
             row =>
             {

diff --git a/src/ServiceToolkit/src/SIL.ServiceToolkit/Utils/ParallelCorpusPreprocessor.cs b/src/ServiceToolkit/src/SIL.ServiceToolkit/Utils/ParallelCorpusPreprocessor.cs
@@ -2,13 +2,13 @@ namespace SIL.ServiceToolkit.Utils;
 
 public class ParallelCorpusPreprocessor
 {
-    private readonly ICorpusService _corpusService;
+    public ICorpusService CorpusService { get; init; }
     private int _seed = 1234;
     private Random _random;
 
     public ParallelCorpusPreprocessor()
     {
-        _corpusService = new CorpusService();
+        CorpusService = new CorpusService();
         _random = new Random(_seed);
     }
 
@@ -45,7 +45,7 @@ public void Preprocess(
         foreach (ParallelCorpus corpus in corpora)
         {
             (MonolingualCorpus Corpus, ITextCorpus TextCorpus)[] sourceCorpora = corpus
-                .SourceCorpora.SelectMany(c => _corpusService.CreateTextCorpora(c.Files).Select(tc => (c, tc)))
+                .SourceCorpora.SelectMany(c => CorpusService.CreateTextCorpora(c.Files).Select(tc => (c, tc)))
                 .ToArray();
             ITextCorpus[] sourceTrainingCorpora = sourceCorpora
                 .Select(sc =>
@@ -78,7 +78,7 @@ row.Ref is not ScriptureRef sr
                 .ToArray();
 
             (MonolingualCorpus Corpus, ITextCorpus TextCorpus)[] targetCorpora = corpus
-                .TargetCorpora.SelectMany(c => _corpusService.CreateTextCorpora(c.Files).Select(tc => (c, tc)))
+                .TargetCorpora.SelectMany(c => CorpusService.CreateTextCorpora(c.Files).Select(tc => (c, tc)))
                 .ToArray();
             ITextCorpus[] targetTrainingCorpora = targetCorpora
                 .Select(tc =>
@@ -138,10 +138,10 @@ row.Ref is not ScriptureRef sr
 
             if (useKeyTerms)
             {
-                ITextCorpus? sourceTermCorpus = _corpusService
+                ITextCorpus? sourceTermCorpus = CorpusService
                     .CreateTermCorpora(corpus.SourceCorpora.SelectMany(sc => sc.Files).ToList())
                     .FirstOrDefault();
-                ITextCorpus? targetTermCorpus = _corpusService
+                ITextCorpus? targetTermCorpus = CorpusService
                     .CreateTermCorpora(corpus.TargetCorpora.SelectMany(tc => tc.Files).ToList())
                     .FirstOrDefault();
                 if (sourceTermCorpus is not null && targetTermCorpus is not null)