339 lines
13 KiB
Java
339 lines
13 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.ibb;
|
|
|
|
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.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.InitExtensions;
|
|
import org.jivesoftware.smackx.bytestreams.BytestreamRequest;
|
|
import org.jivesoftware.smackx.bytestreams.ibb.packet.Open;
|
|
|
|
import org.junit.jupiter.api.BeforeEach;
|
|
import org.junit.jupiter.api.Test;
|
|
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 extends InitExtensions {
|
|
|
|
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 String sessionID = "session_id";
|
|
|
|
private XMPPConnection connection;
|
|
private InBandBytestreamManager byteStreamManager;
|
|
private InitiationListener initiationListener;
|
|
private Open initBytestream;
|
|
|
|
/**
|
|
* Initialize fields used in the tests.
|
|
*/
|
|
@BeforeEach
|
|
public void setup() {
|
|
|
|
// mock connection
|
|
connection = mock(XMPPConnection.class);
|
|
|
|
// initialize InBandBytestreamManager to get the InitiationListener
|
|
byteStreamManager = InBandBytestreamManager.getByteStreamManager(connection);
|
|
|
|
// get the InitiationListener from InBandByteStreamManager
|
|
initiationListener = Whitebox.getInternalState(byteStreamManager, "initiationListener", InitiationListener.class);
|
|
|
|
// create a In-Band Bytestream open packet
|
|
initBytestream = new Open(sessionID, 4096);
|
|
initBytestream.setFrom(initiatorJID);
|
|
initBytestream.setTo(targetJID);
|
|
|
|
}
|
|
|
|
/**
|
|
* If no listeners are registered for incoming In-Band 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);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// capture reply to the In-Band Bytestream open request
|
|
ArgumentCaptor<IQ> argument = ArgumentCaptor.forClass(IQ.class);
|
|
verify(connection).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());
|
|
|
|
}
|
|
|
|
/**
|
|
* Open request with a block size that exceeds the maximum block size should be replied with an
|
|
* resource-constraint error.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldRejectRequestWithTooBigBlockSize() throws Exception {
|
|
byteStreamManager.setMaximumBlockSize(1024);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// capture reply to the In-Band Bytestream open request
|
|
ArgumentCaptor<IQ> argument = ArgumentCaptor.forClass(IQ.class);
|
|
verify(connection).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.resource_constraint,
|
|
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
|
|
InBandBytestreamListener listener = mock(InBandBytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(listener);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert listener is called once
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(listener).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
|
|
InBandBytestreamListener listener = mock(InBandBytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(listener, initiatorJID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert listener is called once
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(listener).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"
|
|
InBandBytestreamListener listener = mock(InBandBytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(listener, JidCreate.from("other_" + initiatorJID));
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert listener is not called
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(listener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// capture reply to the In-Band Bytestream open request
|
|
ArgumentCaptor<IQ> argument = ArgumentCaptor.forClass(IQ.class);
|
|
verify(connection).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
|
|
InBandBytestreamListener allRequestsListener = mock(InBandBytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(allRequestsListener);
|
|
|
|
// add listener for request of user "initiator"
|
|
InBandBytestreamListener userRequestsListener = mock(InBandBytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(userRequestsListener, initiatorJID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert user request listener is called once
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(userRequestsListener).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
|
|
InBandBytestreamListener allRequestsListener = mock(InBandBytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(allRequestsListener);
|
|
|
|
// add listener for request of user "other_initiator"
|
|
InBandBytestreamListener userRequestsListener = mock(InBandBytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(userRequestsListener, JidCreate.from("other_"
|
|
+ initiatorJID));
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// 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 called
|
|
byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(allRequestsListener).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 shouldIgnoreInBandBytestreamRequestOnce() throws Exception {
|
|
|
|
// add listener for all request
|
|
InBandBytestreamListener allRequestsListener = mock(InBandBytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(allRequestsListener);
|
|
|
|
// add listener for request of user "initiator"
|
|
InBandBytestreamListener userRequestsListener = mock(InBandBytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(userRequestsListener, initiatorJID);
|
|
|
|
// ignore session ID
|
|
byteStreamManager.ignoreBytestreamRequestOnce(sessionID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// 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);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert user request listener is called on the second request with the
|
|
// same session ID
|
|
verify(userRequestsListener).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert all requests listener is not called
|
|
byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(allRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
}
|
|
|
|
}
|