Fix proxy test

Azure · Oct 11, 2017 · 981072a · 981072a
1 parent f2fc333
commit 981072a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/microsoft-azure-storage-test/src/com/microsoft/azure/storage/GenericTests.java b/microsoft-azure-storage-test/src/com/microsoft/azure/storage/GenericTests.java
@@ -42,6 +42,7 @@
 import java.net.HttpURLConnection;
 import java.net.InetSocketAddress;
 import java.net.Proxy;
+import java.net.SocketException;
 import java.net.SocketTimeoutException;
 import java.net.URISyntaxException;
 import java.text.ParseException;
@@ -141,7 +142,8 @@ public void testProxy() throws URISyntaxException, StorageException {
             fail("Bad proxy should throw an exception.");
         } catch (StorageException e) {
             if (e.getCause().getClass() != ConnectException.class &&
-                    e.getCause().getClass() != SocketTimeoutException.class) {
+                    e.getCause().getClass() != SocketTimeoutException.class &&
+                    e.getCause().getClass() != SocketException.class) {
                 Assert.fail("Unepected exception for bad proxy");
             }
         }
@@ -167,7 +169,8 @@ public void testDefaultProxy() throws URISyntaxException, StorageException {
             fail("Bad proxy should throw an exception.");
         } catch (StorageException e) {
             if (e.getCause().getClass() != ConnectException.class &&
-                    e.getCause().getClass() != SocketTimeoutException.class) {
+                    e.getCause().getClass() != SocketTimeoutException.class &&
+                    e.getCause().getClass() != SocketException.class) {
                 Assert.fail("Unepected exception for bad proxy");
             }
         }