123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191 |
- /*
- * Copyright (c) 2006-2017 DMDirc Developers
- *
- * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
- * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
- * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
- * permit persons to whom the Software is furnished to do so, subject to the following conditions:
- *
- * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
- * Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
- * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS
- * OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
- * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
- */
-
- package com.dmdirc.addons.channelwho;
-
- import com.dmdirc.config.binding.ConfigBinder;
- import com.dmdirc.events.ChannelUserAwayEvent;
- import com.dmdirc.events.DisplayProperty;
- import com.dmdirc.events.ServerNumericEvent;
- import com.dmdirc.interfaces.Connection;
- import com.dmdirc.interfaces.ConnectionManager;
- import com.dmdirc.events.eventbus.EventBus;
- import com.dmdirc.interfaces.GroupChat;
- import com.dmdirc.interfaces.GroupChatManager;
- import com.dmdirc.interfaces.GroupChatUser;
- import com.dmdirc.interfaces.User;
- import com.dmdirc.interfaces.WindowModel;
- import com.dmdirc.interfaces.config.AggregateConfigProvider;
-
- import com.google.common.collect.Lists;
-
- import java.util.Optional;
- import java.util.concurrent.ScheduledExecutorService;
- import java.util.concurrent.ScheduledFuture;
- import java.util.concurrent.TimeUnit;
-
- import org.junit.Before;
- import org.junit.Test;
- import org.junit.runner.RunWith;
- import org.mockito.ArgumentCaptor;
- import org.mockito.Captor;
- import org.mockito.Mock;
- import org.mockito.runners.MockitoJUnitRunner;
-
- import static org.junit.Assert.assertEquals;
- import static org.mockito.Matchers.any;
- import static org.mockito.Matchers.anyLong;
- import static org.mockito.Matchers.eq;
- import static org.mockito.Mockito.doReturn;
- import static org.mockito.Mockito.never;
- import static org.mockito.Mockito.verify;
- import static org.mockito.Mockito.when;
-
- @RunWith(MockitoJUnitRunner.class)
- public class ConnectionHandlerTest {
-
- @Mock private AggregateConfigProvider config;
- @Mock private ConfigBinder configBinder;
- @Mock private WindowModel windowModel;
- @Mock private EventBus eventBus;
- @Mock private ScheduledExecutorService scheduledExecutorService;
- @Mock private ScheduledFuture<?> scheduledFuture;
- @Mock private ConnectionManager connectionManager;
- @Mock private Connection connection;
- @Mock private GroupChat groupChat;
- @Mock private GroupChatUser groupChatUser;
- @Mock private User user;
- @Mock private GroupChatManager groupChatManager;
- @Mock private ServerNumericEvent serverNumericEvent;
- @Mock private ChannelUserAwayEvent channelUserAwayEvent;
- @Captor private ArgumentCaptor<ChannelUserAwayEvent> eventArgumentCaptor;
- private ConnectionHandler instance;
-
- @Before
- public void setUp() throws Exception {
- doReturn(scheduledFuture).when(scheduledExecutorService)
- .scheduleAtFixedRate(any(Runnable.class), anyLong(), anyLong(), any());
- when(windowModel.getEventBus()).thenReturn(eventBus);
- when(connection.getWindowModel()).thenReturn(windowModel);
- when(config.getBinder()).thenReturn(configBinder);
- when(connectionManager.getConnections()).thenReturn(Lists.newArrayList(connection));
- when(connection.getGroupChatManager()).thenReturn(groupChatManager);
- when(groupChatManager.getChannels()).thenReturn(Lists.newArrayList(groupChat));
- when(groupChat.getWindowModel()).thenReturn(windowModel);
- when(groupChat.getConnection()).thenReturn(Optional.of(connection));
- when(configBinder.withDefaultDomain("domain")).thenReturn(configBinder);
- when(windowModel.getConfigManager()).thenReturn(config);
- when(groupChatUser.getNickname()).thenReturn("nickname");
- when(channelUserAwayEvent.getUser()).thenReturn(groupChatUser);
- when(channelUserAwayEvent.getChannel()).thenReturn(groupChat);
- when(groupChatUser.getUser()).thenReturn(user);
- instance = new ConnectionHandler(config, scheduledExecutorService, eventBus,
- "domain", connection);
- instance.handleWhoInterval(5);
-
- when(serverNumericEvent.getConnection()).thenReturn(connection);
- when(channelUserAwayEvent.getChannel()).thenReturn(groupChat);
- }
-
- @Test
- public void testLoad() throws Exception {
- instance.load();
- verify(configBinder).bind(instance, ConnectionHandler.class);
- verify(eventBus).subscribe(instance);
- verify(scheduledExecutorService).scheduleAtFixedRate(any(Runnable.class), eq(5L), eq(5L),
- eq(TimeUnit.MILLISECONDS));
- }
-
- @Test
- public void testUnload() throws Exception {
- instance.unload();
- verify(configBinder).unbind(instance);
- verify(scheduledExecutorService).shutdown();
- verify(eventBus).unsubscribe(instance);
- }
-
- @Test
- public void testHandleWhoInterval() throws Exception {
- instance.handleWhoInterval(10);
- verify(scheduledFuture).cancel(false);
- verify(scheduledExecutorService).scheduleAtFixedRate(any(Runnable.class), eq(5L), eq(5L),
- eq(TimeUnit.MILLISECONDS));
- verify(scheduledExecutorService).scheduleAtFixedRate(any(Runnable.class), eq(10L), eq(10L),
- eq(TimeUnit.MILLISECONDS));
- }
-
- @Test
- public void testCheckWho_True() throws Exception {
- when(config.getOptionBool("domain", "sendwho")).thenReturn(true);
- instance.checkWho();
- verify(config).getOptionBool("domain", "sendwho");
- verify(groupChat).requestUsersInfo();
- }
-
- @Test
- public void testCheckWho_False() throws Exception {
- when(config.getOptionBool("domain", "sendwho")).thenReturn(false);
- instance.checkWho();
- verify(config).getOptionBool("domain", "sendwho");
- verify(groupChat, never()).requestUsersInfo();
- }
-
- @Test
- public void testHandleAwayEvent_WithReason() throws Exception {
- when(channelUserAwayEvent.getReason()).thenReturn(Optional.of("reason"));
- instance.load();
- instance.handleAwayEvent(channelUserAwayEvent);
- verify(channelUserAwayEvent, never())
- .setDisplayProperty(DisplayProperty.DO_NOT_DISPLAY, true);
- verify(connection, never()).requestUserInfo(any());
- }
-
- @Test
- public void testHandleAwayEvent_WithoutReason() throws Exception {
- when(channelUserAwayEvent.getReason()).thenReturn(Optional.empty());
- instance.load();
- instance.handleAwayEvent(channelUserAwayEvent);
- verify(channelUserAwayEvent).setDisplayProperty(DisplayProperty.DO_NOT_DISPLAY, true);
- verify(connection).requestUserInfo(any());
- }
-
- @Test
- public void testHandleServerNumericEvent_301() throws Exception {
- when(serverNumericEvent.getNumeric()).thenReturn(301);
- when(serverNumericEvent.getArgs()).thenReturn(
- new String[]{"", "", "", "nickname", "reason"});
- instance.load();
- when(channelUserAwayEvent.getReason()).thenReturn(Optional.empty());
- when(groupChatUser.getGroupChat()).thenReturn(groupChat);
- instance.handleAwayEvent(channelUserAwayEvent);
- instance.handleServerNumericEvent(serverNumericEvent);
- verify(eventBus).publishAsync(eventArgumentCaptor.capture());
- assertEquals("nickname", eventArgumentCaptor.getValue().getUser().getNickname());
- assertEquals("reason", eventArgumentCaptor.getValue().getReason().get());
- }
-
- @Test
- public void testHandleServerNumericEvent_101() throws Exception {
- when(serverNumericEvent.getNumeric()).thenReturn(101);
- when(channelUserAwayEvent.getReason()).thenReturn(Optional.empty());
- instance.load();
- instance.handleAwayEvent(channelUserAwayEvent);
- instance.handleServerNumericEvent(serverNumericEvent);
- verify(serverNumericEvent, never()).getArgs();
- verify(eventBus, never()).publishAsync(any());
- }
- }
|