add setter for the headers Key Id value

lbalmaceda · lbalmaceda · commit 970fb3cdc5bd · 2017-02-24T11:34:39.000-03:00
diff --git a/lib/src/main/java/com/auth0/jwt/JWTCreator.java b/lib/src/main/java/com/auth0/jwt/JWTCreator.java
@@ -73,6 +73,17 @@ public Builder withHeader(Map<String, Object> headerClaims) {
             return this;
         }
 
+        /**
+         * Add a specific Key Id ("kid") claim to the Header.
+         *
+         * @param keyId the Key Id value.
+         * @return this same Builder instance.
+         */
+        public Builder withKeyId(String keyId) {
+            this.headerClaims.put(PublicClaims.KEY_ID, keyId);
+            return this;
+        }
+
         /**
          * Add a specific Issuer ("iss") claim to the Payload.
          *
diff --git a/lib/src/test/java/com/auth0/jwt/JWTCreatorTest.java b/lib/src/test/java/com/auth0/jwt/JWTCreatorTest.java
@@ -39,6 +39,16 @@ public void shouldAddHeader() throws Exception {
         assertThat(TokenUtils.splitToken(signed)[0], is("eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCIsImFzZCI6MTIzfQ"));
     }
 
+    @Test
+    public void shouldAddKeyId() throws Exception {
+        String signed = JWTCreator.init()
+                .withKeyId("56a8bd44da435300010000015f5ed")
+                .sign(Algorithm.HMAC256("secret"));
+
+        assertThat(signed, is(notNullValue()));
+        assertThat(TokenUtils.splitToken(signed)[0], is("eyJraWQiOiI1NmE4YmQ0NGRhNDM1MzAwMDEwMDAwMDE1ZjVlZCIsInR5cCI6IkpXVCIsImFsZyI6IkhTMjU2In0"));
+    }
+
     @Test
     public void shouldAddIssuer() throws Exception {
         String signed = JWTCreator.init()
