mirror of
https://github.com/vanitasvitae/Smack.git
synced 2025-09-10 09:39:39 +02:00
Add support for XEP-0198: Stream Management
- De-duplicate code by moving it into AbstractXMPPConnection - Introduce TopLevelStreamElement as superclass for all XMPP stream elements. - Add SynchronizationPoint, ParserUtils - Add ParserUtils Fixes SMACK-333 and SMACK-521
This commit is contained in:
parent
07c10a7444
commit
fc51f3df48
69 changed files with 3277 additions and 1083 deletions
|
@ -40,19 +40,22 @@ public class PacketWriterTest {
|
|||
*
|
||||
* @throws InterruptedException
|
||||
* @throws BrokenBarrierException
|
||||
* @throws NotConnectedException
|
||||
*/
|
||||
@SuppressWarnings("javadoc")
|
||||
@Test
|
||||
public void shouldBlockAndUnblockTest() throws InterruptedException, BrokenBarrierException, NotConnectedException {
|
||||
XMPPTCPConnection connection = new XMPPTCPConnection("foobar.com");
|
||||
final PacketWriter pw = connection.new PacketWriter();
|
||||
pw.setWriter(new BlockingStringWriter());
|
||||
pw.startup();
|
||||
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.sendPacket(new Message());
|
||||
pw.sendStreamElement(new Message());
|
||||
}
|
||||
|
||||
|
||||
final CyclicBarrier barrier = new CyclicBarrier(2);
|
||||
shutdown = false;
|
||||
prematureUnblocked = false;
|
||||
|
@ -61,7 +64,7 @@ public class PacketWriterTest {
|
|||
public void run() {
|
||||
try {
|
||||
barrier.await();
|
||||
pw.sendPacket(new Message());
|
||||
pw.sendStreamElement(new Message());
|
||||
// should only return after the pw was interrupted
|
||||
if (!shutdown) {
|
||||
prematureUnblocked = true;
|
||||
|
@ -85,9 +88,9 @@ public class PacketWriterTest {
|
|||
Thread.sleep(250);
|
||||
|
||||
// Set to true for testing purposes, so that shutdown() won't wait packet writer
|
||||
pw.shutdownDone.set(true);
|
||||
pw.shutdownDone.reportSuccess();
|
||||
// Shutdown the packetwriter
|
||||
pw.shutdown();
|
||||
pw.shutdown(false);
|
||||
shutdown = true;
|
||||
barrier.await();
|
||||
if (prematureUnblocked) {
|
||||
|
|
|
@ -0,0 +1,154 @@
|
|||
/**
|
||||
*
|
||||
* Copyright 2014 Vyacheslav Blinov
|
||||
*
|
||||
* 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.sm.provider;
|
||||
|
||||
import com.jamesmurty.utils.XMLBuilder;
|
||||
import org.jivesoftware.smack.packet.XMPPError;
|
||||
import org.jivesoftware.smack.tcp.sm.packet.StreamManagement;
|
||||
import org.jivesoftware.smack.util.PacketParserUtils;
|
||||
import org.junit.Test;
|
||||
import org.xmlpull.v1.XmlPullParser;
|
||||
import org.xmlpull.v1.XmlPullParserException;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.Properties;
|
||||
|
||||
import static org.hamcrest.CoreMatchers.equalTo;
|
||||
import static org.hamcrest.CoreMatchers.is;
|
||||
import static org.hamcrest.CoreMatchers.notNullValue;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertThat;
|
||||
|
||||
public class ParseStreamManagementTest {
|
||||
private static final Properties outputProperties = initOutputProperties();
|
||||
|
||||
@Test
|
||||
public void testParseEnabled() throws Exception {
|
||||
String stanzaID = "zid615d9";
|
||||
boolean resume = true;
|
||||
String location = "test";
|
||||
int max = 42;
|
||||
|
||||
String enabledStanza = XMLBuilder.create("enabled")
|
||||
.a("xmlns", "urn:xmpp:sm:3")
|
||||
.a("id", "zid615d9")
|
||||
.a("resume", String.valueOf(resume))
|
||||
.a("location", location)
|
||||
.a("max", String.valueOf(max))
|
||||
.asString(outputProperties);
|
||||
|
||||
StreamManagement.Enabled enabledPacket = ParseStreamManagement.enabled(
|
||||
PacketParserUtils.getParserFor(enabledStanza));
|
||||
|
||||
assertThat(enabledPacket, is(notNullValue()));
|
||||
assertThat(enabledPacket.getId(), equalTo(stanzaID));
|
||||
assertThat(enabledPacket.getLocation(), equalTo(location));
|
||||
assertThat(enabledPacket.isResumeSet(), equalTo(resume));
|
||||
assertThat(enabledPacket.getMaxResumptionTime(), equalTo(max));
|
||||
}
|
||||
|
||||
|
||||
@Test
|
||||
public void testParseEnabledInvariant() throws XmlPullParserException, IOException {
|
||||
String enabledString = (new StreamManagement.Enabled("stream-id", false)).toXML().toString();
|
||||
XmlPullParser parser = PacketParserUtils.getParserFor(enabledString);
|
||||
StreamManagement.Enabled enabled = ParseStreamManagement.enabled(parser);
|
||||
|
||||
assertEquals(enabledString, enabled.toXML().toString());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testParseFailed() throws Exception {
|
||||
String failedStanza = XMLBuilder.create("failed")
|
||||
.a("xmlns", "urn:xmpp:sm:3")
|
||||
.asString(outputProperties);
|
||||
|
||||
StreamManagement.Failed failedPacket = ParseStreamManagement.failed(
|
||||
PacketParserUtils.getParserFor(failedStanza));
|
||||
|
||||
assertThat(failedPacket, is(notNullValue()));
|
||||
XMPPError error = failedPacket.getXMPPError();
|
||||
|
||||
assertThat(error, is(notNullValue()));
|
||||
assertThat(error.getCondition(), equalTo("unknown"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testParseFailedError() throws Exception {
|
||||
String errorCondition = "failure";
|
||||
|
||||
String failedStanza = XMLBuilder.create("failed")
|
||||
.a("xmlns", "urn:xmpp:sm:3")
|
||||
.element(errorCondition, XMPPError.NAMESPACE)
|
||||
.asString(outputProperties);
|
||||
|
||||
System.err.println(failedStanza);
|
||||
|
||||
StreamManagement.Failed failedPacket = ParseStreamManagement.failed(
|
||||
PacketParserUtils.getParserFor(failedStanza));
|
||||
|
||||
assertThat(failedPacket, is(notNullValue()));
|
||||
XMPPError error = failedPacket.getXMPPError();
|
||||
|
||||
assertThat(error, is(notNullValue()));
|
||||
assertThat(error.getCondition(), equalTo(errorCondition));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testParseResumed() throws Exception {
|
||||
long handledPackets = 42;
|
||||
String previousID = "zid615d9";
|
||||
|
||||
String resumedStanza = XMLBuilder.create("resumed")
|
||||
.a("xmlns", "urn:xmpp:sm:3")
|
||||
.a("h", String.valueOf(handledPackets))
|
||||
.a("previd", previousID)
|
||||
.asString(outputProperties);
|
||||
|
||||
StreamManagement.Resumed resumedPacket = ParseStreamManagement.resumed(
|
||||
PacketParserUtils.getParserFor(resumedStanza));
|
||||
|
||||
assertThat(resumedPacket, is(notNullValue()));
|
||||
assertThat(resumedPacket.getHandledCount(), equalTo(handledPackets));
|
||||
assertThat(resumedPacket.getPrevId(), equalTo(previousID));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testParseAckAnswer() throws Exception {
|
||||
long handledPackets = 42 + 42;
|
||||
|
||||
String ackStanza = XMLBuilder.create("a")
|
||||
.a("xmlns", "urn:xmpp:sm:3")
|
||||
.a("h", String.valueOf(handledPackets))
|
||||
.asString(outputProperties);
|
||||
|
||||
StreamManagement.AckAnswer acknowledgementPacket = ParseStreamManagement.ackAnswer(
|
||||
PacketParserUtils.getParserFor(ackStanza));
|
||||
|
||||
assertThat(acknowledgementPacket, is(notNullValue()));
|
||||
assertThat(acknowledgementPacket.getHandledCount(), equalTo(handledPackets));
|
||||
}
|
||||
|
||||
|
||||
private static Properties initOutputProperties() {
|
||||
Properties properties = new Properties();
|
||||
properties.put(javax.xml.transform.OutputKeys.OMIT_XML_DECLARATION, "yes");
|
||||
return properties;
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue