ActiveResourceTest.java

package com.aol.micro.server.rest.resources;

import static org.hamcrest.Matchers.is;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.util.Map;

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature;
import org.junit.Before;
import org.junit.Test;

import com.aol.micro.server.events.JobName;
import com.aol.micro.server.events.JobsBeingExecuted;
import com.aol.micro.server.events.RequestEvents;
import com.aol.micro.server.events.RequestTypes;
import com.aol.micro.server.events.RequestsBeingExecuted;
import com.aol.micro.server.events.ScheduledJob;
import com.aol.micro.server.rest.jackson.JacksonUtil;
import com.google.common.collect.ImmutableMap;
import com.google.common.eventbus.EventBus;

public class ActiveResourceTest {

    ActiveResource active;
    RequestsBeingExecuted queries1;
    RequestsBeingExecuted queries2;
    JobsBeingExecuted jobs;
    EventBus bus;

    @Before
    public void setUp() throws Exception {
        bus = new EventBus();
        queries1 = new RequestsBeingExecuted(
                                             bus);
        queries2 = new RequestsBeingExecuted(
                                             bus, "partition");
        jobs = new JobsBeingExecuted(
                                     new EventBus(), 10, JobName.Types.SIMPLE);
        RequestTypes types = new RequestTypes(
                                              bus, true);
        types.getMap()
             .put(queries1.getType(), queries1);
        types.getMap()
             .put(queries2.getType(), queries2);
        active = new ActiveResource(
                                    types, jobs);
    }

    @Test
    public void testactiveRequests() {
        bus.post(RequestEvents.start("query", 1l));
        bus.post(RequestEvents.start("query", 2l, "partition", ImmutableMap.of()));
        MockAsyncResponse<String> response = new MockAsyncResponse<>();
        active.activeRequests(response, null);
        assertThat(convert(response.response()).get("events"), is(1));
        active.activeRequests(response, "partition");
        assertThat(convert(response.response()).get("events"), is(1));
    }

    @Test
    public void whenQueriesWithTheSameIdToDifferentTypesEventsIs1ForBoth() {
        bus.post(RequestEvents.start("query", 1l));
        bus.post(RequestEvents.start("query", 1l, "partition", ImmutableMap.of()));
        MockAsyncResponse<String> response = new MockAsyncResponse<>();
        active.activeRequests(response, null);
        assertThat(convert(response.response()).get("events"), is(1));
        active.activeRequests(response, "partition");
        assertThat(convert(response.response()).get("events"), is(1));
    }

    @Test
    public void whenQueriesWithTheSameIdButSameTypesEventsIs2ButSizeIs1() {
        bus.post(RequestEvents.start("query", 1l));
        bus.post(RequestEvents.start("query", 1l));
        MockAsyncResponse<String> response = new MockAsyncResponse<>();
        active.activeRequests(response, null);
        assertThat(convert(response.response()).get("events"), is(2));
        active.activeRequests(response, null);
        Map map = convert(response.response());
        assertThat(((Map) map.get("active")).size(), is(1));
    }

    @Test
    public void testActiveJobs() throws Throwable {
        ProceedingJoinPoint pjp = mock(ProceedingJoinPoint.class);
        Signature signature = mock(Signature.class);
        when(pjp.getSignature()).thenReturn(signature);
        when(pjp.getTarget()).thenReturn(this);
        when(signature.getDeclaringType()).thenReturn(ScheduledJob.class);
        jobs.aroundScheduledJob(pjp);
        MockAsyncResponse<String> response = new MockAsyncResponse<>();
        active.activeJobs(response);
        assertThat(convert(response.response()).get("events"), is(1));

    }

    private Map convert(String str) {
        return JacksonUtil.convertFromJson(str, Map.class);
    }
}