Mobai
/
magic-Dependency
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix test.

master
Joe Zhou 2023-06-22 10:40:46 +08:00
parent a5fb4e0414
commit ad010927f6
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
webmagic-core/src/test/java/us/codecraft/webmagic/proxy/SimpleProxyProviderTest.java
View File

@ -1,6 +1,9 @@
package us.codecraft.webmagic.proxy; package us.codecraft.webmagic.proxy;
import org.junit.Test; import org.junit.Test;
import org.mockito.Mockito;
import us.codecraft.webmagic.Request;
import us.codecraft.webmagic.Site; import us.codecraft.webmagic.Site;
import us.codecraft.webmagic.Task; import us.codecraft.webmagic.Task;
@ -20,11 +23,12 @@ public class SimpleProxyProviderTest {
Proxy originProxy1 = new Proxy("127.0.0.1", 1087); Proxy originProxy1 = new Proxy("127.0.0.1", 1087);
Proxy originProxy2 = new Proxy("127.0.0.1", 1088); Proxy originProxy2 = new Proxy("127.0.0.1", 1088);
SimpleProxyProvider proxyProvider = SimpleProxyProvider.from(originProxy1, originProxy2); SimpleProxyProvider proxyProvider = SimpleProxyProvider.from(originProxy1, originProxy2);
Proxy proxy = proxyProvider.getProxy(null, TASK); Request request = Mockito.mock(Request.class);
Proxy proxy = proxyProvider.getProxy(request, TASK);
assertThat(proxy).isEqualTo(originProxy1); assertThat(proxy).isEqualTo(originProxy1);
proxy = proxyProvider.getProxy(TASK); proxy = proxyProvider.getProxy(request, TASK);
assertThat(proxy).isEqualTo(originProxy2); assertThat(proxy).isEqualTo(originProxy2);
proxy = proxyProvider.getProxy(TASK); proxy = proxyProvider.getProxy(request, TASK);
assertThat(proxy).isEqualTo(originProxy1); assertThat(proxy).isEqualTo(originProxy1);
} }
} }