Report upload progress when sending a byte array

Author: kevinsawicki

lib/src/main/java/com/github/kevinsawicki/http/HttpRequest.java

@@ -2925,6 +2925,8 @@ public HttpRequest send(final File input) throws HttpRequestException {
  * @throws HttpRequestException
  */
  public HttpRequest send(final byte[] input) throws HttpRequestException {
+ if (input != null)
+ totalSize += input.length;
  return send(new ByteArrayInputStream(input));
  }
 

lib/src/test/java/com/github/kevinsawicki/http/HttpRequestTest.java

@@ -3520,6 +3520,35 @@ public void onUpload(int transferred, int total) {
  assertEquals(file.length(), tx.get());
  }
 
+ /**
+ * Verify progress callback when sending from a byte array
+ *
+ * @throws Exception
+ */
+ @Test
+ public void uploadProgressSendByteArray() throws Exception {
+ final AtomicReference<String> body = new AtomicReference<String>();
+ handler = new RequestHandler() {
+
+ @Override
+ public void handle(Request request, HttpServletResponse response) {
+ body.set(new String(read()));
+ response.setStatus(HTTP_OK);
+ }
+ };
+
+ final byte[] bytes = "hello".getBytes(CHARSET_UTF8);
+ final AtomicInteger tx = new AtomicInteger(0);
+ UploadProgress progress = new UploadProgress() {
+ public void onUpload(int transferred, int total) {
+ assertEquals(bytes.length, total);
+ assertEquals(tx.incrementAndGet(), transferred);
+ }
+ };
+ post(url).bufferSize(1).progress(progress).send(bytes).code();
+ assertEquals(bytes.length, tx.get());
+ }
+
  /**
  * Verify progress callback when sending from a Reader
  *