231 lines
7.9 KiB
Java
231 lines
7.9 KiB
Java
/**
|
|
*
|
|
* 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.smackx.caps;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.assertNotNull;
|
|
import static org.junit.Assert.assertTrue;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.util.Collection;
|
|
import java.util.LinkedList;
|
|
|
|
import org.jivesoftware.smack.packet.IQ;
|
|
import org.jivesoftware.smack.util.StringUtils;
|
|
import org.jivesoftware.smack.util.stringencoder.Base32;
|
|
import org.jivesoftware.smack.util.stringencoder.StringEncoder;
|
|
|
|
import org.jivesoftware.smackx.InitExtensions;
|
|
import org.jivesoftware.smackx.caps.cache.EntityCapsPersistentCache;
|
|
import org.jivesoftware.smackx.caps.cache.SimpleDirectoryPersistentCache;
|
|
import org.jivesoftware.smackx.disco.packet.DiscoverInfo;
|
|
import org.jivesoftware.smackx.xdata.FormField;
|
|
import org.jivesoftware.smackx.xdata.packet.DataForm;
|
|
|
|
import org.junit.Test;
|
|
import org.jxmpp.jid.impl.JidCreate;
|
|
import org.jxmpp.stringprep.XmppStringprepException;
|
|
|
|
|
|
public class EntityCapsManagerTest extends InitExtensions {
|
|
|
|
/**
|
|
* <a href="http://xmpp.org/extensions/xep-0115.html#ver-gen-complex">XEP-
|
|
* 0115 Complex Generation Example</a>.
|
|
* @throws XmppStringprepException
|
|
*/
|
|
@Test
|
|
public void testComplexGenerationExample() throws XmppStringprepException {
|
|
DiscoverInfo di = createComplexSamplePacket();
|
|
|
|
CapsVersionAndHash versionAndHash = EntityCapsManager.generateVerificationString(di, StringUtils.SHA1);
|
|
assertEquals("q07IKJEyjvHSyhy//CH0CxmKi8w=", versionAndHash.version);
|
|
}
|
|
|
|
@Test
|
|
public void testSimpleDirectoryCacheBase32() throws IOException {
|
|
EntityCapsManager.persistentCache = null;
|
|
testSimpleDirectoryCache(Base32.getStringEncoder());
|
|
}
|
|
|
|
@Test
|
|
public void testVerificationDuplicateFeatures() throws XmppStringprepException {
|
|
DiscoverInfo di = createMalformedDiscoverInfo();
|
|
assertTrue(di.containsDuplicateFeatures());
|
|
}
|
|
|
|
@Test
|
|
public void testVerificationDuplicateIdentities() throws XmppStringprepException {
|
|
DiscoverInfo di = createMalformedDiscoverInfo();
|
|
assertTrue(di.containsDuplicateIdentities());
|
|
}
|
|
|
|
@SuppressWarnings("UnusedVariable")
|
|
private static void testSimpleDirectoryCache(StringEncoder<String> stringEncoder) throws IOException {
|
|
|
|
EntityCapsPersistentCache cache = new SimpleDirectoryPersistentCache(createTempDirectory());
|
|
EntityCapsManager.setPersistentCache(cache);
|
|
|
|
DiscoverInfo di = createComplexSamplePacket();
|
|
CapsVersionAndHash versionAndHash = EntityCapsManager.generateVerificationString(di, StringUtils.SHA1);
|
|
String nodeVer = di.getNode() + "#" + versionAndHash.version;
|
|
|
|
// Save the data in EntityCapsManager
|
|
EntityCapsManager.addDiscoverInfoByNode(nodeVer, di);
|
|
|
|
// Lose all the data
|
|
EntityCapsManager.clearMemoryCache();
|
|
|
|
DiscoverInfo restored_di = EntityCapsManager.getDiscoveryInfoByNodeVer(nodeVer);
|
|
assertNotNull(restored_di);
|
|
assertEquals(di.toXML().toString(), restored_di.toXML().toString());
|
|
}
|
|
|
|
private static DiscoverInfo createComplexSamplePacket() throws XmppStringprepException {
|
|
DiscoverInfo di = new DiscoverInfo();
|
|
di.setFrom(JidCreate.from("benvolio@capulet.lit/230193"));
|
|
di.setStanzaId("disco1");
|
|
di.setTo(JidCreate.from("juliet@capulet.lit/chamber"));
|
|
di.setType(IQ.Type.result);
|
|
|
|
Collection<DiscoverInfo.Identity> identities = new LinkedList<DiscoverInfo.Identity>();
|
|
DiscoverInfo.Identity i = new DiscoverInfo.Identity("client", "pc", "Psi 0.11", "en");
|
|
identities.add(i);
|
|
i = new DiscoverInfo.Identity("client", "pc", "Ψ 0.11", "el");
|
|
identities.add(i);
|
|
di.addIdentities(identities);
|
|
|
|
di.addFeature("http://jabber.org/protocol/disco#items");
|
|
di.addFeature(EntityCapsManager.NAMESPACE);
|
|
di.addFeature("http://jabber.org/protocol/muc");
|
|
di.addFeature("http://jabber.org/protocol/disco#info");
|
|
|
|
DataForm df = new DataForm(DataForm.Type.result);
|
|
|
|
FormField ff = new FormField("os");
|
|
ff.addValue("Mac");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("FORM_TYPE");
|
|
ff.setType(FormField.Type.hidden);
|
|
ff.addValue("urn:xmpp:dataforms:softwareinfo");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("ip_version");
|
|
ff.addValue("ipv4");
|
|
ff.addValue("ipv6");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("os_version");
|
|
ff.addValue("10.5.1");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("software");
|
|
ff.addValue("Psi");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("software_version");
|
|
ff.addValue("0.11");
|
|
df.addField(ff);
|
|
|
|
di.addExtension(df);
|
|
return di;
|
|
}
|
|
|
|
private static DiscoverInfo createMalformedDiscoverInfo() throws XmppStringprepException {
|
|
DiscoverInfo di = new DiscoverInfo();
|
|
di.setFrom(JidCreate.from("benvolio@capulet.lit/230193"));
|
|
di.setStanzaId("disco1");
|
|
di.setTo(JidCreate.from(")juliet@capulet.lit/chamber"));
|
|
di.setType(IQ.Type.result);
|
|
|
|
Collection<DiscoverInfo.Identity> identities = new LinkedList<DiscoverInfo.Identity>();
|
|
DiscoverInfo.Identity i = new DiscoverInfo.Identity("client", "pc", "Psi 0.11", "en");
|
|
identities.add(i);
|
|
i = new DiscoverInfo.Identity("client", "pc", "Ψ 0.11", "el");
|
|
identities.add(i);
|
|
di.addIdentities(identities);
|
|
// Failure 1: Duplicate identities
|
|
i = new DiscoverInfo.Identity("client", "pc", "Ψ 0.11", "el");
|
|
identities.add(i);
|
|
di.addIdentities(identities);
|
|
|
|
di.addFeature("http://jabber.org/protocol/disco#items");
|
|
di.addFeature(EntityCapsManager.NAMESPACE);
|
|
di.addFeature("http://jabber.org/protocol/muc");
|
|
di.addFeature("http://jabber.org/protocol/disco#info");
|
|
// Failure 2: Duplicate features
|
|
di.addFeature("http://jabber.org/protocol/disco#info");
|
|
|
|
DataForm df = new DataForm(DataForm.Type.result);
|
|
|
|
FormField ff = new FormField("os");
|
|
ff.addValue("Mac");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("FORM_TYPE");
|
|
ff.setType(FormField.Type.hidden);
|
|
ff.addValue("urn:xmpp:dataforms:softwareinfo");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("ip_version");
|
|
ff.addValue("ipv4");
|
|
ff.addValue("ipv6");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("os_version");
|
|
ff.addValue("10.5.1");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("software");
|
|
ff.addValue("Psi");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("software_version");
|
|
ff.addValue("0.11");
|
|
df.addField(ff);
|
|
|
|
di.addExtension(df);
|
|
|
|
// Failure 3: Another service discovery information form with the same
|
|
// FORM_TYPE
|
|
df = new DataForm(DataForm.Type.result);
|
|
|
|
ff = new FormField("FORM_TYPE");
|
|
ff.setType(FormField.Type.hidden);
|
|
ff.addValue("urn:xmpp:dataforms:softwareinfo");
|
|
df.addField(ff);
|
|
|
|
ff = new FormField("software");
|
|
ff.addValue("smack");
|
|
df.addField(ff);
|
|
|
|
di.addExtension(df);
|
|
|
|
return di;
|
|
}
|
|
|
|
public static File createTempDirectory() throws IOException {
|
|
File tmp = File.createTempFile("entity", "caps");
|
|
tmp.delete();
|
|
tmp.mkdir();
|
|
return tmp;
|
|
}
|
|
|
|
}
|