← Back to branch summary

~s-friedemann/tomdroid/sshfs

« back to all changes in this revision

Viewing changes to lib/signpost/signpost-core/src/test/java/oauth/signpost/basic/OAuthProviderTest.java

  • Committer: Benoit Garret
  • Date: 2010-03-06 20:39:56 UTC
  • mto: (185.4.1 sync-ui)
  • mto: This revision was merged to the branch mainline in revision 186.
  • Revision ID: benoit.garret_launchpad@gadz.org-20100306203956-k6vlss6hk2d91j4n
Put the signpost and commons-codec libraries in lib/

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/signpost/signpost-core/src/test/java/oauth/signpost/basic/OAuthProviderTest.java
 
1
package oauth.signpost.basic;
 
2
 
 
3
import static org.junit.Assert.assertEquals;
 
4
import static org.junit.Assert.assertNotNull;
 
5
import static org.junit.Assert.assertTrue;
 
6
import static org.mockito.Matchers.anyObject;
 
7
import static org.mockito.Mockito.verify;
 
8
import static org.mockito.Mockito.when;
 
9
 
 
10
import java.io.ByteArrayInputStream;
 
11
import java.io.ByteArrayOutputStream;
 
12
import java.io.InputStream;
 
13
import java.io.ObjectInputStream;
 
14
import java.io.ObjectOutputStream;
 
15
import java.net.HttpURLConnection;
 
16
import java.net.URL;
 
17
 
 
18
import oauth.signpost.OAuth;
 
19
import oauth.signpost.OAuthConsumer;
 
20
import oauth.signpost.SignpostTestBase;
 
21
import oauth.signpost.exception.OAuthExpectationFailedException;
 
22
import oauth.signpost.http.HttpRequest;
 
23
import oauth.signpost.signature.SignatureMethod;
 
24
 
 
25
import org.junit.Before;
 
26
import org.junit.Test;
 
27
import org.junit.runner.RunWith;
 
28
import org.mockito.Mock;
 
29
import org.mockito.MockitoAnnotations;
 
30
import org.mockito.runners.MockitoJUnit44Runner;
 
31
 
 
32
@RunWith(MockitoJUnit44Runner.class)
 
33
public class OAuthProviderTest extends SignpostTestBase {
 
34
 
 
35
    DefaultOAuthProvider provider;
 
36
 
 
37
    @Mock
 
38
    OAuthConsumer consumerMock;
 
39
 
 
40
    @Mock
 
41
    HttpURLConnection connectionMock;
 
42
 
 
43
    @Before
 
44
    public void prepare() throws Exception {
 
45
 
 
46
        MockitoAnnotations.initMocks(this);
 
47
 
 
48
        // init consumer mock
 
49
        when(consumerMock.getConsumerKey()).thenReturn(CONSUMER_KEY);
 
50
        when(consumerMock.getConsumerSecret()).thenReturn(CONSUMER_SECRET);
 
51
        when(consumerMock.getToken()).thenReturn(TOKEN);
 
52
        when(consumerMock.getTokenSecret()).thenReturn(TOKEN_SECRET);
 
53
 
 
54
        // init connection mock
 
55
        String responseBody = OAuth.OAUTH_TOKEN + "=" + TOKEN + "&"
 
56
                + OAuth.OAUTH_TOKEN_SECRET + "=" + TOKEN_SECRET;
 
57
        InputStream is = new ByteArrayInputStream(responseBody.getBytes());
 
58
        when(connectionMock.getResponseCode()).thenReturn(200);
 
59
        when(connectionMock.getInputStream()).thenReturn(is);
 
60
 
 
61
        provider = new DefaultOAuthProvider(consumerMock,
 
62
                REQUEST_TOKEN_ENDPOINT_URL, ACCESS_TOKEN_ENDPOINT_URL,
 
63
                AUTHORIZE_WEBSITE_URL);
 
64
        provider.setHttpUrlConnection(connectionMock);
 
65
    }
 
66
 
 
67
    @Test(expected = OAuthExpectationFailedException.class)
 
68
    public void shouldThrowExpectationFailedIfConsumerKeyNotSet()
 
69
            throws Exception {
 
70
        provider = new DefaultOAuthProvider(new DefaultOAuthConsumer(null,
 
71
                CONSUMER_SECRET, SignatureMethod.HMAC_SHA1),
 
72
                REQUEST_TOKEN_ENDPOINT_URL, ACCESS_TOKEN_ENDPOINT_URL,
 
73
                AUTHORIZE_WEBSITE_URL);
 
74
        provider.retrieveRequestToken(REQUEST_TOKEN_ENDPOINT_URL);
 
75
    }
 
76
 
 
77
    @Test(expected = OAuthExpectationFailedException.class)
 
78
    public void shouldThrowExpectationFailedIfConsumerSecretNotSet()
 
79
            throws Exception {
 
80
        provider = new DefaultOAuthProvider(new DefaultOAuthConsumer(
 
81
                CONSUMER_KEY, null, SignatureMethod.HMAC_SHA1),
 
82
                REQUEST_TOKEN_ENDPOINT_URL, ACCESS_TOKEN_ENDPOINT_URL,
 
83
                AUTHORIZE_WEBSITE_URL);
 
84
        provider.retrieveRequestToken(REQUEST_TOKEN_ENDPOINT_URL);
 
85
    }
 
86
 
 
87
    @Test
 
88
    public void shouldRetrieveRequestTokenAndUpdateConsumer() throws Exception {
 
89
 
 
90
        String callbackUrl = "http://www.example.com";
 
91
        String result = provider.retrieveRequestToken(callbackUrl);
 
92
 
 
93
        verify(consumerMock).sign((HttpRequest) anyObject());
 
94
        verify(consumerMock).setTokenWithSecret(TOKEN, TOKEN_SECRET);
 
95
        verify(connectionMock).disconnect();
 
96
 
 
97
        assertEquals(AUTHORIZE_WEBSITE_URL + "?" + OAuth.OAUTH_TOKEN + "="
 
98
                + TOKEN + "&" + OAuth.OAUTH_CALLBACK + "="
 
99
                + "http%3A%2F%2Fwww.example.com", result);
 
100
    }
 
101
 
 
102
    @Test
 
103
    public void shouldRespectCustomQueryParametersInAuthWebsiteUrl()
 
104
            throws Exception {
 
105
        provider = new DefaultOAuthProvider(consumerMock,
 
106
                REQUEST_TOKEN_ENDPOINT_URL, ACCESS_TOKEN_ENDPOINT_URL,
 
107
                "http://provider.com/authorize?q=1");
 
108
        provider.setHttpUrlConnection(connectionMock);
 
109
 
 
110
        String callbackUrl = "http://www.example.com";
 
111
        // the URL ctor checks for URL validity
 
112
        URL url = new URL(provider.retrieveRequestToken(callbackUrl));
 
113
        assertTrue(url.getQuery().startsWith("q=1&oauth_token="));
 
114
    }
 
115
 
 
116
    @Test(expected = OAuthExpectationFailedException.class)
 
117
    public void shouldThrowWhenGettingAccessTokenAndRequestTokenNotSet()
 
118
            throws Exception {
 
119
        when(consumerMock.getToken()).thenReturn(null);
 
120
        provider.retrieveAccessToken(null);
 
121
    }
 
122
 
 
123
    @Test(expected = OAuthExpectationFailedException.class)
 
124
    public void shouldThrowWhenGettingAccessTokenAndRequestTokenSecretNotSet()
 
125
            throws Exception {
 
126
        when(consumerMock.getTokenSecret()).thenReturn(null);
 
127
        provider.retrieveAccessToken(null);
 
128
    }
 
129
 
 
130
    @Test
 
131
    public void shouldRetrieveAccessTokenAndUpdateConsumer() throws Exception {
 
132
 
 
133
        provider.retrieveAccessToken(null);
 
134
 
 
135
        verify(consumerMock).sign((HttpRequest) anyObject());
 
136
        verify(consumerMock).setTokenWithSecret(TOKEN, TOKEN_SECRET);
 
137
        verify(connectionMock).disconnect();
 
138
    }
 
139
 
 
140
    @Test
 
141
    public void shouldMakeSpecialResponseParametersAvailableToConsumer()
 
142
            throws Exception {
 
143
 
 
144
        assertTrue(provider.getResponseParameters().isEmpty());
 
145
 
 
146
        String responseBody = OAuth.OAUTH_TOKEN + "=" + TOKEN + "&"
 
147
                + OAuth.OAUTH_TOKEN_SECRET + "=" + TOKEN_SECRET + "&a=1";
 
148
        InputStream is = new ByteArrayInputStream(responseBody.getBytes());
 
149
        when(connectionMock.getInputStream()).thenReturn(is);
 
150
 
 
151
        provider.retrieveRequestToken(null);
 
152
 
 
153
        assertEquals(1, provider.getResponseParameters().size());
 
154
        assertTrue(provider.getResponseParameters().containsKey("a"));
 
155
        assertEquals("1", provider.getResponseParameters().get("a"));
 
156
 
 
157
        responseBody = OAuth.OAUTH_TOKEN + "=" + TOKEN + "&"
 
158
                + OAuth.OAUTH_TOKEN_SECRET + "=" + TOKEN_SECRET + "&b=2&c=3";
 
159
        is = new ByteArrayInputStream(responseBody.getBytes());
 
160
        when(connectionMock.getInputStream()).thenReturn(is);
 
161
 
 
162
        provider.setHttpUrlConnection(connectionMock);
 
163
        provider.retrieveAccessToken(null);
 
164
 
 
165
        assertEquals(2, provider.getResponseParameters().size());
 
166
        assertTrue(provider.getResponseParameters().containsKey("b"));
 
167
        assertTrue(provider.getResponseParameters().containsKey("c"));
 
168
        assertEquals("2", provider.getResponseParameters().get("b"));
 
169
        assertEquals("3", provider.getResponseParameters().get("c"));
 
170
    }
 
171
 
 
172
    @Test
 
173
    public void shouldBeSerializable() throws Exception {
 
174
        // prepare a provider that has response params set
 
175
        DefaultOAuthProvider provider = new DefaultOAuthProvider(consumerMock,
 
176
                REQUEST_TOKEN_ENDPOINT_URL, ACCESS_TOKEN_ENDPOINT_URL,
 
177
                AUTHORIZE_WEBSITE_URL);
 
178
        String responseBody = OAuth.OAUTH_TOKEN + "=" + TOKEN + "&"
 
179
                + OAuth.OAUTH_TOKEN_SECRET + "=" + TOKEN_SECRET + "&a=1";
 
180
        InputStream is = new ByteArrayInputStream(responseBody.getBytes());
 
181
        when(connectionMock.getInputStream()).thenReturn(is);
 
182
        provider.setHttpUrlConnection(connectionMock);
 
183
        provider.retrieveRequestToken(null);
 
184
 
 
185
        // the mock consumer isn't serializable, thus set a normal one
 
186
        OAuthConsumer consumer = new DefaultOAuthConsumer(CONSUMER_KEY,
 
187
                CONSUMER_SECRET, SignatureMethod.HMAC_SHA1);
 
188
        consumer.setTokenWithSecret(TOKEN, TOKEN_SECRET);
 
189
        provider.setConsumer(consumer);
 
190
        provider.setOAuth10a(true);
 
191
 
 
192
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
 
193
        ObjectOutputStream ostream = new ObjectOutputStream(baos);
 
194
        ostream.writeObject(provider);
 
195
 
 
196
        ObjectInputStream istream = new ObjectInputStream(
 
197
                new ByteArrayInputStream(baos.toByteArray()));
 
198
        provider = (DefaultOAuthProvider) istream.readObject();
 
199
 
 
200
        assertEquals(REQUEST_TOKEN_ENDPOINT_URL,
 
201
                provider.getRequestTokenEndpointUrl());
 
202
        assertEquals(ACCESS_TOKEN_ENDPOINT_URL,
 
203
                provider.getAccessTokenEndpointUrl());
 
204
        assertEquals(AUTHORIZE_WEBSITE_URL,
 
205
                provider.getAuthorizationWebsiteUrl());
 
206
        assertEquals(true, provider.isOAuth10a());
 
207
        assertNotNull(provider.getConsumer());
 
208
        assertNotNull(provider.getResponseParameters());
 
209
        assertEquals("1", provider.getResponseParameters().get("a"));
 
210
    }
 
211
}