Created
June 4, 2016 07:26
-
-
Save slawekradzyminski/7d503a49db1b38f6988aa71310d99b01 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import org.apache.commons.io.FileUtils; | |
import org.apache.http.HttpResponse; | |
import org.apache.http.client.HttpClient; | |
import org.apache.http.client.methods.HttpGet; | |
import org.apache.http.client.methods.HttpRequestBase; | |
import org.apache.http.client.protocol.HttpClientContext; | |
import org.apache.http.impl.client.BasicCookieStore; | |
import org.apache.http.impl.client.HttpClientBuilder; | |
import org.apache.http.impl.client.LaxRedirectStrategy; | |
import org.apache.http.impl.cookie.BasicClientCookie; | |
import org.apache.http.protocol.BasicHttpContext; | |
import org.openqa.selenium.Cookie; | |
import org.openqa.selenium.WebDriver; | |
import java.io.File; | |
import java.io.IOException; | |
import java.net.URI; | |
import java.net.URISyntaxException; | |
import java.util.Set; | |
class Request { | |
private URI linkToCheck; | |
private WebDriver driver; | |
Request(WebDriver driver) { | |
this.driver = driver; | |
} | |
Request() { | |
} | |
void setURIToCheck(String linkToCheck) throws URISyntaxException { | |
this.linkToCheck = new URI(linkToCheck); | |
} | |
int getHTTPStatusCodeFromResponse() throws IOException { | |
return getHttpResponse().getStatusLine().getStatusCode(); | |
} | |
private HttpResponse getHttpResponse() throws IOException { | |
HttpClient client = initializeHttpClient(); | |
BasicHttpContext httpContext = new BasicHttpContext(); | |
if (driver != null) { | |
addCookies(httpContext); | |
} | |
HttpRequestBase request = buildRequest(); | |
return client.execute(request, httpContext); | |
} | |
private HttpClient initializeHttpClient() { | |
return HttpClientBuilder | |
.create() | |
.setRedirectStrategy(new LaxRedirectStrategy()) | |
.build(); | |
} | |
private HttpRequestBase buildRequest() { | |
HttpRequestBase requestMethod = new HttpGet(); | |
requestMethod.setURI(this.linkToCheck); | |
return requestMethod; | |
} | |
private void addCookies(BasicHttpContext httpContext) { | |
BasicCookieStore cookies = getCurrentDriverCookies(driver.manage().getCookies()); | |
httpContext.setAttribute(HttpClientContext.COOKIE_STORE, cookies); | |
} | |
private BasicCookieStore getCurrentDriverCookies(Set<Cookie> cookies) { | |
BasicCookieStore mimicWebDriverCookieStore = new BasicCookieStore(); | |
for (Cookie seleniumCookie : cookies) { | |
BasicClientCookie duplicateCookie = new BasicClientCookie(seleniumCookie.getName(), seleniumCookie.getValue()); | |
duplicateCookie.setDomain(seleniumCookie.getDomain()); | |
duplicateCookie.setSecure(seleniumCookie.isSecure()); | |
duplicateCookie.setExpiryDate(seleniumCookie.getExpiry()); | |
duplicateCookie.setPath(seleniumCookie.getPath()); | |
mimicWebDriverCookieStore.addCookie(duplicateCookie); | |
} | |
return mimicWebDriverCookieStore; | |
} | |
File downloadFile() throws Exception { | |
File downloadedFile = File.createTempFile("download", ".tmp"); | |
HttpResponse fileToDownload = getHttpResponse(); | |
FileUtils.copyInputStreamToFile(fileToDownload.getEntity().getContent(), downloadedFile); | |
return downloadedFile; | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment