mirror of
https://github.com/RatzzFatzz/MKVAudioSubtitleChanger.git
synced 2026-02-11 10:05:58 +01:00
Improve subtitle selection
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
package at.pcgamingfreaks.mkvaudiosubtitlechanger.impl;
|
||||
|
||||
import at.pcgamingfreaks.mkvaudiosubtitlechanger.model.FileAttribute;
|
||||
import at.pcgamingfreaks.mkvaudiosubtitlechanger.model.LaneType;
|
||||
import org.junit.jupiter.params.ParameterizedTest;
|
||||
import org.junit.jupiter.params.provider.Arguments;
|
||||
import org.junit.jupiter.params.provider.MethodSource;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.stream.Collectors;
|
||||
import java.util.stream.Stream;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.*;
|
||||
|
||||
class SubtitleTrackComparatorTest {
|
||||
private static final SubtitleTrackComparator comparator = new SubtitleTrackComparator(new String[]{"unstyled"});
|
||||
|
||||
private static Stream<Arguments> compareArguments() {
|
||||
return Stream.of(
|
||||
Arguments.of(List.of(attr("unstyled sub", false), attr("styled sub", false)),
|
||||
List.of(attr("unstyled sub", false), attr("styled sub", false))),
|
||||
Arguments.of(List.of(attr("styled sub", false), attr("unstyled sub", false)),
|
||||
List.of(attr("unstyled sub", false), attr("styled sub", false))),
|
||||
|
||||
Arguments.of(List.of(attr("unstyled sub", true), attr("styled sub", false)),
|
||||
List.of(attr("unstyled sub", true), attr("styled sub", false))),
|
||||
Arguments.of(List.of(attr("styled sub", true), attr("unstyled sub", false)),
|
||||
List.of(attr("unstyled sub", false), attr("styled sub", true))),
|
||||
|
||||
Arguments.of(List.of(attr("unstyled sub", true), attr("unstyled sub", false)),
|
||||
List.of(attr("unstyled sub", true), attr("unstyled sub", false)))
|
||||
);
|
||||
}
|
||||
|
||||
@ParameterizedTest
|
||||
@MethodSource("compareArguments")
|
||||
void compare(List<FileAttribute> input, List<FileAttribute> expected) {
|
||||
List<FileAttribute> result = input.stream().sorted(comparator.reversed()).collect(Collectors.toList());
|
||||
|
||||
assertArrayEquals(expected.toArray(new FileAttribute[0]), result.toArray(new FileAttribute[0]));
|
||||
}
|
||||
|
||||
private static FileAttribute attr(String trackName, boolean defaultTrack) {
|
||||
return new FileAttribute(0, "", trackName, defaultTrack, false, LaneType.SUBTITLES);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user