package com.atlassian.sal.testresources.net;
   
   import com.atlassian.sal.api.net.Request;
   import com.atlassian.sal.api.net.RequestFactory;
   
   import java.util.HashMap;
   import java.util.Map;
   
   /**
   * Mock request factory.  Either mock a request and call addRequest() with the URL you want that request
   * returned for, or the factory, if no request has been set for that URL, will create a MockRequest for you.
   */
  public class MockRequestFactory implements RequestFactory<Request<?, ?>> {
      private final Map<String, Request<?, ?>> requestMap = new HashMap<>();
  
      public void addRequest(final String url, final Request<?, ?> request) {
          requestMap.put(url, request);
      }
  
      public Request<?, ?> createRequest(final Request.MethodType methodType, final String url) {
          if (requestMap.containsKey(url)) {
              return requestMap.get(url);
          } else {
              return new MockRequest(methodType, url);
          }
      }
  
      public boolean supportsHeader() {
          return true;
      }
  }