Merge pull request Azure#315 from RikkiGibson/WithRequestPolicies

Add withRequestPolicies
sima-zhu · Dec 5, 2017 · f9f781c · f9f781c
2 parents 7fc95c1 + eb28d89
commit f9f781c
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/client-runtime/src/main/java/com/microsoft/rest/v2/http/HttpPipeline.java b/client-runtime/src/main/java/com/microsoft/rest/v2/http/HttpPipeline.java
@@ -243,6 +243,19 @@ public Builder withRequestPolicy(int index, RequestPolicy.Factory requestPolicyF
             return this;
         }
 
+        /**
+         * Add the provided RequestPolicy factories to this HttpPipeline builder.
+         * @param requestPolicyFactories The RequestPolicy factories to add to this
+         *                               HttpPipeline builder.
+         * @return This HttpPipeline builder.
+         */
+        public Builder withRequestPolicies(RequestPolicy.Factory... requestPolicyFactories) {
+            for (RequestPolicy.Factory factory : requestPolicyFactories) {
+                withRequestPolicy(factory);
+            }
+            return this;
+        }
+
         /**
          * Add the provided RequestPolicy factory to this HttpPipeline builder
          * directly before the first instance of the provided RequestPolicy

diff --git a/client-runtime/src/test/java/com/microsoft/rest/v2/http/HttpPipelineBuilderTests.java b/client-runtime/src/test/java/com/microsoft/rest/v2/http/HttpPipelineBuilderTests.java
@@ -110,4 +110,37 @@ public void withRequestPolicyAfter() {
         assertEquals(LoggingPolicy.Factory.class, builder.requestPolicyFactories().get(2).getClass());
         assertEquals(PortPolicy.Factory.class, builder.requestPolicyFactories().get(3).getClass());
     }
+
+    @Test
+    public void withRequestPolicyArray() {
+        final HttpPipeline.Builder builder = new HttpPipeline.Builder();
+
+        builder.withRequestPolicies(
+                new ProtocolPolicy.Factory("http"),
+                new PortPolicy.Factory(80),
+                new LoggingPolicy.Factory(LoggingPolicy.LogLevel.BODY));
+
+        assertEquals(3, builder.requestPolicyFactories().size());
+        assertEquals(LoggingPolicy.Factory.class, builder.requestPolicyFactories().get(0).getClass());
+        assertEquals(PortPolicy.Factory.class, builder.requestPolicyFactories().get(1).getClass());
+        assertEquals(ProtocolPolicy.Factory.class, builder.requestPolicyFactories().get(2).getClass());
+    }
+
+    @Test
+    public void appendingRequestPolicyArray() {
+        final HttpPipeline.Builder builder = new HttpPipeline.Builder();
+
+        builder.withRequestPolicy(new RetryPolicy.Factory());
+        builder.withRequestPolicies(
+                new ProtocolPolicy.Factory("http"),
+                new PortPolicy.Factory(80),
+                new LoggingPolicy.Factory(LoggingPolicy.LogLevel.BODY));
+
+        assertEquals(4, builder.requestPolicyFactories().size());
+        assertEquals(LoggingPolicy.Factory.class, builder.requestPolicyFactories().get(0).getClass());
+        assertEquals(PortPolicy.Factory.class, builder.requestPolicyFactories().get(1).getClass());
+        assertEquals(ProtocolPolicy.Factory.class, builder.requestPolicyFactories().get(2).getClass());
+        assertEquals(RetryPolicy.Factory.class, builder.requestPolicyFactories().get(3).getClass());
+
+    }
 }