1
0
Fork 0
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:
Thiago Camargo 2007-02-16 01:59:01 +00:00 committed by thiago
parent e155c302ad
commit 42d1ecf83b

View file

@ -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;
} }