mirror of
https://github.com/vanitasvitae/Smack.git
synced 2024-09-26 01:29:32 +02:00
c81cd34561
Introducing a clean split between the constant connection configuration parameters, which are now all in ConnectionConfiguration and the dynamic connection state (e.g. hostAddresses) which are now in AbstractXMPPConnection. Also removed all arguments of login() since the username, password, resource and callback handler need now to be configured via ConnectionConfiguration. Also remove documentation/extensions/messageevents.md, as it's already in documentation/legacy
123 lines
4.1 KiB
Java
123 lines
4.1 KiB
Java
/**
|
|
*
|
|
* Copyright 2014 Florian Schmaus
|
|
*
|
|
* 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.smack.tcp;
|
|
|
|
import java.io.IOException;
|
|
import java.io.Writer;
|
|
import java.util.concurrent.BrokenBarrierException;
|
|
import java.util.concurrent.CyclicBarrier;
|
|
|
|
import org.jivesoftware.smack.SmackException.NotConnectedException;
|
|
import org.jivesoftware.smack.packet.Message;
|
|
import org.jivesoftware.smack.tcp.XMPPTCPConnection.PacketWriter;
|
|
import org.junit.Test;
|
|
|
|
import static org.junit.Assert.fail;
|
|
|
|
public class PacketWriterTest {
|
|
volatile boolean shutdown;
|
|
volatile boolean prematureUnblocked;
|
|
|
|
/**
|
|
* Make sure that packet writer does block once the queue reaches
|
|
* {@link PacketWriter#QUEUE_SIZE} and that
|
|
* {@link PacketWriter#sendPacket(org.jivesoftware.smack.tcp.packet.Packet)} does unblock after the
|
|
* interrupt.
|
|
*
|
|
* @throws InterruptedException
|
|
* @throws BrokenBarrierException
|
|
* @throws NotConnectedException
|
|
*/
|
|
@SuppressWarnings("javadoc")
|
|
@Test
|
|
public void shouldBlockAndUnblockTest() throws InterruptedException, BrokenBarrierException, NotConnectedException {
|
|
XMPPTCPConnection connection = new XMPPTCPConnection("user", "pass", "example.org");
|
|
final PacketWriter pw = connection.new PacketWriter();
|
|
connection.packetWriter = pw;
|
|
connection.packetReader = connection.new PacketReader();
|
|
connection.setWriter(new BlockingStringWriter());
|
|
pw.init();
|
|
|
|
for (int i = 0; i < XMPPTCPConnection.PacketWriter.QUEUE_SIZE; i++) {
|
|
pw.sendStreamElement(new Message());
|
|
}
|
|
|
|
final CyclicBarrier barrier = new CyclicBarrier(2);
|
|
shutdown = false;
|
|
prematureUnblocked = false;
|
|
Thread t = new Thread(new Runnable() {
|
|
@Override
|
|
public void run() {
|
|
try {
|
|
barrier.await();
|
|
pw.sendStreamElement(new Message());
|
|
// should only return after the pw was interrupted
|
|
if (!shutdown) {
|
|
prematureUnblocked = true;
|
|
}
|
|
}
|
|
catch (Exception e) {
|
|
}
|
|
try {
|
|
barrier.await();
|
|
}
|
|
catch (InterruptedException | BrokenBarrierException e) {
|
|
}
|
|
}
|
|
});
|
|
t.start();
|
|
// This barrier is not strictly necessary, but may increases the chances that the threat
|
|
// will block before we call shutdown. Otherwise we may get false positives (which is still
|
|
// better then false negatives).
|
|
barrier.await();
|
|
// Not really cool, but may increases the chances for 't' to block in sendPacket.
|
|
Thread.sleep(250);
|
|
|
|
// Set to true for testing purposes, so that shutdown() won't wait packet writer
|
|
pw.shutdownDone.reportSuccess();
|
|
// Shutdown the packetwriter
|
|
pw.shutdown(false);
|
|
shutdown = true;
|
|
barrier.await();
|
|
if (prematureUnblocked) {
|
|
fail("Should not unblock before the thread got shutdown");
|
|
}
|
|
synchronized (t) {
|
|
t.notify();
|
|
}
|
|
}
|
|
|
|
public class BlockingStringWriter extends Writer {
|
|
@Override
|
|
public void write(char[] cbuf, int off, int len) throws IOException {
|
|
try {
|
|
wait();
|
|
}
|
|
catch (InterruptedException e) {
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void flush() throws IOException {
|
|
}
|
|
|
|
@Override
|
|
public void close() throws IOException {
|
|
}
|
|
}
|
|
}
|