Examples with TestMonkeyContext com.netflix.simianarmy.TestMonkeyContext used on opensource projects

Example 1 with TestMonkeyContext

use of com.netflix.simianarmy.TestMonkeyContext in project SimianArmy by Netflix.

the class TestBasicScheduler method testRunner.

@Test
public void testRunner() throws InterruptedException {
    BasicScheduler sched = new BasicScheduler(200, TimeUnit.MILLISECONDS, 1);
    Monkey mockMonkey = mock(Monkey.class);
    when(mockMonkey.context()).thenReturn(new TestMonkeyContext(Enums.MONKEY));
    when(mockMonkey.type()).thenReturn(Enums.MONKEY).thenReturn(Enums.MONKEY);
    final AtomicLong counter = new AtomicLong(0L);
    sched.start(mockMonkey, new Runnable() {

        @Override
        public void run() {
            counter.incrementAndGet();
        }
    });
    Thread.sleep(100);
    Assert.assertEquals(counter.get(), 1);
    Thread.sleep(200);
    Assert.assertEquals(counter.get(), 2);
    sched.stop(mockMonkey);
    Thread.sleep(200);
    Assert.assertEquals(counter.get(), 2);
}

Example 2 with TestMonkeyContext

use of com.netflix.simianarmy.TestMonkeyContext in project SimianArmy by Netflix.

the class TestBasicScheduler method testDelayedStart.

@Test
public void testDelayedStart() throws Exception {
    BasicScheduler sched = new BasicScheduler(1, TimeUnit.HOURS, 1);
    TestMonkeyContext context = new TestMonkeyContext(Enums.MONKEY);
    Monkey mockMonkey = mock(Monkey.class);
    when(mockMonkey.context()).thenReturn(context).thenReturn(context);
    when(mockMonkey.type()).thenReturn(Enums.MONKEY).thenReturn(Enums.MONKEY);
    // first monkey has no previous events, so it runs practically immediately
    FutureTask<Void> task = new FutureTask<Void>(Callables.<Void>returning(null));
    sched.start(mockMonkey, task);
    // make sure that the task gets completed within 100ms
    task.get(100L, TimeUnit.MILLISECONDS);
    sched.stop(mockMonkey);
    // create an event 5 min ago
    Calendar cal = Calendar.getInstance();
    cal.add(Calendar.MINUTE, -5);
    BasicRecorderEvent evt = new BasicRecorderEvent(Enums.MONKEY, EventEnums.EVENT, "region", "test-id", cal.getTime().getTime());
    context.recorder().recordEvent(evt);
    // this time when it runs it will not run immediately since it should be scheduled for 55m from now.
    task = new FutureTask<Void>(Callables.<Void>returning(null));
    sched.start(mockMonkey, task);
    try {
        task.get(100, TimeUnit.MILLISECONDS);
        Assert.fail("The task shouldn't have been completed in 100ms");
    } catch (TimeoutException e) {
    // NOPMD - This is an expected exception
    }
    sched.stop(mockMonkey);
}