Merge pull request #866 from cyian-1756/testFixes

Test fixes
This commit is contained in:
cyian-1756 2018-08-20 16:34:46 -04:00 committed by GitHub
commit 364fa7de52
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 32 additions and 2 deletions

View File

@ -47,7 +47,7 @@ public abstract class AbstractRipper
// Everytime addUrlToDownload skips a already downloaded url this increases by 1
public int alreadyDownloadedUrls = 0;
private boolean shouldStop = false;
private boolean thisIsATest = false;
private static boolean thisIsATest = false;
public void stop() {
shouldStop = true;
@ -611,7 +611,7 @@ public abstract class AbstractRipper
LOGGER.debug("THIS IS A TEST RIP");
thisIsATest = true;
}
protected boolean isThisATest() {
protected static boolean isThisATest() {
return thisIsATest;
}

View File

@ -23,6 +23,7 @@ import org.jsoup.HttpStatusException;
import com.rarchives.ripme.ui.RipStatusMessage.STATUS;
import com.rarchives.ripme.utils.Utils;
import com.rarchives.ripme.ripper.AbstractRipper;
/**
* Thread for downloading files.
@ -222,6 +223,14 @@ class DownloadFileThread extends Thread {
observer.setBytesCompleted(bytesDownloaded);
observer.sendUpdate(STATUS.COMPLETED_BYTES, bytesDownloaded);
}
// If this is a test and we're downloading a large file
if (AbstractRipper.isThisATest() && bytesTotal / 10000000 >= 10) {
logger.debug("Not downloading whole file because it is over 10mb and this is a test");
bis.close();
fos.close();
break;
}
}
bis.close();
fos.close();

View File

@ -0,0 +1,21 @@
package com.rarchives.ripme.tst.ripper.rippers;
import com.rarchives.ripme.ripper.rippers.HqpornerRipper;
import java.io.IOException;
import java.net.URL;
public class HqpornerRipperTest extends RippersTest{
public void testRip() throws IOException {
HqpornerRipper ripper = new HqpornerRipper(new URL("https://hqporner.com/hdporn/84636-pool_lesson_with_a_cheating_husband.html"));
testRipper(ripper);
}
public void testGetGID() throws IOException {
URL poolURL = new URL("https://hqporner.com/hdporn/84636-pool_lesson_with_a_cheating_husband.html");
HqpornerRipper ripper = new HqpornerRipper(poolURL);
assertEquals("84636-pool_lesson_with_a_cheating_husband", ripper.getGID(poolURL));
}
}