• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 package com.xtremelabs.robolectric.shadows;
2 
3 import com.xtremelabs.robolectric.Robolectric;
4 import com.xtremelabs.robolectric.internal.Implementation;
5 import com.xtremelabs.robolectric.internal.Implements;
6 import com.xtremelabs.robolectric.internal.RealObject;
7 import com.xtremelabs.robolectric.tester.org.apache.http.HttpRequestInfo;
8 import org.apache.commons.logging.Log;
9 import org.apache.commons.logging.LogFactory;
10 import org.apache.http.ConnectionReuseStrategy;
11 import org.apache.http.HttpException;
12 import org.apache.http.HttpHost;
13 import org.apache.http.HttpRequest;
14 import org.apache.http.HttpResponse;
15 import org.apache.http.client.AuthenticationHandler;
16 import org.apache.http.client.HttpRequestRetryHandler;
17 import org.apache.http.client.RedirectHandler;
18 import org.apache.http.client.UserTokenHandler;
19 import org.apache.http.conn.ClientConnectionManager;
20 import org.apache.http.conn.ConnectionKeepAliveStrategy;
21 import org.apache.http.conn.routing.HttpRoutePlanner;
22 import org.apache.http.impl.client.DefaultRequestDirector;
23 import org.apache.http.params.HttpParams;
24 import org.apache.http.protocol.HttpContext;
25 import org.apache.http.protocol.HttpProcessor;
26 import org.apache.http.protocol.HttpRequestExecutor;
27 
28 import java.io.IOException;
29 
30 @SuppressWarnings({"UnusedDeclaration"})
31 @Implements(DefaultRequestDirector.class)
32 public class ShadowDefaultRequestDirector {
33     @RealObject DefaultRequestDirector realObject;
34 
35     protected Log log;
36     protected ClientConnectionManager connectionManager;
37     protected HttpRoutePlanner httpRoutePlanner;
38     protected ConnectionReuseStrategy connectionReuseStrategy;
39     protected ConnectionKeepAliveStrategy connectionKeepAliveStrategy;
40     protected HttpRequestExecutor httpRequestExecutor;
41     protected HttpProcessor httpProcessor;
42     protected HttpRequestRetryHandler httpRequestRetryHandler;
43     protected RedirectHandler redirectHandler;
44     protected AuthenticationHandler targetAuthenticationHandler;
45     protected AuthenticationHandler proxyAuthenticationHandler;
46     protected UserTokenHandler userTokenHandler;
47     protected HttpParams httpParams;
48 
49     com.xtremelabs.robolectric.tester.org.apache.http.impl.client.DefaultRequestDirector redirector;
50 
__constructor__( Log log, HttpRequestExecutor requestExec, ClientConnectionManager conman, ConnectionReuseStrategy reustrat, ConnectionKeepAliveStrategy kastrat, HttpRoutePlanner rouplan, HttpProcessor httpProcessor, HttpRequestRetryHandler retryHandler, RedirectHandler redirectHandler, AuthenticationHandler targetAuthHandler, AuthenticationHandler proxyAuthHandler, UserTokenHandler userTokenHandler, HttpParams params)51     public void __constructor__(
52             Log log,
53             HttpRequestExecutor requestExec,
54             ClientConnectionManager conman,
55             ConnectionReuseStrategy reustrat,
56             ConnectionKeepAliveStrategy kastrat,
57             HttpRoutePlanner rouplan,
58             HttpProcessor httpProcessor,
59             HttpRequestRetryHandler retryHandler,
60             RedirectHandler redirectHandler,
61             AuthenticationHandler targetAuthHandler,
62             AuthenticationHandler proxyAuthHandler,
63             UserTokenHandler userTokenHandler,
64             HttpParams params) {
65         this.log = log;
66         this.httpRequestExecutor = requestExec;
67         this.connectionManager = conman;
68         this.connectionReuseStrategy = reustrat;
69         this.connectionKeepAliveStrategy = kastrat;
70         this.httpRoutePlanner = rouplan;
71         this.httpProcessor = httpProcessor;
72         this.httpRequestRetryHandler = retryHandler;
73         this.redirectHandler = redirectHandler;
74         this.targetAuthenticationHandler = targetAuthHandler;
75         this.proxyAuthenticationHandler = proxyAuthHandler;
76         this.userTokenHandler = userTokenHandler;
77         this.httpParams = params;
78 
79         try {
80             redirector = new com.xtremelabs.robolectric.tester.org.apache.http.impl.client.DefaultRequestDirector(
81                   log,
82                   requestExec,
83                   conman,
84                   reustrat,
85                   kastrat,
86                   rouplan,
87                   httpProcessor,
88                   retryHandler,
89                   redirectHandler,
90                   targetAuthHandler,
91                   proxyAuthHandler,
92                   userTokenHandler,
93                   params
94             );
95         } catch (IllegalArgumentException ignored) {
96             Robolectric.getFakeHttpLayer().interceptHttpRequests(true);
97         }
98     }
99 
__constructor__( HttpRequestExecutor requestExec, ClientConnectionManager conman, ConnectionReuseStrategy reustrat, ConnectionKeepAliveStrategy kastrat, HttpRoutePlanner rouplan, HttpProcessor httpProcessor, HttpRequestRetryHandler retryHandler, RedirectHandler redirectHandler, AuthenticationHandler targetAuthHandler, AuthenticationHandler proxyAuthHandler, UserTokenHandler userTokenHandler, HttpParams params)100     public void __constructor__(
101             HttpRequestExecutor requestExec,
102             ClientConnectionManager conman,
103             ConnectionReuseStrategy reustrat,
104             ConnectionKeepAliveStrategy kastrat,
105             HttpRoutePlanner rouplan,
106             HttpProcessor httpProcessor,
107             HttpRequestRetryHandler retryHandler,
108             RedirectHandler redirectHandler,
109             AuthenticationHandler targetAuthHandler,
110             AuthenticationHandler proxyAuthHandler,
111             UserTokenHandler userTokenHandler,
112             HttpParams params) {
113         __constructor__(
114                 LogFactory.getLog(DefaultRequestDirector.class),
115                 requestExec,
116                 conman,
117                 reustrat,
118                 kastrat,
119                 rouplan,
120                 httpProcessor,
121                 retryHandler,
122                 redirectHandler,
123                 targetAuthHandler,
124                 proxyAuthHandler,
125                 userTokenHandler,
126                 params);
127     }
128 
getSentHttpRequest(int index)129     public static HttpRequest getSentHttpRequest(int index) {
130         return getSentHttpRequestInfo(index).getHttpRequest();
131     }
132 
getLatestSentHttpRequest()133     public static HttpRequest getLatestSentHttpRequest() {
134         return getLatestSentHttpRequestInfo().getHttpRequest();
135     }
136 
getLatestSentHttpRequestInfo()137     public static HttpRequestInfo getLatestSentHttpRequestInfo() {
138         int requestCount = Robolectric.getFakeHttpLayer().getSentHttpRequestInfos().size();
139         return Robolectric.getFakeHttpLayer().getSentHttpRequestInfo(requestCount - 1);
140     }
141 
getSentHttpRequestInfo(int index)142     public static HttpRequestInfo getSentHttpRequestInfo(int index) {
143         return Robolectric.getFakeHttpLayer().getSentHttpRequestInfo(index);
144     }
145 
146     @Implementation
execute(HttpHost httpHost, HttpRequest httpRequest, HttpContext httpContext)147     public HttpResponse execute(HttpHost httpHost, HttpRequest httpRequest, HttpContext httpContext) throws HttpException, IOException {
148         if (Robolectric.getFakeHttpLayer().isInterceptingHttpRequests()) {
149             return Robolectric.getFakeHttpLayer().emulateRequest(httpHost, httpRequest, httpContext, realObject);
150         } else {
151             return redirector.execute(httpHost, httpRequest, httpContext);
152         }
153     }
154 
getLog()155     public Log getLog() {
156         return log;
157     }
158 
getConnectionManager()159     public ClientConnectionManager getConnectionManager() {
160         return connectionManager;
161     }
162 
getHttpRoutePlanner()163     public HttpRoutePlanner getHttpRoutePlanner() {
164         return httpRoutePlanner;
165     }
166 
getConnectionReuseStrategy()167     public ConnectionReuseStrategy getConnectionReuseStrategy() {
168         return connectionReuseStrategy;
169     }
170 
getConnectionKeepAliveStrategy()171     public ConnectionKeepAliveStrategy getConnectionKeepAliveStrategy() {
172         return connectionKeepAliveStrategy;
173     }
174 
getHttpRequestExecutor()175     public HttpRequestExecutor getHttpRequestExecutor() {
176         return httpRequestExecutor;
177     }
178 
getHttpProcessor()179     public HttpProcessor getHttpProcessor() {
180         return httpProcessor;
181     }
182 
getHttpRequestRetryHandler()183     public HttpRequestRetryHandler getHttpRequestRetryHandler() {
184         return httpRequestRetryHandler;
185     }
186 
getRedirectHandler()187     public RedirectHandler getRedirectHandler() {
188         return redirectHandler;
189     }
190 
getTargetAuthenticationHandler()191     public AuthenticationHandler getTargetAuthenticationHandler() {
192         return targetAuthenticationHandler;
193     }
194 
getProxyAuthenticationHandler()195     public AuthenticationHandler getProxyAuthenticationHandler() {
196         return proxyAuthenticationHandler;
197     }
198 
getUserTokenHandler()199     public UserTokenHandler getUserTokenHandler() {
200         return userTokenHandler;
201     }
202 
getHttpParams()203     public HttpParams getHttpParams() {
204         return httpParams;
205     }
206 }