/*******************************************************************************
* 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.websocket.impl;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.testng.MockitoTestNGListener;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Listeners;
import org.testng.annotations.Test;
import javax.websocket.RemoteEndpoint;
import javax.websocket.Session;
import java.io.IOException;
import java.util.Optional;
import static java.util.Collections.emptySet;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.anyString;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
/**
* Test for {@link BasicWebSocketMessageTransmitter}
*
* @author Dmitry Kuleshov
*/
@Listeners(MockitoTestNGListener.class)
public class BasicWebSocketMessageTransmitterTest {
private static final String MESSAGE = "message";
private static final String ENDPOINT_ID = "id";
@Mock
private WebSocketSessionRegistry registry;
@Mock
private MessagesReSender reSender;
@InjectMocks
private BasicWebSocketMessageTransmitter transmitter;
@Mock
private Session session;
@Mock
private RemoteEndpoint.Basic remote;
@BeforeMethod
public void setUp() throws Exception {
when(session.getBasicRemote()).thenReturn(remote);
when(session.isOpen()).thenReturn(true);
when(registry.get(ENDPOINT_ID)).thenReturn(Optional.of(session));
when(registry.getSessions()).thenReturn(emptySet());
}
@Test
public void shouldSendDirectMessageIfSessionIsOpenAndEndpointIsSet() throws IOException {
transmitter.transmit(ENDPOINT_ID, MESSAGE);
verify(session).getBasicRemote();
verify(remote).sendText(MESSAGE);
verify(reSender, never()).add(eq(ENDPOINT_ID), anyString());
}
@Test
public void shouldAddMessageToPendingIfSessionIsNotOpenedAndEndpointIsSet() throws IOException {
when(session.isOpen()).thenReturn(false);
transmitter.transmit(ENDPOINT_ID, MESSAGE);
verify(session, never()).getBasicRemote();
verify(remote, never()).sendText(MESSAGE);
verify(reSender).add(ENDPOINT_ID, MESSAGE);
}
}