|
12 | 12 | */
|
13 | 13 | package org.asynchttpclient;
|
14 | 14 |
|
15 |
| -import org.asynchttpclient.Realm; |
16 |
| -import org.asynchttpclient.Realm.AuthScheme; |
17 |
| -import org.asynchttpclient.Realm.RealmBuilder; |
18 |
| -import org.testng.Assert; |
| 15 | +import static org.testng.Assert.assertEquals; |
| 16 | + |
19 | 17 | import java.math.BigInteger;
|
20 | 18 | import java.security.MessageDigest;
|
21 | 19 |
|
| 20 | +import org.asynchttpclient.Realm.AuthScheme; |
| 21 | +import org.asynchttpclient.Realm.RealmBuilder; |
22 | 22 | import org.testng.annotations.Test;
|
23 | 23 |
|
24 | 24 | public class RealmTest {
|
25 | 25 | @Test(groups = "fast")
|
26 | 26 | public void testClone() {
|
27 | 27 | RealmBuilder builder = new RealmBuilder();
|
28 |
| - builder.setPrincipal( "user" ).setPassword( "pass" ); |
29 |
| - builder.setEnconding( "enc" ).setUsePreemptiveAuth( true ); |
30 |
| - builder.setRealmName( "realm" ).setAlgorithm( "algo" ); |
31 |
| - builder.setScheme( AuthScheme.BASIC ); |
| 28 | + builder.setPrincipal("user").setPassword("pass"); |
| 29 | + builder.setEnconding("enc").setUsePreemptiveAuth(true); |
| 30 | + builder.setRealmName("realm").setAlgorithm("algo"); |
| 31 | + builder.setScheme(AuthScheme.BASIC); |
32 | 32 | Realm orig = builder.build();
|
33 |
| - |
34 |
| - Realm clone = new RealmBuilder().clone( orig ).build(); |
35 |
| - Assert.assertEquals( clone.getPrincipal(), orig.getPrincipal() ); |
36 |
| - Assert.assertEquals( clone.getPassword(), orig.getPassword() ); |
37 |
| - Assert.assertEquals( clone.getEncoding(), orig.getEncoding() ); |
38 |
| - Assert.assertEquals( clone.getUsePreemptiveAuth(), orig.getUsePreemptiveAuth() ); |
39 |
| - Assert.assertEquals( clone.getRealmName(), orig.getRealmName() ); |
40 |
| - Assert.assertEquals( clone.getAlgorithm(), orig.getAlgorithm() ); |
41 |
| - Assert.assertEquals( clone.getAuthScheme(), orig.getAuthScheme() ); |
| 33 | + |
| 34 | + Realm clone = new RealmBuilder().clone(orig).build(); |
| 35 | + assertEquals(clone.getPrincipal(), orig.getPrincipal()); |
| 36 | + assertEquals(clone.getPassword(), orig.getPassword()); |
| 37 | + assertEquals(clone.getEncoding(), orig.getEncoding()); |
| 38 | + assertEquals(clone.getUsePreemptiveAuth(), orig.getUsePreemptiveAuth()); |
| 39 | + assertEquals(clone.getRealmName(), orig.getRealmName()); |
| 40 | + assertEquals(clone.getAlgorithm(), orig.getAlgorithm()); |
| 41 | + assertEquals(clone.getAuthScheme(), orig.getAuthScheme()); |
42 | 42 | }
|
| 43 | + |
43 | 44 | @Test(groups = "fast")
|
44 | 45 | public void testOldDigestEmptyString() {
|
45 |
| - String qop=""; |
| 46 | + String qop = ""; |
46 | 47 | testOldDigest(qop);
|
47 | 48 | }
|
| 49 | + |
48 | 50 | @Test(groups = "fast")
|
49 | 51 | public void testOldDigestNull() {
|
50 |
| - String qop=null; |
| 52 | + String qop = null; |
51 | 53 | testOldDigest(qop);
|
52 | 54 | }
|
53 | 55 |
|
54 |
| - private void testOldDigest(String qop){ |
55 |
| - String user="user"; |
56 |
| - String pass="pass"; |
57 |
| - String realm="realm"; |
58 |
| - String nonce="nonce"; |
59 |
| - String method="GET"; |
60 |
| - String uri="/foo"; |
| 56 | + private void testOldDigest(String qop) { |
| 57 | + String user = "user"; |
| 58 | + String pass = "pass"; |
| 59 | + String realm = "realm"; |
| 60 | + String nonce = "nonce"; |
| 61 | + String method = "GET"; |
| 62 | + String uri = "/foo"; |
61 | 63 | RealmBuilder builder = new RealmBuilder();
|
62 |
| - builder.setPrincipal( user ).setPassword( pass ); |
63 |
| - builder.setNonce( nonce ); |
64 |
| - builder.setUri( uri ); |
| 64 | + builder.setPrincipal(user).setPassword(pass); |
| 65 | + builder.setNonce(nonce); |
| 66 | + builder.setUri(uri); |
65 | 67 | builder.setMethodName(method);
|
66 |
| - builder.setRealmName( realm ); |
| 68 | + builder.setRealmName(realm); |
67 | 69 | builder.setQop(qop);
|
68 |
| - builder.setScheme( AuthScheme.DIGEST ); |
| 70 | + builder.setScheme(AuthScheme.DIGEST); |
69 | 71 | Realm orig = builder.build();
|
70 | 72 |
|
71 |
| - String ha1=getMd5(user +":" + realm +":"+pass); |
72 |
| - String ha2=getMd5(method +":"+ uri); |
73 |
| - String expectedResponse=getMd5(ha1 +":" + nonce +":" + ha2); |
| 73 | + String ha1 = getMd5(user + ":" + realm + ":" + pass); |
| 74 | + String ha2 = getMd5(method + ":" + uri); |
| 75 | + String expectedResponse = getMd5(ha1 + ":" + nonce + ":" + ha2); |
74 | 76 |
|
75 |
| - Assert.assertEquals(expectedResponse,orig.getResponse()); |
| 77 | + assertEquals(expectedResponse, orig.getResponse()); |
76 | 78 | }
|
77 | 79 |
|
78 | 80 | @Test(groups = "fast")
|
79 | 81 | public void testStrongDigest() {
|
80 |
| - String user="user"; |
81 |
| - String pass="pass"; |
82 |
| - String realm="realm"; |
83 |
| - String nonce="nonce"; |
84 |
| - String method="GET"; |
85 |
| - String uri="/foo"; |
86 |
| - String qop="auth"; |
| 82 | + String user = "user"; |
| 83 | + String pass = "pass"; |
| 84 | + String realm = "realm"; |
| 85 | + String nonce = "nonce"; |
| 86 | + String method = "GET"; |
| 87 | + String uri = "/foo"; |
| 88 | + String qop = "auth"; |
87 | 89 | RealmBuilder builder = new RealmBuilder();
|
88 |
| - builder.setPrincipal( user ).setPassword( pass ); |
89 |
| - builder.setNonce( nonce ); |
90 |
| - builder.setUri( uri ); |
| 90 | + builder.setPrincipal(user).setPassword(pass); |
| 91 | + builder.setNonce(nonce); |
| 92 | + builder.setUri(uri); |
91 | 93 | builder.setMethodName(method);
|
92 |
| - builder.setRealmName( realm ); |
| 94 | + builder.setRealmName(realm); |
93 | 95 | builder.setQop(qop);
|
94 |
| - builder.setScheme( AuthScheme.DIGEST ); |
| 96 | + builder.setScheme(AuthScheme.DIGEST); |
95 | 97 | Realm orig = builder.build();
|
96 | 98 |
|
97 | 99 | String nc = orig.getNc();
|
98 | 100 | String cnonce = orig.getCnonce();
|
99 |
| - String ha1=getMd5(user +":" + realm +":"+pass); |
100 |
| - String ha2=getMd5(method +":"+ uri); |
101 |
| - String expectedResponse=getMd5(ha1 +":" + nonce +":" + nc + ":" + cnonce +":" + qop + ":" + ha2); |
| 101 | + String ha1 = getMd5(user + ":" + realm + ":" + pass); |
| 102 | + String ha2 = getMd5(method + ":" + uri); |
| 103 | + String expectedResponse = getMd5(ha1 + ":" + nonce + ":" + nc + ":" + cnonce + ":" + qop + ":" + ha2); |
102 | 104 |
|
103 |
| - Assert.assertEquals(expectedResponse,orig.getResponse()); |
| 105 | + assertEquals(expectedResponse, orig.getResponse()); |
104 | 106 | }
|
105 | 107 |
|
106 |
| - private String getMd5(String what){ |
107 |
| - try { |
| 108 | + private String getMd5(String what) { |
| 109 | + try { |
108 | 110 | MessageDigest md = MessageDigest.getInstance("MD5");
|
109 |
| - md.update(what.getBytes("ISO-8859-1")); |
110 |
| - byte[] hash = md.digest(); |
111 |
| - BigInteger bi = new BigInteger(1, hash); |
112 |
| - String result = bi.toString(16); |
113 |
| - if (result.length() % 2 != 0) { |
114 |
| - return "0" + result; |
115 |
| - } |
116 |
| - return result; |
117 |
| - } catch (Exception e) { |
118 |
| - throw new RuntimeException(e); |
| 111 | + md.update(what.getBytes("ISO-8859-1")); |
| 112 | + byte[] hash = md.digest(); |
| 113 | + BigInteger bi = new BigInteger(1, hash); |
| 114 | + String result = bi.toString(16); |
| 115 | + if (result.length() % 2 != 0) { |
| 116 | + return "0" + result; |
119 | 117 | }
|
| 118 | + return result; |
| 119 | + } catch (Exception e) { |
| 120 | + throw new RuntimeException(e); |
| 121 | + } |
120 | 122 | }
|
121 | 123 | }
|
0 commit comments