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 1
1
package us .codecraft .webmagic .proxy ;
2
2
3
3
import org .junit .Test ;
4
+ import org .mockito .Mockito ;
5
+
6
+ import us .codecraft .webmagic .Request ;
4
7
import us .codecraft .webmagic .Site ;
5
8
import us .codecraft .webmagic .Task ;
6
9
@@ -20,11 +23,12 @@ public void test_get_proxy() throws Exception {
20
23
Proxy originProxy1 = new Proxy ("127.0.0.1" , 1087 );
21
24
Proxy originProxy2 = new Proxy ("127.0.0.1" , 1088 );
22
25
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 );
24
28
assertThat (proxy ).isEqualTo (originProxy1 );
25
- proxy = proxyProvider .getProxy (TASK );
29
+ proxy = proxyProvider .getProxy (request , TASK );
26
30
assertThat (proxy ).isEqualTo (originProxy2 );
27
- proxy = proxyProvider .getProxy (TASK );
31
+ proxy = proxyProvider .getProxy (request , TASK );
28
32
assertThat (proxy ).isEqualTo (originProxy1 );
29
33
}
30
34
}
You can’t perform that action at this time.
0 commit comments