mirror of
https://github.com/vanitasvitae/Smack.git
synced 2024-06-29 14:54:50 +02:00
This also removes the powermock dependency. Although powermock is a fine library, it currently prevents dropping Junit4. And since we only use the Whitebox API of powermock, this simply replaced powermock's Whitebox with our own.
292 lines
12 KiB
Java
292 lines
12 KiB
Java
/**
|
|
*
|
|
* Copyright the original author or authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
package org.jivesoftware.smackx.bytestreams.socks5;
|
|
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
import static org.mockito.Mockito.mock;
|
|
import static org.mockito.Mockito.never;
|
|
import static org.mockito.Mockito.timeout;
|
|
import static org.mockito.Mockito.verify;
|
|
|
|
import org.jivesoftware.smack.XMPPConnection;
|
|
import org.jivesoftware.smack.packet.IQ;
|
|
import org.jivesoftware.smack.packet.StanzaError;
|
|
import org.jivesoftware.smack.test.util.Whitebox;
|
|
|
|
import org.jivesoftware.smackx.bytestreams.BytestreamRequest;
|
|
import org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream;
|
|
import org.jivesoftware.smackx.disco.ServiceDiscoveryManager;
|
|
|
|
import org.junit.jupiter.api.BeforeEach;
|
|
import org.junit.jupiter.api.Test;
|
|
import org.jxmpp.jid.DomainBareJid;
|
|
import org.jxmpp.jid.EntityFullJid;
|
|
import org.jxmpp.jid.JidTestUtil;
|
|
import org.jxmpp.jid.impl.JidCreate;
|
|
import org.mockito.ArgumentCaptor;
|
|
|
|
/**
|
|
* Test for the InitiationListener class.
|
|
*
|
|
* @author Henning Staib
|
|
*/
|
|
public class InitiationListenerTest {
|
|
|
|
private static final int TIMEOUT = 10000;
|
|
|
|
private static final EntityFullJid initiatorJID = JidTestUtil.DUMMY_AT_EXAMPLE_ORG_SLASH_DUMMYRESOURCE;
|
|
private static final EntityFullJid targetJID = JidTestUtil.FULL_JID_1_RESOURCE_1;
|
|
private static final DomainBareJid proxyJID = JidTestUtil.MUC_EXAMPLE_ORG;
|
|
private static final String proxyAddress = "127.0.0.1";
|
|
private static final String sessionID = "session_id";
|
|
|
|
private XMPPConnection connection;
|
|
private Socks5BytestreamManager byteStreamManager;
|
|
private InitiationListener initiationListener;
|
|
private Bytestream initBytestream;
|
|
|
|
/**
|
|
* Initialize fields used in the tests.
|
|
*/
|
|
@BeforeEach
|
|
public void setup() {
|
|
|
|
// mock connection
|
|
connection = mock(XMPPConnection.class);
|
|
|
|
// create service discovery manager for mocked connection
|
|
ServiceDiscoveryManager.getInstanceFor(connection);
|
|
|
|
// initialize Socks5ByteStreamManager to get the InitiationListener
|
|
byteStreamManager = Socks5BytestreamManager.getBytestreamManager(connection);
|
|
|
|
// get the InitiationListener from Socks5ByteStreamManager
|
|
initiationListener = Whitebox.getInternalState(byteStreamManager, "initiationListener", InitiationListener.class);
|
|
|
|
// create a SOCKS5 Bytestream initiation packet
|
|
initBytestream = Socks5PacketUtils.createBytestreamInitiation(initiatorJID, targetJID,
|
|
sessionID);
|
|
initBytestream.addStreamHost(proxyJID, proxyAddress, 7777);
|
|
|
|
}
|
|
|
|
/**
|
|
* If no listeners are registered for incoming SOCKS5 Bytestream requests, all request should be
|
|
* rejected with an error.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldRespondWithError() throws Exception {
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// capture reply to the SOCKS5 Bytestream initiation
|
|
ArgumentCaptor<IQ> argument = ArgumentCaptor.forClass(IQ.class);
|
|
verify(connection, timeout(TIMEOUT)).sendStanza(argument.capture());
|
|
|
|
// assert that reply is the correct error packet
|
|
assertEquals(initiatorJID, argument.getValue().getTo());
|
|
assertEquals(IQ.Type.error, argument.getValue().getType());
|
|
assertEquals(StanzaError.Condition.not_acceptable,
|
|
argument.getValue().getError().getCondition());
|
|
|
|
}
|
|
|
|
/**
|
|
* If a listener for all requests is registered it should be notified on incoming requests.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldInvokeListenerForAllRequests() throws Exception {
|
|
|
|
// add listener
|
|
Socks5BytestreamListener listener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(listener);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// assert listener is called once
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(listener, timeout(TIMEOUT)).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert that listener is called for the correct request
|
|
assertEquals(initiatorJID, byteStreamRequest.getValue().getFrom());
|
|
|
|
}
|
|
|
|
/**
|
|
* If a listener for a specific user in registered it should be notified on incoming requests
|
|
* for that user.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldInvokeListenerForUser() throws Exception {
|
|
|
|
// add listener
|
|
Socks5BytestreamListener listener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(listener, initiatorJID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// assert listener is called once
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(listener, timeout(TIMEOUT)).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert that reply is the correct error packet
|
|
assertEquals(initiatorJID, byteStreamRequest.getValue().getFrom());
|
|
|
|
}
|
|
|
|
/**
|
|
* If listener for a specific user is registered it should not be notified on incoming requests
|
|
* from other users.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldNotInvokeListenerForUser() throws Exception {
|
|
|
|
// add listener for request of user "other_initiator"
|
|
Socks5BytestreamListener listener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(listener, JidCreate.from("other_" + initiatorJID));
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// assert listener is not called
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(listener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// capture reply to the SOCKS5 Bytestream initiation
|
|
ArgumentCaptor<IQ> argument = ArgumentCaptor.forClass(IQ.class);
|
|
verify(connection, timeout(TIMEOUT)).sendStanza(argument.capture());
|
|
|
|
// assert that reply is the correct error packet
|
|
assertEquals(initiatorJID, argument.getValue().getTo());
|
|
assertEquals(IQ.Type.error, argument.getValue().getType());
|
|
assertEquals(StanzaError.Condition.not_acceptable,
|
|
argument.getValue().getError().getCondition());
|
|
}
|
|
|
|
/**
|
|
* If a user specific listener and an all requests listener is registered only the user specific
|
|
* listener should be notified.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldNotInvokeAllRequestsListenerIfUserListenerExists() throws Exception {
|
|
|
|
// add listener for all request
|
|
Socks5BytestreamListener allRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(allRequestsListener);
|
|
|
|
// add listener for request of user "initiator"
|
|
Socks5BytestreamListener userRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(userRequestsListener, initiatorJID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// assert user request listener is called once
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(userRequestsListener, timeout(TIMEOUT)).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert all requests listener is not called
|
|
byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(allRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
}
|
|
|
|
/**
|
|
* If a user specific listener and an all requests listener is registered only the all requests
|
|
* listener should be notified on an incoming request for another user.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldInvokeAllRequestsListenerIfUserListenerExists() throws Exception {
|
|
// add listener for all request
|
|
Socks5BytestreamListener allRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(allRequestsListener);
|
|
|
|
// add listener for request of user "other_initiator"
|
|
Socks5BytestreamListener userRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(userRequestsListener, JidCreate.from("other_"
|
|
+ initiatorJID));
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
// assert all requests listener is called
|
|
verify(allRequestsListener, timeout(TIMEOUT)).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert user request listener is not called
|
|
verify(userRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
}
|
|
|
|
/**
|
|
* If a request with a specific session ID should be ignored no listeners should be notified.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldIgnoreSocks5BytestreamRequestOnce() throws Exception {
|
|
|
|
// add listener for all request
|
|
Socks5BytestreamListener allRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(allRequestsListener);
|
|
|
|
// add listener for request of user "initiator"
|
|
Socks5BytestreamListener userRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(userRequestsListener, initiatorJID);
|
|
|
|
// ignore session ID
|
|
byteStreamManager.ignoreBytestreamRequestOnce(sessionID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// assert user request listener is not called
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(userRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert all requests listener is not called
|
|
byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(allRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// run the listener with the initiation packet again
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// assert user request listener is called on the second request with the same session ID
|
|
verify(userRequestsListener, timeout(TIMEOUT)).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert all requests listener is not called
|
|
byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(allRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
}
|
|
|
|
}
|