Fix test.

pull/1127/head
Joe Zhou
parent a5fb4e0414
commit ad010927f6

@ -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);
} }
} }

Loading…
Cancel
Save