Reputation: 445
I have a method for downloading a file from a URL. I need to write unit test cases to resume a download, so is there any way to disable/enable the network connection so that resume download method can invoke?
My download method is as follows:
public String download(LatestVersionInfo info) throws FileNotFoundException, XMLStreamException,NoSuchAlgorithmException, MalformedURLException {
boolean isCached = checkCache(info, location);
String filePath = null;
RandomAccessFile randomAccessFile = null;
InputStream inputStream = null;
if (!isCached) {
try {
createLocation();
int responseCode = 0;
HttpsURLConnection connection = getSSLCertificate(info
.getLatestVersionUrl().toString());
connection.setRequestMethod("GET");
connection.setRequestProperty("Range", "bytes=" + downloaded
+ "-");
connection.connect();
responseCode = connection.getResponseCode();
if (responseCode / 100 != 2) {
error();
}
int contentLength = connection.getContentLength();
if (contentLength < 1) {
error();
}
if (size == -1) {
size = contentLength;
stateChanged();
}
randomAccessFile = new RandomAccessFile(
getDownLoadPath(info.getLatestVersionUrl()), "rw");
randomAccessFile.seek(downloaded);
inputStream = connection.getInputStream();
while (status == DOWNLOADING) {
byte buffer[];
float finalSize = size - downloaded;
if (finalSize > MAX_BUFFER_SIZE) {
buffer = new byte[(int) finalSize];
} else {
buffer = new byte[MAX_BUFFER_SIZE];
}
int read = inputStream.read(buffer);
if (read == -1)
break;
randomAccessFile.write(buffer, 0, read);
downloaded += read;
stateChanged();
}
if (status == DOWNLOADING) {
status = COMPLETE;
stateChanged();
}
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
if (randomAccessFile != null)
randomAccessFile.close();
} catch (IOException e) {
e.printStackTrace();
}
try {
if (inputStream != null)
inputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
return filePath;
}
Is there any way to disable/enable the network connection so that resume download method can invoke.
Thanks.
Upvotes: 1
Views: 258
Reputation: 32437
Looks like you're trying to resume after a failed connection rather than pause the underlying TCP transfer in some way (which would be much harder).
So you can just kill the HttpsURLConnection
halfway through (from a separate thread). If you put the HttpsURLConnection connection
in a field, the other thread can call connection.getInputStream().close()
, which will cause the connection to 'fail'.
Upvotes: 1