JsonRpcMessageReceiverTest.java

/*******************************************************************************
 * Copyright (c) 2012-2017 Codenvy, S.A.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Codenvy, S.A. - initial API and implementation
 *******************************************************************************/
package org.eclipse.che.api.core.jsonrpc.commons;

import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.testng.MockitoTestNGListener;
import org.testng.annotations.Listeners;
import org.testng.annotations.Test;

import static java.util.Collections.singletonList;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

/**
 * Tests for {@link JsonRpcMessageReceiver}
 */
@Listeners(MockitoTestNGListener.class)
public class JsonRpcMessageReceiverTest {

    static final String MESSAGE     = "message";
    static final String ENDPOINT_ID = "endpoint-id";

    @Mock
    RequestDispatcher       requestDispatcher;
    @Mock
    ResponseDispatcher      responseDispatcher;
    @Mock
    JsonRpcErrorTransmitter errorTransmitter;
    @Mock
    JsonRpcQualifier        jsonRpcQualifier;
    @Mock
    JsonRpcUnmarshaller     jsonRpcUnmarshaller;
    @InjectMocks
    JsonRpcMessageReceiver  jsonRpcMessageReceiver;

    @Test
    public void shouldValidateMessage() throws Exception {
        jsonRpcMessageReceiver.receive(ENDPOINT_ID, MESSAGE);

        verify(jsonRpcQualifier).isValidJson(MESSAGE);
    }

    @Test
    public void shouldTransmitErrorWhenValidationFailed() throws Exception {
        when(jsonRpcQualifier.isValidJson(MESSAGE)).thenReturn(false);

        jsonRpcMessageReceiver.receive(ENDPOINT_ID, MESSAGE);

        verify(errorTransmitter).transmit(eq(ENDPOINT_ID), any(JsonRpcException.class));
    }

    @Test
    public void shouldNotTransmitErrorWhenValidationSucceeded() throws Exception {
        when(jsonRpcQualifier.isValidJson(MESSAGE)).thenReturn(true);

        jsonRpcMessageReceiver.receive(ENDPOINT_ID, MESSAGE);

        verify(errorTransmitter, never()).transmit(eq(ENDPOINT_ID), any(JsonRpcException.class));
    }

    @Test
    public void shouldUnmarshalArray() throws Exception {
        jsonRpcMessageReceiver.receive(ENDPOINT_ID, MESSAGE);

        verify(jsonRpcUnmarshaller).unmarshalArray(MESSAGE);
    }

    @Test
    public void shouldDispatchResponseIfResponseReceived() throws Exception {
        when(jsonRpcQualifier.isJsonRpcResponse(MESSAGE)).thenReturn(true);
        when(jsonRpcQualifier.isJsonRpcRequest(MESSAGE)).thenReturn(false);
        when(jsonRpcUnmarshaller.unmarshalArray(any())).thenReturn(singletonList(MESSAGE));

        jsonRpcMessageReceiver.receive(ENDPOINT_ID, MESSAGE);

        verify(responseDispatcher).dispatch(eq(ENDPOINT_ID), any(JsonRpcResponse.class));
    }

    @Test
    public void shouldDispatchRequestIfRequestReceived() throws Exception {
        when(jsonRpcQualifier.isJsonRpcRequest(MESSAGE)).thenReturn(true);
        when(jsonRpcQualifier.isJsonRpcResponse(MESSAGE)).thenReturn(false);
        when(jsonRpcUnmarshaller.unmarshalArray(any())).thenReturn(singletonList(MESSAGE));

        jsonRpcMessageReceiver.receive(ENDPOINT_ID, MESSAGE);

        verify(requestDispatcher).dispatch(eq(ENDPOINT_ID), any(JsonRpcRequest.class));
    }
}