Refactor SchedulerDead tests

This commit is contained in:
str4d
2015-07-27 11:50:01 +00:00
parent e704baddd8
commit 45fe238227
3 changed files with 51 additions and 19 deletions

View File

@@ -0,0 +1,34 @@
package net.i2p.client.streaming.impl;
import static org.hamcrest.Matchers.*;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;
import net.i2p.I2PAppContext;
import net.i2p.client.I2PSession;
import net.i2p.util.SimpleTimer2;
import org.junit.Rule;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;
public class ConnectionTest {
@Rule
public MockitoRule rule = MockitoJUnit.rule();
@Mock private ConnectionManager manager;
@Mock private I2PSession session;
@Mock private SchedulerChooser chooser;
@Mock private SimpleTimer2 timer;
@Mock private PacketQueue queue;
@Mock private ConnectionPacketHandler handler;
@Mock private ConnectionOptions opts;
@Test
public void test() {
//Connection conn = new Connection(I2PAppContext.getGlobalContext(), manager, session, chooser, timer, queue, handler, opts, false);
}
}

View File

@@ -6,19 +6,13 @@ import static org.mockito.Mockito.*;
import org.junit.Test;
import org.mockito.Mock;
public class SchedulerDeadTest extends SchedulerImplTestBase {
public class SchedulerDeadTest extends TaskSchedulerTestBase {
private SchedulerDead s;
@Mock private Connection con;
@Mock private ConnectionOptions opts;
protected void initScheduler() {
s = new SchedulerDead(context);
}
@Test
public void testAccept_null() {
assertFalse(s.accept(null));
protected TaskScheduler createScheduler() {
return new SchedulerDead(context);
}
private void setMocks(int now, int discSchOn, int connTimeout, int lifetime, int sendStreamId) {
@@ -33,24 +27,24 @@ public class SchedulerDeadTest extends SchedulerImplTestBase {
@Test
public void testAccept_nothingLeftToDo() {
setMocks(10*60*1000, 9*60*1000 - Connection.DISCONNECT_TIMEOUT, 0, 0, 0);
assertTrue(s.accept(con));
assertTrue(scheduler.accept(con));
}
@Test
public void testAccept_noDisconnectScheduled() {
setMocks(10*60*1000, 0, 0, 0, 0);
assertFalse(s.accept(con));
assertFalse(scheduler.accept(con));
}
@Test
public void testAccept_timedOut() {
setMocks(0, 0, Connection.DISCONNECT_TIMEOUT/2, Connection.DISCONNECT_TIMEOUT, 0);
assertTrue(s.accept(con));
assertTrue(scheduler.accept(con));
}
@Test
public void testEventOccurred() {
s.eventOccurred(con);
scheduler.eventOccurred(con);
verify(con).disconnectComplete();
}
}

View File

@@ -1,7 +1,6 @@
package net.i2p.client.streaming.impl;
import static org.hamcrest.Matchers.*;
import static org.junit.Assert.*;
import static org.junit.Assert.assertFalse;
import static org.mockito.Mockito.*;
import net.i2p.I2PAppContext;
import net.i2p.util.Clock;
@@ -15,7 +14,7 @@ import org.mockito.Spy;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;
public abstract class SchedulerImplTestBase {
public abstract class TaskSchedulerTestBase {
@Rule
public MockitoRule rule = MockitoJUnit.rule();
@@ -24,15 +23,20 @@ public abstract class SchedulerImplTestBase {
@Mock protected Clock clock;
@Mock protected SimpleTimer2 timer;
protected SchedulerDead scheduler;
protected TaskScheduler scheduler;
@Before
public void setUp() {
when(context.clock()).thenReturn(clock);
when(context.simpleTimer2()).thenReturn(timer);
initScheduler();
scheduler = createScheduler();
}
protected abstract void initScheduler();
protected abstract TaskScheduler createScheduler();
@Test
public void testAccept_null() {
assertFalse(scheduler.accept(null));
}
}