Add a test about the server replying 401 while the client uploads a big file

Author: Stephane Landelle

src/test/java/com/ning/http/client/async/FastUnauthorizedUploadTest.java

@@ -0,0 +1,67 @@
+/*
+ * Copyright (c) 2010-2012 Sonatype, Inc. All rights reserved.
+ *
+ * This program is licensed to you under the Apache License Version 2.0,
+ * and you may not use this file except in compliance with the Apache License Version 2.0.
+ * You may obtain a copy of the Apache License Version 2.0 at http://www.apache.org/licenses/LICENSE-2.0.
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the Apache License Version 2.0 is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the Apache License Version 2.0 for the specific language governing permissions and limitations there under.
+ */
+package com.ning.http.client.async;
+
+import java.io.File;
+import java.io.IOException;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.eclipse.jetty.server.Request;
+import org.eclipse.jetty.server.handler.AbstractHandler;
+import org.testng.Assert;
+import org.testng.annotations.Test;
+
+import com.ning.http.client.AsyncHttpClient;
+import com.ning.http.client.AsyncHttpClient.BoundRequestBuilder;
+import com.ning.http.client.FilePart;
+import com.ning.http.client.Response;
+
+public abstract class FastUnauthorizedUploadTest extends AbstractBasicTest {
+
+    @Override
+    public AbstractHandler configureHandler() throws Exception {
+        return new AbstractHandler() {
+
+            public void handle(String target, Request baseRequest, HttpServletRequest req, HttpServletResponse resp) throws IOException, ServletException {
+
+                resp.setStatus(401);
+                resp.getOutputStream().flush();
+                resp.getOutputStream().close();
+
+                baseRequest.setHandled(true);
+            }
+        };
+    }
+
+    @Test(groups = { "standalone", "default_provider" }, enabled = true)
+    public void testUnauthorizedWhileUploading() throws Exception {
+        byte[] bytes = "RatherLargeFileRatherLargeFileRatherLargeFileRatherLargeFile".getBytes("UTF-16");
+        long repeats = (1024 * 1024 / bytes.length) + 1;
+        File largeFile = FilePartLargeFileTest.createTempFile(bytes, (int) repeats);
+
+        AsyncHttpClient client = getAsyncHttpClient(null);
+        try {
+            BoundRequestBuilder rb = client.preparePut(getTargetUrl());
+
+            rb.addBodyPart(new FilePart("test", largeFile, "application/octet-stream", "UTF-8"));
+
+            Response response = rb.execute().get();
+            Assert.assertEquals(401, response.getStatusCode());
+        } finally {
+            client.close();
+        }
+    }
+}

src/test/java/com/ning/http/client/async/FilePartLargeFileTest.java

@@ -28,7 +28,6 @@
 import org.eclipse.jetty.server.Request;
 import org.eclipse.jetty.server.handler.AbstractHandler;
 import org.testng.Assert;
-import org.testng.annotations.AfterMethod;
 import org.testng.annotations.Test;
 
 import com.ning.http.client.AsyncHttpClient;
@@ -39,11 +38,9 @@
 
 public abstract class FilePartLargeFileTest extends AbstractBasicTest {
 
-    private File largeFile;
-
     @Test(groups = { "standalone", "default_provider" }, enabled = true)
     public void testPutImageFile() throws Exception {
-        largeFile = getTestFile();
+        File largeFile = getTestFile();
         AsyncHttpClientConfig config = new AsyncHttpClientConfig.Builder().setRequestTimeoutInMs(100 * 6000).build();
         AsyncHttpClient client = getAsyncHttpClient(config);
         try {
@@ -62,7 +59,7 @@ public void testPutImageFile() throws Exception {
     public void testPutLargeTextFile() throws Exception {
         byte[] bytes = "RatherLargeFileRatherLargeFileRatherLargeFileRatherLargeFile".getBytes("UTF-16");
         long repeats = (1024 * 1024 / bytes.length) + 1;
-        largeFile = createTempFile(bytes, (int) repeats);
+        File largeFile = createTempFile(bytes, (int) repeats);
 
         AsyncHttpClient client = getAsyncHttpClient(null);
         try {
@@ -93,11 +90,6 @@ private static File getTestFile() {
         return testResource1File;
     }
 
-    @AfterMethod
-    public void after() {
-        largeFile.delete();
-    }
-
     @Override
     public AbstractHandler configureHandler() throws Exception {
         return new AbstractHandler() {

src/test/java/com/ning/http/client/async/netty/NettyFastUnauthorizedUploadTest.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright (c) 2010-2012 Sonatype, Inc. All rights reserved.
+ *
+ * This program is licensed to you under the Apache License Version 2.0,
+ * and you may not use this file except in compliance with the Apache License Version 2.0.
+ * You may obtain a copy of the Apache License Version 2.0 at http://www.apache.org/licenses/LICENSE-2.0.
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the Apache License Version 2.0 is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the Apache License Version 2.0 for the specific language governing permissions and limitations there under.
+ */
+package com.ning.http.client.async.netty;
+
+import org.testng.annotations.Test;
+
+import com.ning.http.client.AsyncHttpClient;
+import com.ning.http.client.AsyncHttpClientConfig;
+import com.ning.http.client.async.FastUnauthorizedUploadTest;
+import com.ning.http.client.async.ProviderUtil;
+
+@Test
+public class NettyFastUnauthorizedUploadTest extends FastUnauthorizedUploadTest {
+    @Override
+    public AsyncHttpClient getAsyncHttpClient(AsyncHttpClientConfig config) {
+        return ProviderUtil.nettyProvider(config);
+    }
+}