use of com.axway.ats.core.filetransfer.HttpsClient in project ats-framework by Axway.
the class HtmlFileDownloader method downloadFile.
/**
* Perform the file/image download.
*
* @param downloadUrl
* @return download file absolute path
* @throws IOException
* @throws FileTransferClientException
*/
public String downloadFile(String downloadUrl) throws IOException {
if (StringUtils.isNullOrEmpty(downloadUrl)) {
throw new UiElementException("The element you have specified does not link to anything!");
}
URL url = new URL(downloadUrl);
String fileName = url.getPath();
if (fileName.indexOf('/') > -1) {
fileName = fileName.substring(fileName.lastIndexOf('/') + 1);
}
HttpsClient httpsClient = new HttpsClient();
httpsClient.connect(url.getHost());
if (this.mimicWebDriverCookieState) {
for (org.openqa.selenium.Cookie cookie : this.webDriver.manage().getCookies()) {
httpsClient.addCookie(cookie.getName(), cookie.getValue(), cookie.getDomain(), cookie.getPath(), cookie.getExpiry(), cookie.isSecure());
}
if (this.webDriver instanceof org.openqa.selenium.phantomjs.PhantomJSDriver && System.getProperty(PhantomJsDriver.HTTP_ONLY_COOKIES_PROPERTY) != null) {
for (org.openqa.selenium.Cookie cookie : PhantomJsDriver.getHttpOnlyCookies()) {
httpsClient.addCookie(cookie.getName(), cookie.getValue(), cookie.getDomain(), cookie.getPath(), cookie.getExpiry(), cookie.isSecure());
}
}
}
File downloadedFile = new File(this.downloadDir + fileName);
log.debug("Downloading file: " + downloadedFile.getName());
org.apache.commons.io.FileUtils.copyInputStreamToFile(httpsClient.performGetRequest(url.getFile()), downloadedFile);
String downloadedFileAbsolutePath = downloadedFile.getAbsolutePath();
log.debug("File downloaded to '" + downloadedFileAbsolutePath + "'");
return downloadedFileAbsolutePath;
}
Aggregations