You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

QAuthManagerTest.java 7.9KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211
  1. /*
  2. * Copyright (c) 2006-2015 DMDirc Developers
  3. *
  4. * Permission is hereby granted, free of charge, to any person obtaining a copy
  5. * of this software and associated documentation files (the "Software"), to deal
  6. * in the Software without restriction, including without limitation the rights
  7. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  8. * copies of the Software, and to permit persons to whom the Software is
  9. * furnished to do so, subject to the following conditions:
  10. *
  11. * The above copyright notice and this permission notice shall be included in
  12. * all copies or substantial portions of the Software.
  13. *
  14. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  20. * SOFTWARE.
  21. */
  22. package com.dmdirc.addons.qauth;
  23. import com.dmdirc.DMDircMBassador;
  24. import com.dmdirc.Invite;
  25. import com.dmdirc.Query;
  26. import com.dmdirc.config.ConfigBinder;
  27. import com.dmdirc.events.QueryMessageEvent;
  28. import com.dmdirc.events.ServerConnectedEvent;
  29. import com.dmdirc.events.ServerInviteReceivedEvent;
  30. import com.dmdirc.events.ServerNoticeEvent;
  31. import com.dmdirc.interfaces.Connection;
  32. import com.dmdirc.interfaces.User;
  33. import com.dmdirc.interfaces.config.AggregateConfigProvider;
  34. import com.dmdirc.plugins.PluginInfo;
  35. import java.util.Optional;
  36. import org.junit.Before;
  37. import org.junit.Test;
  38. import org.junit.runner.RunWith;
  39. import org.mockito.Mock;
  40. import org.mockito.runners.MockitoJUnitRunner;
  41. import static org.mockito.Matchers.anyString;
  42. import static org.mockito.Mockito.never;
  43. import static org.mockito.Mockito.verify;
  44. import static org.mockito.Mockito.verifyZeroInteractions;
  45. import static org.mockito.Mockito.when;
  46. @RunWith(MockitoJUnitRunner.class)
  47. public class QAuthManagerTest {
  48. @Mock private DMDircMBassador eventBus;
  49. @Mock private PluginInfo pluginInfo;
  50. @Mock private AggregateConfigProvider config;
  51. @Mock private ConfigBinder configBinder;
  52. @Mock private ServerConnectedEvent serverConnectedEvent;
  53. @Mock private ServerInviteReceivedEvent serverInviteReceivedEvent;
  54. @Mock private ServerNoticeEvent serverNoticeEvent;
  55. @Mock private QueryMessageEvent queryMessageEvent;
  56. @Mock private User user;
  57. @Mock private Query query;
  58. @Mock private Invite invite;
  59. @Mock private Connection connection;
  60. private QAuthManager instance;
  61. @Before
  62. public void setUp() throws Exception {
  63. when(serverConnectedEvent.getConnection()).thenReturn(connection);
  64. when(serverInviteReceivedEvent.getConnection()).thenReturn(connection);
  65. when(serverInviteReceivedEvent.getInvite()).thenReturn(invite);
  66. when(serverInviteReceivedEvent.getChannel()).thenReturn("#channel");
  67. when(serverInviteReceivedEvent.getUser()).thenReturn(user);
  68. when(serverNoticeEvent.getConnection()).thenReturn(connection);
  69. when(serverNoticeEvent.getUser()).thenReturn(user);
  70. when(serverNoticeEvent.getMessage()).thenReturn("You are now logged in as username.");
  71. when(queryMessageEvent.getQuery()).thenReturn(query);
  72. when(queryMessageEvent.getUser()).thenReturn(user);
  73. when(queryMessageEvent.getMessage()).thenReturn("You are now logged in as username.");
  74. when(user.getNickname()).thenReturn("Q");
  75. when(query.getConnection()).thenReturn(Optional.of(connection));
  76. when(connection.getNetwork()).thenReturn("Quakenet");
  77. when(config.getBinder()).thenReturn(configBinder);
  78. when(configBinder.withDefaultDomain("pluginDomain")).thenReturn(configBinder);
  79. instance = new QAuthManager("pluginDomain", pluginInfo, config, eventBus);
  80. instance.handleUsername("username");
  81. instance.handlePassword("password");
  82. instance.handleAcceptInvites(true);
  83. instance.handleAutoInvite(true);
  84. }
  85. @Test
  86. public void testLoad() {
  87. instance.load();
  88. verify(configBinder).bind(instance, QAuthManager.class);
  89. verify(eventBus).subscribe(instance);
  90. }
  91. @Test
  92. public void testUnload() {
  93. instance.unload();
  94. verify(configBinder).unbind(instance);
  95. verify(eventBus).unsubscribe(instance);
  96. }
  97. @Test
  98. public void testConnected_IncorrectNetwork() {
  99. when(connection.getNetwork()).thenReturn("NotQuakenet");
  100. instance.handleConnect(serverConnectedEvent);
  101. verify(connection, never()).sendMessage(anyString(), anyString());
  102. }
  103. @Test
  104. public void testConnected_Quakenet() {
  105. instance.handleConnect(serverConnectedEvent);
  106. verify(connection).sendMessage("Q@Cserve.quakenet.org", "auth username password");
  107. }
  108. @Test
  109. public void testInvites_NoHandle() {
  110. instance.handleAcceptInvites(false);
  111. instance.handleInvite(serverInviteReceivedEvent);
  112. verifyZeroInteractions(invite);
  113. }
  114. @Test
  115. public void testInvites_NotQuakenet() {
  116. when(connection.getNetwork()).thenReturn("NotQuakenet");
  117. instance.handleInvite(serverInviteReceivedEvent);
  118. verifyZeroInteractions(invite);
  119. }
  120. @Test
  121. public void testInvites_Quakenet_NotQ() {
  122. when(user.getNickname()).thenReturn("NotQ");
  123. instance.handleAcceptInvites(false);
  124. instance.handleInvite(serverInviteReceivedEvent);
  125. verifyZeroInteractions(invite);
  126. }
  127. @Test
  128. public void testInvites_Quakenet_Q() {
  129. instance.handleInvite(serverInviteReceivedEvent);
  130. verify(invite).accept();
  131. }
  132. @Test
  133. public void testCommunication_Message_NoConnection() {
  134. when(queryMessageEvent.getQuery()).thenReturn(query);
  135. when(query.getConnection()).thenReturn(Optional.<Connection>empty());
  136. instance.handleMessages(queryMessageEvent);
  137. verifyZeroInteractions(connection);
  138. }
  139. @Test
  140. public void testCommunication_Message_NotQuakenet() {
  141. when(connection.getNetwork()).thenReturn("NotQuakenet");
  142. instance.handleMessages(queryMessageEvent);
  143. verify(connection, never()).sendMessage(anyString(), anyString());
  144. }
  145. @Test
  146. public void testCommunication_Message_Quakenet_NotQ() {
  147. when(user.getNickname()).thenReturn("NotQ");
  148. instance.handleMessages(queryMessageEvent);
  149. verify(connection, never()).sendMessage(anyString(), anyString());
  150. }
  151. @Test
  152. public void testCommunication_Message_Quakenet_Q() {
  153. instance.handleMessages(queryMessageEvent);
  154. verify(connection).sendMessage("Q@Cserve.quakenet.org", "invite");
  155. }
  156. @Test
  157. public void testCommunication_Message_Off() {
  158. instance.handleAutoInvite(false);
  159. instance.handleMessages(queryMessageEvent);
  160. verify(connection, never()).sendMessage("Q@Cserve.quakenet.org", "invite");
  161. }
  162. @Test
  163. public void testCommunication_Notice_NotQuakenet() {
  164. when(connection.getNetwork()).thenReturn("NotQuakenet");
  165. instance.handleNotices(serverNoticeEvent);
  166. verify(connection, never()).sendMessage(anyString(), anyString());
  167. }
  168. @Test
  169. public void testCommunication_Notice_Quakenet_NotQ() {
  170. when(user.getNickname()).thenReturn("NotQ");
  171. instance.handleNotices(serverNoticeEvent);
  172. verify(connection, never()).sendMessage(anyString(), anyString());
  173. }
  174. @Test
  175. public void testCommunication_Notice_Quakenet_Q() {
  176. instance.handleNotices(serverNoticeEvent);
  177. verify(connection).sendMessage("Q@Cserve.quakenet.org", "invite");
  178. }
  179. @Test
  180. public void testCommunication_Notice_Off() {
  181. instance.handleAutoInvite(false);
  182. instance.handleNotices(serverNoticeEvent);
  183. verify(connection, never()).sendMessage("Q@Cserve.quakenet.org", "invite");
  184. }
  185. }