From c175ea88c0bcc2a35055f7d851fc98bbd37e4838 Mon Sep 17 00:00:00 2001 From: "yihua.huang" Date: Sat, 11 Mar 2017 11:43:18 +0800 Subject: [PATCH] #more test #484 --- .../DuplicateRemovedSchedulerTest.java | 23 ++++++++++++++----- 1 file changed, 17 insertions(+), 6 deletions(-) diff --git a/webmagic-core/src/test/java/us/codecraft/webmagic/scheduler/DuplicateRemovedSchedulerTest.java b/webmagic-core/src/test/java/us/codecraft/webmagic/scheduler/DuplicateRemovedSchedulerTest.java index da69129..a098049 100644 --- a/webmagic-core/src/test/java/us/codecraft/webmagic/scheduler/DuplicateRemovedSchedulerTest.java +++ b/webmagic-core/src/test/java/us/codecraft/webmagic/scheduler/DuplicateRemovedSchedulerTest.java @@ -21,14 +21,15 @@ import static org.mockito.Mockito.verify; @RunWith(MockitoJUnitRunner.class) public class DuplicateRemovedSchedulerTest { + private DuplicateRemovedScheduler duplicateRemovedScheduler = new DuplicateRemovedScheduler() { + @Override + public Request poll(Task task) { + return null; + } + }; + @Test public void test_no_duplicate_removed_for_post_request() throws Exception { - DuplicateRemovedScheduler duplicateRemovedScheduler = new DuplicateRemovedScheduler() { - @Override - public Request poll(Task task) { - return null; - } - }; DuplicateRemover duplicateRemover = Mockito.mock(DuplicateRemover.class); duplicateRemovedScheduler.setDuplicateRemover(duplicateRemover); Request request = new Request("https://www.google.com/"); @@ -36,4 +37,14 @@ public class DuplicateRemovedSchedulerTest { duplicateRemovedScheduler.push(request, null); verify(duplicateRemover,times(0)).isDuplicate(any(Request.class),any(Task.class)); } + + @Test + public void test_duplicate_removed_for_get_request() throws Exception { + DuplicateRemover duplicateRemover = Mockito.mock(DuplicateRemover.class); + duplicateRemovedScheduler.setDuplicateRemover(duplicateRemover); + Request request = new Request("https://www.google.com/"); + request.setMethod(HttpConstant.Method.GET); + duplicateRemovedScheduler.push(request, null); + verify(duplicateRemover,times(1)).isDuplicate(any(Request.class),any(Task.class)); + } }