Merge pull request #208 from dispatch/0.14/realm-with-scheme

[0.14] Allow to set scheme in as keyword
dispatch · Nov 18, 2018 · e871029 · e871029
2 parents 0243e8b + ba00ab7
commit e871029
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/core/src/main/scala/requests.scala b/core/src/main/scala/requests.scala
@@ -271,8 +271,8 @@ trait ParamVerbs extends RequestVerbs {
 }
 
 trait AuthVerbs extends RequestVerbs {
-  def as(user: String, password: String): Req =
-    this.as(new Realm.Builder(user, password).build())
+  def as(user: String, password: String, scheme: AuthScheme): Req =
+    this.as(new Realm.Builder(user, password).setScheme(scheme).build())
 
   /** Basic auth, use with care. */
   def as_!(user: String, password: String): Req =

diff --git a/core/src/test/scala/basic.scala b/core/src/test/scala/basic.scala
@@ -2,7 +2,10 @@ package dispatch.spec
 
 import java.nio.charset.Charset
 
+import org.asynchttpclient.Realm
+import org.asynchttpclient.Realm.AuthScheme
 import org.scalacheck._
+import org.scalacheck.Prop._
 
 object BasicSpecification
 extends Properties("Basic")
@@ -165,4 +168,19 @@ with DispatchCleanup {
       req.toRequest.getUrl ?= "http://127.0.0.1:%d/?%s".format(server.port, expectedParams.mkString("&"))
     }
   }
+
+  property("Building a Realm without a scheme should throw NPE") = {
+    forAll(Gen.zip(Gen.alphaStr, Gen.alphaStr)) { case (user, password) =>
+      throws(classOf[NullPointerException])(new Realm.Builder(user, password).build())
+    }
+  }
+
+  property("Build a Realm using as") = {
+    forAll(Gen.zip(Gen.alphaStr, Gen.alphaStr, Gen.oneOf(AuthScheme.values()))) { case (user, password, scheme) =>
+      val realm = localhost.as(user, password, scheme).toRequest.getRealm
+      realm.getScheme ?= scheme
+      realm.getPrincipal ?= user
+      realm.getPassword ?= password
+    }
+  }
 }