View Javadoc

1   /**
2    *       Copyright 2010 Newcastle University
3    *
4    *          http://research.ncl.ac.uk/smart/
5    *
6    * Licensed to the Apache Software Foundation (ASF) under one or more
7    * contributor license agreements.  See the NOTICE file distributed with
8    * this work for additional information regarding copyright ownership.
9    * The ASF licenses this file to You under the Apache License, Version 2.0
10   * (the "License"); you may not use this file except in compliance with
11   * the License.  You may obtain a copy of the License at
12   *
13   *      http://www.apache.org/licenses/LICENSE-2.0
14   *
15   * Unless required by applicable law or agreed to in writing, software
16   * distributed under the License is distributed on an "AS IS" BASIS,
17   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18   * See the License for the specific language governing permissions and
19   * limitations under the License.
20   */
21  
22  package org.apache.amber.oauth2.rs.extractor;
23  
24  import static org.easymock.EasyMock.createStrictMock;
25  import static org.easymock.EasyMock.expect;
26  import static org.easymock.EasyMock.replay;
27  import static org.easymock.EasyMock.verify;
28  
29  import javax.servlet.http.HttpServletRequest;
30  
31  import junit.framework.Assert;
32  
33  import org.junit.Test;
34  
35  
36  /**
37   *
38   *
39   *
40   */
41  public class QueryTokenExtractorTest {
42  
43      @Test
44      public void testGetAccessToken() throws Exception {
45  
46          HttpServletRequest request = createStrictMock(HttpServletRequest.class);
47          // expect(request.getParameter(OAuth.OAUTH_BEARER_TOKEN)).andStubReturn("sometoken");
48          expect(request.getQueryString()).andStubReturn("access_token=sometoken");
49          replay(request);
50          BearerQueryTokenExtractor qte = new BearerQueryTokenExtractor();
51          Assert.assertEquals("sometoken", qte.getAccessToken(request));
52          verify(request);
53  
54      }
55  
56      @Test
57      public void testGetAccessTokenNull() throws Exception {
58  
59          HttpServletRequest request = createStrictMock(HttpServletRequest.class);
60          expect(request.getQueryString()).andStubReturn(null);
61  
62  //        expect(request.getParameter(OAuth.OAUTH_TOKEN)).andStubReturn(null);
63  //        expect(request.getParameter(OAuth.OAUTH_BEARER_TOKEN)).andStubReturn(null);
64          replay(request);
65          BearerQueryTokenExtractor qte = new BearerQueryTokenExtractor();
66          Assert.assertNull(qte.getAccessToken(request));
67          verify(request);
68  
69      }
70  
71  }