File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed
webmagic-core/src/test/java/us/codecraft/webmagic/proxy Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change 11package us .codecraft .webmagic .proxy ;
22
33import org .junit .Test ;
4+ import org .mockito .Mockito ;
5+
6+ import us .codecraft .webmagic .Request ;
47import us .codecraft .webmagic .Site ;
58import us .codecraft .webmagic .Task ;
69
@@ -20,11 +23,12 @@ public void test_get_proxy() throws Exception {
2023 Proxy originProxy1 = new Proxy ("127.0.0.1" , 1087 );
2124 Proxy originProxy2 = new Proxy ("127.0.0.1" , 1088 );
2225 SimpleProxyProvider proxyProvider = SimpleProxyProvider .from (originProxy1 , originProxy2 );
23- Proxy proxy = proxyProvider .getProxy (null , TASK );
26+ Request request = Mockito .mock (Request .class );
27+ Proxy proxy = proxyProvider .getProxy (request , TASK );
2428 assertThat (proxy ).isEqualTo (originProxy1 );
25- proxy = proxyProvider .getProxy (TASK );
29+ proxy = proxyProvider .getProxy (request , TASK );
2630 assertThat (proxy ).isEqualTo (originProxy2 );
27- proxy = proxyProvider .getProxy (TASK );
31+ proxy = proxyProvider .getProxy (request , TASK );
2832 assertThat (proxy ).isEqualTo (originProxy1 );
2933 }
3034}
You can’t perform that action at this time.
0 commit comments