Skip to content

Commit ad01092

Browse files
committed
Fix test.
1 parent a5fb4e0 commit ad01092

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

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

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
package us.codecraft.webmagic.proxy;
22

33
import org.junit.Test;
4+
import org.mockito.Mockito;
5+
6+
import us.codecraft.webmagic.Request;
47
import us.codecraft.webmagic.Site;
58
import 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
}

0 commit comments

Comments
 (0)