Fix unit tests

This commit is contained in:
0xb00bface 2023-04-25 19:23:54 +02:00
parent 86e8f00aaf
commit 19850eab31
2 changed files with 9 additions and 2 deletions

View File

@ -12,6 +12,7 @@ import java.time.ZoneId;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;
import java.util.Locale;
import java.util.Optional;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.any;
@ -38,7 +39,8 @@ class AbstractPlaceholderAwarePostProcessorTest extends AbstractPpTest {
config = mockConfig();
placeHolderAwarePp = new Move();
modelNotesService = mock(ModelNotesService.class);
when(modelNotesService.loadModelNotes(any())).thenReturn(null);
when(modelNotesService.loadModelNotes(any())).thenReturn(Optional.of("tag,_foo,_bar"));
}
@Test
@ -137,7 +139,7 @@ class AbstractPlaceholderAwarePostProcessorTest extends AbstractPpTest {
@Test
void testModelNotesReplacement() {
String input = "asdf_${modelNotes}_asdf";
assertEquals("asdf_tag,_foo,_bar_asdf", placeHolderAwarePp.fillInPlaceHolders(input, createPostProcessingContext(rec, null, config)));
assertEquals("asdf_tag,_foo,_bar_asdf", placeHolderAwarePp.fillInPlaceHolders(input, createPostProcessingContext(rec, null, config, modelNotesService)));
}
@Test

View File

@ -93,11 +93,16 @@ public abstract class AbstractPpTest {
}
PostProcessingContext createPostProcessingContext(Recording rec, RecordingManager recordingManager, Config config) {
return createPostProcessingContext(rec, recordingManager, config, null);
}
PostProcessingContext createPostProcessingContext(Recording rec, RecordingManager recordingManager, Config config, ModelNotesService modelNotesService) {
PostProcessingContext ctx = new PostProcessingContext();
ctx.setConfig(config);
ctx.setRecording(rec);
ctx.setRecorder(mockRecorder());
ctx.setRecordingManager(recordingManager);
ctx.setModelNotesService(modelNotesService);
return ctx;
}
}