Commit c175ea88 authored by yihua.huang's avatar yihua.huang

#more test #484

parent 9b964c0a
...@@ -21,14 +21,15 @@ import static org.mockito.Mockito.verify; ...@@ -21,14 +21,15 @@ import static org.mockito.Mockito.verify;
@RunWith(MockitoJUnitRunner.class) @RunWith(MockitoJUnitRunner.class)
public class DuplicateRemovedSchedulerTest { public class DuplicateRemovedSchedulerTest {
private DuplicateRemovedScheduler duplicateRemovedScheduler = new DuplicateRemovedScheduler() {
@Override
public Request poll(Task task) {
return null;
}
};
@Test @Test
public void test_no_duplicate_removed_for_post_request() throws Exception { 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); DuplicateRemover duplicateRemover = Mockito.mock(DuplicateRemover.class);
duplicateRemovedScheduler.setDuplicateRemover(duplicateRemover); duplicateRemovedScheduler.setDuplicateRemover(duplicateRemover);
Request request = new Request("https://www.google.com/"); Request request = new Request("https://www.google.com/");
...@@ -36,4 +37,14 @@ public class DuplicateRemovedSchedulerTest { ...@@ -36,4 +37,14 @@ public class DuplicateRemovedSchedulerTest {
duplicateRemovedScheduler.push(request, null); duplicateRemovedScheduler.push(request, null);
verify(duplicateRemover,times(0)).isDuplicate(any(Request.class),any(Task.class)); 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));
}
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment