package org.xmlBlaster.test.cluster;
import java.util.logging.Logger;
import org.xmlBlaster.test.util.Client;
import org.xmlBlaster.util.Global;
// for client connections:
import org.xmlBlaster.util.*;
import org.xmlBlaster.client.I_Callback;
import org.xmlBlaster.client.key.PublishKey;
import org.xmlBlaster.client.key.EraseKey;
import org.xmlBlaster.client.key.SubscribeKey;
import org.xmlBlaster.client.key.UnSubscribeKey;
import org.xmlBlaster.client.key.UpdateKey;
import org.xmlBlaster.client.qos.PublishQos;
import org.xmlBlaster.client.qos.PublishReturnQos;
import org.xmlBlaster.client.qos.UpdateQos;
import org.xmlBlaster.client.qos.SubscribeQos;
import org.xmlBlaster.client.qos.UnSubscribeQos;
import org.xmlBlaster.client.qos.UnSubscribeReturnQos;
import org.xmlBlaster.client.qos.EraseQos;
import org.xmlBlaster.client.qos.EraseReturnQos;
import org.xmlBlaster.client.I_XmlBlasterAccess;
import org.xmlBlaster.util.MsgUnit;
import junit.framework.*;
/**
* Test publishing a message from bilbo to heron.
* <p />
* <pre>
* java -Djava.compiler= junit.textui.TestRunner -noloading org.xmlBlaster.test.cluster.SubscribeTest
* </pre>
* NOTE: asserts() in update() methods are routed back to server and are not handled
* by the junit testsuite, so we check double (see code).
* @see <a href="http://www.xmlblaster.org/xmlBlaster/doc/requirements/cluster.html" target="others">Cluster requirement</a>
*/
public class SubscribeTest extends TestCase {
private String ME = "SubscribeTest";
private Global glob;
private static Logger log = Logger.getLogger(SubscribeTest.class.getName());
private ServerHelper serverHelper;
private I_XmlBlasterAccess heronCon, avalonCon, golanCon, frodoCon, bilboCon, bilboCon2;
private int updateCounterBilbo = 0;
private int updateCounterBilbo2 = 0;
private String oid = "SubscribeToBilbo";
private String domain = "RUGBY_NEWS"; // heron is master for RUGBY_NEWS
private String contentStr = "We win";
public SubscribeTest(String name) {
super(name);
this.glob = new Global(null, true, false);
}
/**
* Initialize the test ...
*/
protected void setUp() {
log.info("Entering setUp(), test starts");
updateCounterBilbo = 0;
updateCounterBilbo2 = 0;
serverHelper = new ServerHelper(glob, log, ME);
// Starts a cluster node
serverHelper.startHeron();
serverHelper.startAvalon();
//serverHelper.startGolan();
serverHelper.startFrodo();
serverHelper.startBilbo();
}
/**
* cleaning up ...
*/
protected void tearDown() {
log.info("Entering tearDown(), test is finished");
try { Thread.sleep(1000); } catch( InterruptedException i) {} // Wait some time
if (bilboCon != null) { bilboCon.disconnect(null); bilboCon = null; }
if (bilboCon2 != null) { bilboCon2.disconnect(null); bilboCon2 = null; }
if (frodoCon != null) { frodoCon.disconnect(null); frodoCon = null; }
if (golanCon != null) { golanCon.disconnect(null); golanCon = null; }
if (avalonCon != null) { avalonCon.disconnect(null); avalonCon = null; }
if (heronCon != null) { heronCon.disconnect(null); heronCon = null; }
serverHelper.tearDown();
}
/**
* We start all nodes as described in requirement
* <a href="http://www.xmlblaster.org/xmlBlaster/doc/requirements/cluster.html" target="others">cluster</a>
* <p />
* - Subscribe to RUGBY messages from bilbo twice<br />
* - publish RUGBY messages to avalon (heron is the master)<br />
* - Kill bilbo, restart bilbo and check if we still get them
*/
public void testSubscribeTwice() {
System.err.println("***SubscribeTest.testSubscribeTwice: Subscribe a message from a cluster slave ...");
try {
System.err.println("->Connect to avalon ...");
avalonCon = serverHelper.connect(serverHelper.getAvalonGlob(), null);
{
System.err.println("->Connect to bilbo ...");
bilboCon = serverHelper.connect(serverHelper.getBilboGlob(), new I_Callback() { // Login to xmlBlaster, register for updates
public String update(String cbSessionId, UpdateKey updateKey, byte[] content, UpdateQos updateQos) {
if (updateQos.isErased()) {
log.info("Ignoring erase message");
return "";
}
updateCounterBilbo++;
log.info(
"Receiving update '" + updateKey.getOid() + "' " + updateCounterBilbo + " ...");
assertEquals("Wrong message updated", oid, updateKey.getOid());
return "";
}
});
System.err.println("->Subscribe from bilbo ...");
SubscribeKey sk = new SubscribeKey(glob, oid);
sk.setDomain(domain);
SubscribeQos sq = new SubscribeQos(glob);
bilboCon.subscribe(sk.toXml(), sq.toXml());
}
{
System.err.println("->Connect to bilbo 2 ...");
final Global bilboGlob2 = serverHelper.getBilboGlob().getClone(null);
bilboCon2 = serverHelper.connect(bilboGlob2, new I_Callback() { // Login to xmlBlaster, register for updates
public String update(String cbSessionId, UpdateKey updateKey, byte[] content, UpdateQos updateQos) {
if (updateQos.isErased()) {
log.info("Ignoring erase message");
return "";
}
updateCounterBilbo2++;
log.info(
"Receiving update '" + updateKey.getOid() + "' " + updateCounterBilbo2 + " ...");
assertEquals("#2 Wrong message updated", oid, updateKey.getOid());
return "";
}
});
System.err.println("->Subscribe from bilbo 2 ...");
SubscribeKey sk = new SubscribeKey(glob, oid);
sk.setDomain(domain);
SubscribeQos sq = new SubscribeQos(glob);
bilboCon2.subscribe(sk.toXml(), sq.toXml());
}
// First test subscribe ...
{
System.err.println("->Publish to avalon ...");
PublishKey avalon_pk = new PublishKey(glob, oid, "text/plain", "1.0", domain);
PublishQos avalon_pq = new PublishQos(glob);
MsgUnit avalon_msgUnit = new MsgUnit(avalon_pk, contentStr, avalon_pq);
PublishReturnQos avalon_prq = avalonCon.publish(avalon_msgUnit);
assertEquals("oid changed", oid, avalon_prq.getKeyOid());
try { Thread.sleep(2000); } catch( InterruptedException i) {}
if (1 != updateCounterBilbo) log.severe("Did not expect " + updateCounterBilbo + " updates");
assertEquals("message from avalon", 1, updateCounterBilbo);
if (1 != updateCounterBilbo2) log.severe("Did not expect " + updateCounterBilbo2 + " updates");
assertEquals("message from avalon #2", 1, updateCounterBilbo2);
updateCounterBilbo = 0;
updateCounterBilbo2 = 0;
}
System.err.println("->testSubscribeTwice done, SUCCESS.");
// ... and now test unSubscribe
{
System.err.println("->UnSubscribe from bilbo ...");
UnSubscribeKey usk = new UnSubscribeKey(glob, oid);
usk.setDomain(domain);
UnSubscribeQos usq = new UnSubscribeQos(glob);
UnSubscribeReturnQos[] usrq = bilboCon.unSubscribe(usk, usq);
assertEquals("", 1, usrq.length);
System.err.println("->Publish to avalon ...");
PublishKey avalon_pk = new PublishKey(glob, oid, "text/plain", "1.0", domain);
PublishQos avalon_pq = new PublishQos(glob);
MsgUnit avalon_msgUnit = new MsgUnit(avalon_pk, contentStr, avalon_pq);
PublishReturnQos avalon_prq = avalonCon.publish(avalon_msgUnit);
assertEquals("oid changed", oid, avalon_prq.getKeyOid());
try { Thread.sleep(2000); } catch( InterruptedException i) {}
if (0 != updateCounterBilbo) log.severe("Did not expect " + updateCounterBilbo + " updates");
assertEquals("message from avalon", 0, updateCounterBilbo);
if (1 != updateCounterBilbo2) log.severe("Did not expect " + updateCounterBilbo2 + " updates");
assertEquals("message from avalon #2", 1, updateCounterBilbo2);
updateCounterBilbo = 0;
updateCounterBilbo2 = 0;
}
System.err.println("->Trying to erase the message at the slave node ...");
EraseKey ek = new EraseKey(glob, oid);
ek.setDomain(domain);
EraseQos eq = new EraseQos(glob);
EraseReturnQos[] arr = avalonCon.erase(ek.toXml(), eq.toXml());
assertEquals("Erase", 1, arr.length);
}
catch (XmlBlasterException e) {
e.printStackTrace();
fail("SubscribeToBilbo-Exception: " + e.toString());
}
finally {
if (bilboCon != null) {
bilboCon.disconnect(null);
bilboCon = null;
}
if (bilboCon2 != null) {
bilboCon2.disconnect(null);
bilboCon2 = null;
}
if (avalonCon != null) {
avalonCon.disconnect(null);
avalonCon = null;
}
}
System.err.println("***SubscribeTest.testSubscribeTwice: testSubscribeTwice [SUCCESS]");
}
/**
* We start all nodes as described in requirement
* <a href="http://www.xmlblaster.org/xmlBlaster/doc/requirements/cluster.html" target="others">cluster</a>
* <p />
* 1. publish RUGBY messages to avalon (heron is the master)<br />
* 2. Subscribe those messages from bilbo<br />
* 3. Kill bilbo, restart bilbo and check if we still get them
*/
public void testSubscribe() {
System.err.println("***SubscribeTest: Subscribe a message from a cluster slave ...");
int num = 2;
I_XmlBlasterAccess[] bilboCons = new I_XmlBlasterAccess[num];
try {
System.err.println("->Connect to avalon ...");
avalonCon = serverHelper.connect(serverHelper.getAvalonGlob(), null);
try { Thread.sleep(1000); } catch( InterruptedException i) {} // Wait some time
for (int ii=0; ii<num; ii++) {
System.err.println("->Connect to bilbo #" + ii + " ...");
final Global bilboGlobii = serverHelper.getBilboGlob().getClone(null);
bilboCons[ii] = serverHelper.connect(bilboGlobii, new I_Callback() { // Login to xmlBlaster, register for updates
public String update(String cbSessionId, UpdateKey updateKey, byte[] content, UpdateQos updateQos) {
log.info(
"Receiving update '" + updateKey.getOid() + "' state=" + updateQos.getState() + ", " + updateCounterBilbo + " ...");
if (updateQos.isErased()) {
log.info("Ignoring erase message");
return "";
}
updateCounterBilbo++;
log.info(
"Receiving update '" + updateKey.getOid() + "' " + updateCounterBilbo + " ...");
assertEquals("Wrong message updated", oid, updateKey.getOid());
return "";
}
});
System.err.println("->Publish to avalon #" + ii + " ...");
PublishKey avalon_pk = new PublishKey(glob, oid, "text/plain", "1.0", domain);
PublishQos avalon_pq = new PublishQos(glob);
MsgUnit avalon_msgUnit = new MsgUnit(avalon_pk, contentStr, avalon_pq);
PublishReturnQos avalon_prq = avalonCon.publish(avalon_msgUnit);
assertEquals("oid changed", oid, avalon_prq.getKeyOid());
try { Thread.sleep(1000L); } catch( InterruptedException i) {}
System.err.println("->Subscribe from bilbo #" + ii + ", the message from avalon should arrive ...");
SubscribeKey sk = new SubscribeKey(glob, oid);
sk.setDomain(domain);
SubscribeQos sq = new SubscribeQos(glob);
bilboCons[ii].subscribe(sk.toXml(), sq.toXml());
waitOnUpdate(2000L, 1);
try { Thread.sleep(1000); } catch( InterruptedException i) {} // wait longer to check if too many arrive
if (1 != updateCounterBilbo) log.severe("Did not expect " + updateCounterBilbo + " updates");
assertEquals("message from avalon", 1, updateCounterBilbo);
updateCounterBilbo = 0;
System.err.println("->Trying to erase the message at the slave node ...");
EraseKey ek = new EraseKey(glob, oid);
ek.setDomain(domain);
EraseQos eq = new EraseQos(glob);
EraseReturnQos[] arr = avalonCon.erase(ek.toXml(), eq.toXml());
assertEquals("Erase", 1, arr.length);
// Wait on erase events
try { Thread.sleep(1000); } catch( InterruptedException i) {}
updateCounterBilbo = 0;
updateCounterBilbo2 = 0;
// We stay logged in but kill over callback server ...
Client.shutdownCb(bilboCons[ii], Client.Shutdown.KEEP_LOGGED_IN);
}
System.err.println("->testSubscribe done, SUCCESS.");
}
catch (XmlBlasterException e) {
e.printStackTrace();
fail("SubscribeToBilbo-Exception: " + e.toString());
}
finally {
for (int jj=0; jj<bilboCons.length; jj++) {
if (bilboCons[jj] != null) {
bilboCons[jj].disconnect(null);
bilboCons[jj] = null;
}
}
if (avalonCon != null) {
avalonCon.disconnect(null);
avalonCon = null;
}
}
System.err.println("***SubscribeTest: testSubscribe [SUCCESS]");
}
private void waitOnUpdate(final long timeout, final int numWait) {
long pollingInterval = 50L; // check every 0.05 seconds
if (timeout < 50) pollingInterval = timeout / 10L;
long sum = 0L;
while (updateCounterBilbo < numWait) {
try {
Thread.sleep(pollingInterval);
}
catch( InterruptedException i)
{}
sum += pollingInterval;
if (sum > timeout) {
log.warning("Timeout of " + timeout + " occurred");
break;
}
}
}
}