Fix tests again.

This commit is contained in:
FireMasterK 2021-07-31 14:40:51 +05:30
parent 88c11db4c5
commit f4aad8b014
No known key found for this signature in database
GPG key ID: 49451E4482CC5BCD
2 changed files with 6 additions and 2 deletions

View file

@ -147,8 +147,10 @@ public class YoutubeMusicSearchExtractor extends SearchExtractor {
return false;
}
final boolean corrected = itemSectionRenderer.getArray("contents").getObject(0)
.has("didYouMeanRenderer");
JsonObject firstContent = itemSectionRenderer.getArray("contents").getObject(0);
final boolean corrected = firstContent
.has("didYouMeanRenderer") || firstContent.has("showingResultsForRenderer");
return corrected;
}

View file

@ -133,6 +133,7 @@ public class YoutubeMusicSearchExtractorTest {
public static class Suggestion extends DefaultSearchExtractorTest {
private static SearchExtractor extractor;
private static final String QUERY = "megaman x3";
private static final boolean CORRECTED = true;
@BeforeClass
public static void setUp() throws Exception {
@ -150,6 +151,7 @@ public class YoutubeMusicSearchExtractorTest {
@Override public String expectedSearchString() { return QUERY; }
@Nullable @Override public String expectedSearchSuggestion() { return "mega man x3"; }
@Override public InfoItem.InfoType expectedInfoItemType() { return InfoItem.InfoType.STREAM; }
@Override public boolean isCorrectedSearch() { return CORRECTED; }
}
public static class CorrectedSearch extends DefaultSearchExtractorTest {