Fix test.

code4craft · Jun 22, 2023 · ad01092 · ad01092
1 parent a5fb4e0
commit ad01092
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/webmagic-core/src/test/java/us/codecraft/webmagic/proxy/SimpleProxyProviderTest.java b/webmagic-core/src/test/java/us/codecraft/webmagic/proxy/SimpleProxyProviderTest.java
@@ -1,6 +1,9 @@
 package us.codecraft.webmagic.proxy;
 
 import org.junit.Test;
+import org.mockito.Mockito;
+
+import us.codecraft.webmagic.Request;
 import us.codecraft.webmagic.Site;
 import us.codecraft.webmagic.Task;
 
@@ -20,11 +23,12 @@ public void test_get_proxy() throws Exception {
         Proxy originProxy1 = new Proxy("127.0.0.1", 1087);
         Proxy originProxy2 = new Proxy("127.0.0.1", 1088);
         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);
-        proxy = proxyProvider.getProxy(TASK);
+        proxy = proxyProvider.getProxy(request, TASK);
         assertThat(proxy).isEqualTo(originProxy2);
-        proxy = proxyProvider.getProxy(TASK);
+        proxy = proxyProvider.getProxy(request, TASK);
         assertThat(proxy).isEqualTo(originProxy1);
     }
 }