mirror of
https://github.com/vanitasvitae/Smack.git
synced 2024-11-29 23:42:06 +01:00
PacketFilter Remotion Test Case
git-svn-id: http://svn.igniterealtime.org/svn/repos/smack/trunk@7169 b35dd754-fafc-0310-a699-88a17e54d16e
This commit is contained in:
parent
e155c302ad
commit
42d1ecf83b
1 changed files with 166 additions and 68 deletions
|
@ -1,60 +1,61 @@
|
||||||
/**
|
/**
|
||||||
* $RCSfile$
|
* $RCSfile$
|
||||||
* $Revision$
|
* $Revision$
|
||||||
* $Date$
|
* $Date$
|
||||||
*
|
*
|
||||||
* Copyright (C) 2002-2003 Jive Software. All rights reserved.
|
* Copyright (C) 2002-2003 Jive Software. All rights reserved.
|
||||||
* ====================================================================
|
* ====================================================================
|
||||||
* The Jive Software License (based on Apache Software License, Version 1.1)
|
* The Jive Software License (based on Apache Software License, Version 1.1)
|
||||||
*
|
*
|
||||||
* Redistribution and use in source and binary forms, with or without
|
* Redistribution and use in source and binary forms, with or without
|
||||||
* modification, are permitted provided that the following conditions
|
* modification, are permitted provided that the following conditions
|
||||||
* are met:
|
* are met:
|
||||||
*
|
*
|
||||||
* 1. Redistributions of source code must retain the above copyright
|
* 1. Redistributions of source code must retain the above copyright
|
||||||
* notice, this list of conditions and the following disclaimer.
|
* notice, this list of conditions and the following disclaimer.
|
||||||
*
|
*
|
||||||
* 2. Redistributions in binary form must reproduce the above copyright
|
* 2. Redistributions in binary form must reproduce the above copyright
|
||||||
* notice, this list of conditions and the following disclaimer in
|
* notice, this list of conditions and the following disclaimer in
|
||||||
* the documentation and/or other materials provided with the
|
* the documentation and/or other materials provided with the
|
||||||
* distribution.
|
* distribution.
|
||||||
*
|
*
|
||||||
* 3. The end-user documentation included with the redistribution,
|
* 3. The end-user documentation included with the redistribution,
|
||||||
* if any, must include the following acknowledgment:
|
* if any, must include the following acknowledgment:
|
||||||
* "This product includes software developed by
|
* "This product includes software developed by
|
||||||
* Jive Software (http://www.jivesoftware.com)."
|
* Jive Software (http://www.jivesoftware.com)."
|
||||||
* Alternately, this acknowledgment may appear in the software itself,
|
* Alternately, this acknowledgment may appear in the software itself,
|
||||||
* if and wherever such third-party acknowledgments normally appear.
|
* if and wherever such third-party acknowledgments normally appear.
|
||||||
*
|
*
|
||||||
* 4. The names "Smack" and "Jive Software" must not be used to
|
* 4. The names "Smack" and "Jive Software" must not be used to
|
||||||
* endorse or promote products derived from this software without
|
* endorse or promote products derived from this software without
|
||||||
* prior written permission. For written permission, please
|
* prior written permission. For written permission, please
|
||||||
* contact webmaster@jivesoftware.com.
|
* contact webmaster@jivesoftware.com.
|
||||||
*
|
*
|
||||||
* 5. Products derived from this software may not be called "Smack",
|
* 5. Products derived from this software may not be called "Smack",
|
||||||
* nor may "Smack" appear in their name, without prior written
|
* nor may "Smack" appear in their name, without prior written
|
||||||
* permission of Jive Software.
|
* permission of Jive Software.
|
||||||
*
|
*
|
||||||
* THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
|
* THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
|
||||||
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
||||||
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
* DISCLAIMED. IN NO EVENT SHALL JIVE SOFTWARE OR
|
* DISCLAIMED. IN NO EVENT SHALL JIVE SOFTWARE OR
|
||||||
* ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
* ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
|
||||||
* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||||||
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||||
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
|
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
|
||||||
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
||||||
* SUCH DAMAGE.
|
* SUCH DAMAGE.
|
||||||
* ====================================================================
|
* ====================================================================
|
||||||
*/
|
*/
|
||||||
|
|
||||||
package org.jivesoftware.smack;
|
package org.jivesoftware.smack;
|
||||||
|
|
||||||
import org.jivesoftware.smack.filter.FromMatchesFilter;
|
import org.jivesoftware.smack.filter.FromMatchesFilter;
|
||||||
import org.jivesoftware.smack.filter.PacketIDFilter;
|
import org.jivesoftware.smack.filter.PacketIDFilter;
|
||||||
import org.jivesoftware.smack.filter.PacketTypeFilter;
|
import org.jivesoftware.smack.filter.PacketTypeFilter;
|
||||||
|
import org.jivesoftware.smack.filter.PacketFilter;
|
||||||
import org.jivesoftware.smack.packet.*;
|
import org.jivesoftware.smack.packet.*;
|
||||||
import org.jivesoftware.smack.test.SmackTestCase;
|
import org.jivesoftware.smack.test.SmackTestCase;
|
||||||
|
|
||||||
|
@ -63,14 +64,42 @@ import java.util.Date;
|
||||||
|
|
||||||
public class PacketReaderTest extends SmackTestCase {
|
public class PacketReaderTest extends SmackTestCase {
|
||||||
|
|
||||||
|
private int counter;
|
||||||
|
|
||||||
|
private final Object mutex = new Object();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Constructor for PacketReaderTest.
|
* Constructor for PacketReaderTest.
|
||||||
|
*
|
||||||
* @param arg0
|
* @param arg0
|
||||||
*/
|
*/
|
||||||
public PacketReaderTest(String arg0) {
|
public PacketReaderTest(String arg0) {
|
||||||
super(arg0);
|
super(arg0);
|
||||||
|
resetCounter();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Counter management
|
||||||
|
|
||||||
|
private void resetCounter() {
|
||||||
|
synchronized (mutex) {
|
||||||
|
counter = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void incCounter() {
|
||||||
|
synchronized (mutex) {
|
||||||
|
counter++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private int valCounter() {
|
||||||
|
int val;
|
||||||
|
synchronized (mutex) {
|
||||||
|
val = counter;
|
||||||
|
}
|
||||||
|
return val;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Verify that when Smack receives a "not implemented IQ" answers with an IQ packet
|
* Verify that when Smack receives a "not implemented IQ" answers with an IQ packet
|
||||||
* with error code 501.
|
* with error code 501.
|
||||||
|
@ -91,7 +120,7 @@ public class PacketReaderTest extends SmackTestCase {
|
||||||
PacketCollector collector = getConnection(0).createPacketCollector(
|
PacketCollector collector = getConnection(0).createPacketCollector(
|
||||||
new PacketIDFilter(iqPacket.getPacketID()));
|
new PacketIDFilter(iqPacket.getPacketID()));
|
||||||
getConnection(0).sendPacket(iqPacket);
|
getConnection(0).sendPacket(iqPacket);
|
||||||
IQ response = (IQ)collector.nextResult(SmackConfiguration.getPacketReplyTimeout());
|
IQ response = (IQ) collector.nextResult(SmackConfiguration.getPacketReplyTimeout());
|
||||||
if (response == null) {
|
if (response == null) {
|
||||||
fail("No response from the other user.");
|
fail("No response from the other user.");
|
||||||
}
|
}
|
||||||
|
@ -104,23 +133,29 @@ public class PacketReaderTest extends SmackTestCase {
|
||||||
* Tests that PacketReader adds new listeners and also removes them correctly.
|
* Tests that PacketReader adds new listeners and also removes them correctly.
|
||||||
*/
|
*/
|
||||||
public void testRemoveListener() {
|
public void testRemoveListener() {
|
||||||
PacketListener listener = new PacketListener() {
|
|
||||||
public void processPacket(Packet packet) {
|
PacketListener listener = new PacketListener() {
|
||||||
//Ignore
|
public void processPacket(Packet packet) {
|
||||||
}
|
// Do nothing
|
||||||
};
|
}
|
||||||
// Keep number of current listeners
|
};
|
||||||
int listenersSize = getConnection(0).packetReader.listeners.size();
|
// Keep number of current listeners
|
||||||
// Add a new listener
|
int listenersSize = getConnection(0).packetReader.listeners.size();
|
||||||
getConnection(0).addPacketListener(listener, new MockPacketFilter(true));
|
// Add a new listener
|
||||||
// Check that the listener was added
|
getConnection(0).addPacketListener(listener, new MockPacketFilter(true));
|
||||||
assertEquals("Listener was not added", listenersSize + 1,
|
// Check that the listener was added
|
||||||
getConnection(0).packetReader.listeners.size());
|
assertEquals("Listener was not added", listenersSize + 1,
|
||||||
// Remove the listener
|
getConnection(0).packetReader.listeners.size());
|
||||||
getConnection(0).removePacketListener(listener);
|
|
||||||
// Check that the number of listeners is correct (i.e. the listener was removed)
|
Message msg = new Message(getConnection(0).getUser(), Message.Type.normal);
|
||||||
assertEquals("Listener was not removed", listenersSize,
|
|
||||||
getConnection(0).packetReader.listeners.size());
|
getConnection(1).sendPacket(msg);
|
||||||
|
|
||||||
|
// Remove the listener
|
||||||
|
getConnection(0).removePacketListener(listener);
|
||||||
|
// Check that the number of listeners is correct (i.e. the listener was removed)
|
||||||
|
assertEquals("Listener was not removed", listenersSize,
|
||||||
|
getConnection(0).packetReader.listeners.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -164,6 +199,69 @@ public class PacketReaderTest extends SmackTestCase {
|
||||||
assertNotNull("No message was received", collector.nextResult(1000));
|
assertNotNull("No message was received", collector.nextResult(1000));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Tests that PacketReader adds new listeners and also removes them correctly.
|
||||||
|
*/
|
||||||
|
public void testFiltersRemotion() {
|
||||||
|
|
||||||
|
resetCounter();
|
||||||
|
|
||||||
|
int repeat=10;
|
||||||
|
|
||||||
|
for (int j = 0; j < repeat; j++) {
|
||||||
|
|
||||||
|
PacketListener listener = new PacketListener() {
|
||||||
|
public void processPacket(Packet packet) {
|
||||||
|
System.out.println("Packet Captured");
|
||||||
|
incCounter();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
PacketFilter pf = new PacketFilter() {
|
||||||
|
public boolean accept(Packet packet) {
|
||||||
|
System.out.println("Packet Filtered");
|
||||||
|
incCounter();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
getConnection(0).addPacketListener(listener,pf);
|
||||||
|
// Check that the listener was added
|
||||||
|
|
||||||
|
Message msg = new Message(getConnection(0).getUser(), Message.Type.normal);
|
||||||
|
|
||||||
|
getConnection(1).sendPacket(msg);
|
||||||
|
|
||||||
|
try {
|
||||||
|
Thread.sleep(100);
|
||||||
|
}
|
||||||
|
catch (InterruptedException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove the listener
|
||||||
|
getConnection(0).removePacketListener(listener);
|
||||||
|
|
||||||
|
try {
|
||||||
|
Thread.sleep(200);
|
||||||
|
}
|
||||||
|
catch (InterruptedException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 0; i < 10; i++)
|
||||||
|
getConnection(1).sendPacket(msg);
|
||||||
|
|
||||||
|
try {
|
||||||
|
Thread.sleep(100);
|
||||||
|
}
|
||||||
|
catch (InterruptedException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
System.out.println(valCounter());
|
||||||
|
assertEquals(valCounter(),repeat*2);
|
||||||
|
}
|
||||||
|
|
||||||
protected int getMaxConnections() {
|
protected int getMaxConnections() {
|
||||||
return 2;
|
return 2;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue