2014-02-17 23:58:40 +01:00
|
|
|
/**
|
|
|
|
*
|
|
|
|
* 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.smack;
|
|
|
|
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
import static org.junit.Assert.assertNull;
|
|
|
|
|
2015-02-26 18:41:17 +01:00
|
|
|
import org.jivesoftware.smack.filter.StanzaFilter;
|
2015-02-05 11:17:27 +01:00
|
|
|
import org.jivesoftware.smack.packet.Stanza;
|
2014-02-17 23:58:40 +01:00
|
|
|
import org.junit.Test;
|
|
|
|
|
|
|
|
public class PacketCollectorTest
|
|
|
|
{
|
|
|
|
|
|
|
|
@Test
|
2015-02-14 09:43:44 +01:00
|
|
|
public void verifyRollover() throws InterruptedException
|
2014-02-17 23:58:40 +01:00
|
|
|
{
|
|
|
|
TestPacketCollector collector = new TestPacketCollector(null, new OKEverything(), 5);
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
for (int i=0; i<6; i++)
|
|
|
|
{
|
2015-02-05 11:17:27 +01:00
|
|
|
Stanza testPacket = new TestPacket(i);
|
2014-02-17 23:58:40 +01:00
|
|
|
collector.processPacket(testPacket);
|
|
|
|
}
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
// Assert that '0' has rolled off
|
2015-02-05 11:17:27 +01:00
|
|
|
assertEquals("1", collector.nextResultBlockForever().getStanzaId());
|
|
|
|
assertEquals("2", collector.nextResultBlockForever().getStanzaId());
|
|
|
|
assertEquals("3", collector.nextResultBlockForever().getStanzaId());
|
|
|
|
assertEquals("4", collector.nextResultBlockForever().getStanzaId());
|
|
|
|
assertEquals("5", collector.pollResult().getStanzaId());
|
2014-02-17 23:58:40 +01:00
|
|
|
assertNull(collector.pollResult());
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
for (int i=10; i<15; i++)
|
|
|
|
{
|
2015-02-05 11:17:27 +01:00
|
|
|
Stanza testPacket = new TestPacket(i);
|
2014-02-17 23:58:40 +01:00
|
|
|
collector.processPacket(testPacket);
|
|
|
|
}
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2015-02-05 11:17:27 +01:00
|
|
|
assertEquals("10", collector.nextResultBlockForever().getStanzaId());
|
|
|
|
assertEquals("11", collector.nextResultBlockForever().getStanzaId());
|
|
|
|
assertEquals("12", collector.nextResultBlockForever().getStanzaId());
|
|
|
|
assertEquals("13", collector.nextResultBlockForever().getStanzaId());
|
|
|
|
assertEquals("14", collector.pollResult().getStanzaId());
|
2014-02-17 23:58:40 +01:00
|
|
|
assertNull(collector.pollResult());
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
assertNull(collector.nextResult(1000));
|
|
|
|
}
|
|
|
|
|
2015-03-18 21:51:25 +01:00
|
|
|
/**
|
|
|
|
* Although this doesn't guarentee anything due to the nature of threading, it can potentially
|
|
|
|
* catch problems.
|
|
|
|
*/
|
2014-02-17 23:58:40 +01:00
|
|
|
@Test
|
|
|
|
public void verifyThreadSafety()
|
|
|
|
{
|
|
|
|
int insertCount = 500;
|
|
|
|
final TestPacketCollector collector = new TestPacketCollector(null, new OKEverything(), insertCount);
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
Thread consumer1 = new Thread(new Runnable()
|
|
|
|
{
|
|
|
|
@Override
|
|
|
|
public void run()
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
while (true)
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
Thread.sleep(3);
|
|
|
|
}
|
|
|
|
catch (InterruptedException e)
|
|
|
|
{
|
|
|
|
}
|
2014-02-18 15:05:19 +01:00
|
|
|
@SuppressWarnings("unused")
|
2015-02-05 11:17:27 +01:00
|
|
|
Stanza packet = collector.nextResultBlockForever();
|
2014-02-17 23:58:40 +01:00
|
|
|
// System.out.println(Thread.currentThread().getName() + " packet: " + packet);
|
|
|
|
}
|
|
|
|
}
|
2015-02-14 09:43:44 +01:00
|
|
|
catch (InterruptedException e) {
|
|
|
|
throw new RuntimeException(e);
|
|
|
|
}
|
2014-02-17 23:58:40 +01:00
|
|
|
}
|
|
|
|
});
|
|
|
|
consumer1.setName("consumer 1");
|
|
|
|
|
|
|
|
Thread consumer2 = new Thread(new Runnable()
|
|
|
|
{
|
|
|
|
@Override
|
|
|
|
public void run()
|
|
|
|
{
|
2015-02-05 11:17:27 +01:00
|
|
|
Stanza p = null;
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
do
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
Thread.sleep(3);
|
|
|
|
}
|
|
|
|
catch (InterruptedException e)
|
|
|
|
{
|
|
|
|
}
|
2015-02-14 09:43:44 +01:00
|
|
|
try {
|
|
|
|
p = collector.nextResult(1);
|
|
|
|
}
|
|
|
|
catch (InterruptedException e) {
|
|
|
|
throw new RuntimeException(e);
|
|
|
|
}
|
2014-02-17 23:58:40 +01:00
|
|
|
// System.out.println(Thread.currentThread().getName() + " packet: " + p);
|
|
|
|
}
|
|
|
|
while (p != null);
|
|
|
|
}
|
|
|
|
});
|
|
|
|
consumer2.setName("consumer 2");
|
|
|
|
|
|
|
|
Thread consumer3 = new Thread(new Runnable()
|
|
|
|
{
|
|
|
|
@Override
|
|
|
|
public void run()
|
|
|
|
{
|
2015-02-05 11:17:27 +01:00
|
|
|
Stanza p = null;
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
do
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
Thread.sleep(3);
|
|
|
|
}
|
|
|
|
catch (InterruptedException e)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
p = collector.pollResult();
|
|
|
|
// System.out.println(Thread.currentThread().getName() + " packet: " + p);
|
|
|
|
}
|
|
|
|
while (p != null);
|
|
|
|
}
|
|
|
|
});
|
|
|
|
consumer3.setName("consumer 3");
|
|
|
|
|
|
|
|
consumer1.start();
|
|
|
|
consumer2.start();
|
|
|
|
consumer3.start();
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
for(int i=0; i<insertCount; i++)
|
|
|
|
{
|
|
|
|
collector.processPacket(new TestPacket(i));
|
|
|
|
}
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
Thread.sleep(5000);
|
|
|
|
consumer3.join();
|
|
|
|
consumer2.join();
|
|
|
|
consumer1.interrupt();
|
|
|
|
}
|
|
|
|
catch (InterruptedException e)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
//We cannot guarantee that this is going to pass due to the possible issue of timing between consumer 1
|
|
|
|
// and main, but the probability is extremely remote.
|
|
|
|
assertNull(collector.pollResult());
|
|
|
|
}
|
|
|
|
|
2015-02-26 18:41:17 +01:00
|
|
|
class OKEverything implements StanzaFilter
|
2014-02-17 23:58:40 +01:00
|
|
|
{
|
|
|
|
@Override
|
2015-02-05 11:17:27 +01:00
|
|
|
public boolean accept(Stanza packet)
|
2014-02-17 23:58:40 +01:00
|
|
|
{
|
|
|
|
return true;
|
|
|
|
}
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
}
|
2015-03-17 11:33:02 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
class TestPacketCollector extends PacketCollector
|
|
|
|
{
|
2015-02-26 18:41:17 +01:00
|
|
|
protected TestPacketCollector(XMPPConnection conection, StanzaFilter packetFilter, int size)
|
2014-02-17 23:58:40 +01:00
|
|
|
{
|
2015-02-26 18:41:17 +01:00
|
|
|
super(conection, PacketCollector.newConfiguration().setStanzaFilter(packetFilter).setSize(size));
|
2014-02-17 23:58:40 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-02-05 11:17:27 +01:00
|
|
|
class TestPacket extends Stanza
|
2014-02-17 23:58:40 +01:00
|
|
|
{
|
|
|
|
public TestPacket(int i)
|
|
|
|
{
|
2015-02-05 11:17:27 +01:00
|
|
|
setStanzaId(String.valueOf(i));
|
2014-02-17 23:58:40 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public String toXML()
|
|
|
|
{
|
2015-02-05 11:17:27 +01:00
|
|
|
return "<packetId>" + getStanzaId() + "</packetId>";
|
2014-02-17 23:58:40 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|