View Javadoc

1   package com.atlassian.sal.testresources.net;
2   
3   import com.atlassian.sal.api.net.Request;
4   import com.atlassian.sal.api.net.RequestFactory;
5   
6   import java.util.HashMap;
7   import java.util.Map;
8   
9   /**
10   * Mock request factory.  Either mock a request and call addRequest() with the URL you want that request
11   * returned for, or the factory, if no request has been set for that URL, will create a MockRequest for you.
12   */
13  public class MockRequestFactory implements RequestFactory<Request<?, ?>>
14  {
15      private final Map<String, Request<?, ?>> requestMap = new HashMap<String, Request<?, ?>>();
16  
17      public void addRequest(final String url, final Request<?, ?> request)
18      {
19          requestMap.put(url, request);
20      }
21  
22      public Request<?, ?> createRequest(final Request.MethodType methodType, final String url)
23      {
24          if (requestMap.containsKey(url))
25          {
26              return requestMap.get(url);
27          }
28          else
29          {
30              return new MockRequest(methodType, url);
31          }
32      }
33  
34      public boolean supportsHeader()
35      {
36          return true;
37      }
38  }