/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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.apache.vysper.xmpp.modules.extension.xep0060_pubsub.disco; import java.util.Arrays; import java.util.List; import org.apache.vysper.xml.fragment.XMLElement; import org.apache.vysper.xmpp.addressing.Entity; import org.apache.vysper.xmpp.modules.core.base.handler.IQHandler; import org.apache.vysper.xmpp.modules.extension.xep0060_pubsub.AbstractPublishSubscribeTestCase; import org.apache.vysper.xmpp.modules.extension.xep0060_pubsub.feature.PubsubFeatures; import org.apache.vysper.xmpp.modules.extension.xep0060_pubsub.handler.AbstractStanzaGenerator; import org.apache.vysper.xmpp.modules.extension.xep0060_pubsub.model.LeafNode; import org.apache.vysper.xmpp.modules.servicediscovery.handler.DiscoInfoIQHandler; import org.apache.vysper.xmpp.protocol.NamespaceURIs; import org.apache.vysper.xmpp.protocol.ResponseStanzaContainer; import org.apache.vysper.xmpp.stanza.IQStanza; import org.apache.vysper.xmpp.stanza.IQStanzaType; import org.apache.vysper.xmpp.stanza.Stanza; import org.apache.vysper.xmpp.stanza.StanzaBuilder; public class PubSubDiscoInfoTestCase extends AbstractPublishSubscribeTestCase { @Override protected AbstractStanzaGenerator getDefaultStanzaGenerator() { return new DefaultDiscoInfoStanzaGenerator(); } @Override protected IQHandler getHandler() { return new DiscoInfoIQHandler(); } public void testIdentityAndFeature() { DefaultDiscoInfoStanzaGenerator sg = (DefaultDiscoInfoStanzaGenerator) getDefaultStanzaGenerator(); Stanza stanza = sg.getStanza(client, pubsubService.getBareJID(), "id123"); ResponseStanzaContainer result = sendStanza(stanza, true); assertTrue(result.hasResponse()); IQStanza response = new IQStanza(result.getResponseStanza()); assertEquals(IQStanzaType.RESULT.value(), response.getType()); assertEquals("id123", response.getAttributeValue("id")); // IDs must match // get the query Element XMLElement query = response.getFirstInnerElement(); List<XMLElement> inner = query.getInnerElements(); assertEquals("query", query.getName()); // at least we have an identity and a feature element assertTrue(inner.size() >= 2); // ordering etc. is unknown; step through all subelements and pick the ones we need XMLElement identity = null; XMLElement feature = null; for (XMLElement el : inner) { if (el.getName().equals("identity") //&& el.getNamespace().equals(NamespaceURIs.XEP0030_SERVICE_DISCOVERY_INFO) // TODO enable when the parser is fixed && el.getAttributeValue("category").equals("pubsub") && el.getAttributeValue("type").equals("service")) { identity = el; } else if (el.getName().equals("feature") /*&& el.getNamespace().equals(NamespaceURIs.XEP0030_SERVICE_DISCOVERY_INFO)*/// TODO enable when the parser is fixed && el.getAttributeValue("var").equals(NamespaceURIs.XEP0060_PUBSUB)) { feature = el; } } // make sure they were there (booleans would have sufficed) assertNotNull(identity); assertNotNull(feature); } public void testInfoRequestForANode() throws Exception { root.add(new LeafNode(serviceConfiguration, "news", "News", client)); DefaultDiscoInfoStanzaGenerator sg = (DefaultDiscoInfoStanzaGenerator) getDefaultStanzaGenerator(); Stanza stanza = sg.getStanza(client, pubsubService.getBareJID(), "id123", "news"); ResponseStanzaContainer result = sendStanza(stanza, true); assertTrue(result.hasResponse()); IQStanza response = new IQStanza(result.getResponseStanza()); assertEquals(IQStanzaType.RESULT.value(), response.getType()); assertEquals("id123", response.getAttributeValue("id")); // IDs must match // get the query Element XMLElement query = response.getFirstInnerElement(); List<XMLElement> inner = query.getInnerElements(); assertEquals("query", query.getName()); // at least we have an identity element assertTrue(inner.size() >= 1); // ordering etc. is unknown; step through all subelements and pick the ones we need XMLElement identity = null; for (XMLElement el : inner) { if (el.getName().equals("identity") //&& el.getNamespace().equals(NamespaceURIs.XEP0030_SERVICE_DISCOVERY_INFO) // TODO enable when the parser is fixed && el.getAttributeValue("category").equals("pubsub") && el.getAttributeValue("type").equals("leaf")) { identity = el; } } // make sure they were there assertNotNull(identity); String[] featuresList = new String[] { NamespaceURIs.XEP0060_PUBSUB, PubsubFeatures.ACCESS_OPEN.toString(), PubsubFeatures.ITEM_IDS.toString(), PubsubFeatures.PERSISTENT_ITEMS.toString(), PubsubFeatures.MULTI_SUBSCRIBE.toString(), PubsubFeatures.PUBLISH.toString(), PubsubFeatures.SUBSCRIBE.toString(), PubsubFeatures.RETRIEVE_SUBSCRIPTIONS.toString(), PubsubFeatures.RETRIEVE_AFFILIATIONS.toString(), PubsubFeatures.MODIFY_AFFILIATIONS.toString() }; XMLElement[] elementList = collectFeatures(inner, featuresList); for (int idx = 0; idx < elementList.length; ++idx) { assertNotNull(featuresList[idx], elementList[idx]); // add a more descriptive error message if the test fails } } private XMLElement[] collectFeatures(List<XMLElement> inner, String[] features) { XMLElement[] elementList = new XMLElement[features.length]; Arrays.sort(features); for (XMLElement el : inner) { if (el.getName().equals("feature")) /*&& el.getNamespace().equals(NamespaceURIs.XEP0030_SERVICE_DISCOVERY_INFO)*/{ // TODO enable when the parser is fixed int index = Arrays.binarySearch(features, el.getAttributeValue("var")); if (index != -1) { elementList[index] = el; } } } return elementList; } class DefaultDiscoInfoStanzaGenerator extends AbstractStanzaGenerator { @Override protected StanzaBuilder buildInnerElement(Entity client, Entity pubsub, StanzaBuilder sb, String node) { return sb; } @Override protected String getNamespace() { return NamespaceURIs.XEP0030_SERVICE_DISCOVERY_INFO; } @Override protected IQStanzaType getStanzaType() { return IQStanzaType.GET; } public Stanza getStanza(Entity client, Entity pubsub, String id) { StanzaBuilder stanzaBuilder = StanzaBuilder.createIQStanza(client, pubsub, getStanzaType(), id); stanzaBuilder.startInnerElement("query", getNamespace()); return stanzaBuilder.build(); } @Override public Stanza getStanza(Entity client, Entity pubsub, String id, String node) { StanzaBuilder stanzaBuilder = StanzaBuilder.createIQStanza(client, pubsub, getStanzaType(), id); stanzaBuilder.startInnerElement("query", getNamespace()); stanzaBuilder.addAttribute("node", node); stanzaBuilder.endInnerElement(); return stanzaBuilder.build(); } } }