Smack/smack-extensions/src/test/java/org/jivesoftware/smackx/pubsub/ItemValidationTest.java

/**
 *
 * Copyright 2011 Robin Collier
 *
 * 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.pubsub;

import static org.custommonkey.xmlunit.XMLAssert.assertXMLEqual;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import org.jivesoftware.smack.ThreadedDummyConnection;
import org.jivesoftware.smack.packet.Packet;
import org.jivesoftware.smack.packet.PacketExtension;
import org.jivesoftware.smack.util.PacketParserUtils;
import org.jivesoftware.smackx.InitExtensions;
import org.jivesoftware.smackx.pubsub.packet.PubSubNamespace;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.xmlpull.v1.XmlPullParser;

/**
 * 
 * @author Robin Collier
 *
 */
public class ItemValidationTest extends InitExtensions {
	private ThreadedDummyConnection connection;

	@Before
	public void setUp() throws Exception
	{
		// Uncomment this to enable debug output
		// XMPPConnection.DEBUG_ENABLED = true;

		connection = new ThreadedDummyConnection();
		connection.connect();
		connection.login();
	}

	@After
	public void tearDown() throws Exception
	{
		if (connection != null)
			connection.disconnect();
	}

	@Test
	public void verifyBasicItem() throws Exception
	{
		Item simpleItem = new Item();
		String simpleCtrl = "<item />";
		assertXMLEqual(simpleCtrl, simpleItem.toXML());

		Item idItem = new Item("uniqueid");
		String idCtrl = "<item id='uniqueid'/>";
		assertXMLEqual(idCtrl, idItem.toXML());

		Item itemWithNodeId = new Item("testId", "testNode");
		String nodeIdCtrl = "<item id='testId' node='testNode' />";
		assertXMLEqual(nodeIdCtrl, itemWithNodeId.toXML());
	}

	@Test
	public void verifyPayloadItem() throws Exception
	{
		SimplePayload payload = new SimplePayload(null, null, "<data>This is the payload</data>");

		PayloadItem<SimplePayload> simpleItem = new PayloadItem<SimplePayload>(payload);
		String simpleCtrl = "<item>" + payload.toXML() + "</item>";
		assertXMLEqual(simpleCtrl, simpleItem.toXML());

		PayloadItem<SimplePayload> idItem = new PayloadItem<SimplePayload>("uniqueid", payload);
		String idCtrl = "<item id='uniqueid'>" + payload.toXML() + "</item>";
		assertXMLEqual(idCtrl, idItem.toXML());

		PayloadItem<SimplePayload> itemWithNodeId = new PayloadItem<SimplePayload>("testId", "testNode", payload);
		String nodeIdCtrl = "<item id='testId' node='testNode'>" + payload.toXML() + "</item>";
		assertXMLEqual(nodeIdCtrl, itemWithNodeId.toXML());
	}

	@Test
	public void parseBasicItem() throws Exception
	{
        XmlPullParser parser = PacketParserUtils.getParserFor(
            "<message from='pubsub.myserver.com' to='francisco@denmark.lit' id='foo'>" +
                "<event xmlns='http://jabber.org/protocol/pubsub#event'>" +
                    "<items node='testNode'>" +
                        "<item id='testid1' />" +
                    "</items>" + 
                 "</event>" + 
            "</message>");
        
        Packet message = PacketParserUtils.parseMessage(parser);
        PacketExtension eventExt = message.getExtension(PubSubNamespace.EVENT.getXmlns());
        
        assertTrue(eventExt instanceof EventElement);
        EventElement event = (EventElement) eventExt;
        assertEquals(EventElementType.items, event.getEventType());
        assertEquals(1, event.getExtensions().size());
        assertTrue(event.getExtensions().get(0) instanceof ItemsExtension);
        assertEquals(1, ((ItemsExtension)event.getExtensions().get(0)).items.size());
        
        PacketExtension itemExt = ((ItemsExtension)event.getExtensions().get(0)).items.get(0);
        assertTrue(itemExt instanceof Item);
        assertEquals("testid1", ((Item)itemExt).getId());
	}

    @Test
    public void parseSimplePayloadItem() throws Exception
    {
        String itemContent = "<foo xmlns='smack:test'>Some text</foo>";
        
        XmlPullParser parser = PacketParserUtils.getParserFor(
            "<message from='pubsub.myserver.com' to='francisco@denmark.lit' id='foo'>" +
                "<event xmlns='http://jabber.org/protocol/pubsub#event'>" +
                    "<items node='testNode'>" +
                        "<item id='testid1' >" +
                            itemContent +
                        "</item>" + 
                    "</items>" + 
                 "</event>" + 
            "</message>");
        
        Packet message = PacketParserUtils.parseMessage(parser);
        PacketExtension eventExt = message.getExtension(PubSubNamespace.EVENT.getXmlns());
        EventElement event = (EventElement) eventExt;
        PacketExtension itemExt = ((ItemsExtension)event.getExtensions().get(0)).items.get(0);

        assertTrue(itemExt instanceof PayloadItem<?>);
        PayloadItem<?> item = (PayloadItem<?>)itemExt;
        
        assertEquals("testid1", item.getId());
        assertTrue(item.getPayload() instanceof SimplePayload);
        
        SimplePayload payload = (SimplePayload) item.getPayload();
        assertEquals("foo", payload.getElementName());
        assertEquals("smack:test", payload.getNamespace());
        assertXMLEqual(itemContent, payload.toXML().toString());
    }

    @Test
    public void parseComplexItem() throws Exception
    {
        String itemContent = 
                "<entry xmlns='http://www.w3.org/2005/Atom'>" +
                    "<title>Soliloquy</title>" +
                    "<summary>" +
                        "To be, or not to be: that is the question:" +
                        "Whether 'tis nobler in the mind to suffer" +
                        "The slings and arrows of outrageous fortune," +
                        "Or to take arms against a sea of troubles," +
                        "And by opposing end them?" +
                    "</summary>" +
                    "<link rel='alternate' type='text/html' href='http://denmark.lit/2003/12/13/atom03'/>" +
                    "<id>tag:denmark.lit,2003:entry-32397</id>" +
                    "<published>2003-12-13T18:30:02Z</published>" +
                    "<updated>2003-12-13T18:30:02Z</updated>" +
                "</entry>";
        
        XmlPullParser parser = PacketParserUtils.getParserFor(
            "<message from='pubsub.myserver.com' to='francisco@denmark.lit' id='foo'>" +
                "<event xmlns='http://jabber.org/protocol/pubsub#event'>" +
                    "<items node='testNode'>" +
                        "<item id='testid1' >" +
                            itemContent +
                        "</item>" + 
                    "</items>" + 
                 "</event>" + 
            "</message>");
        
        Packet message = PacketParserUtils.parseMessage(parser);
        PacketExtension eventExt = message.getExtension(PubSubNamespace.EVENT.getXmlns());
        EventElement event = (EventElement) eventExt;
        PacketExtension itemExt = ((ItemsExtension)event.getExtensions().get(0)).items.get(0);

        assertTrue(itemExt instanceof PayloadItem<?>);
        PayloadItem<?> item = (PayloadItem<?>)itemExt;
        
        assertEquals("testid1", item.getId());
        assertTrue(item.getPayload() instanceof SimplePayload);
        
        SimplePayload payload = (SimplePayload) item.getPayload();
        assertEquals("entry", payload.getElementName());
        assertEquals("http://www.w3.org/2005/Atom", payload.getNamespace());
        assertXMLEqual(itemContent, payload.toXML().toString());
    }

    @Test
    public void parseEmptyTag() throws Exception
    {
        String itemContent = "<foo xmlns='smack:test'><bar/></foo>";
        
        XmlPullParser parser = PacketParserUtils.getParserFor(
            "<message from='pubsub.myserver.com' to='francisco@denmark.lit' id='foo'>" +
                "<event xmlns='http://jabber.org/protocol/pubsub#event'>" +
                    "<items node='testNode'>" +
                        "<item id='testid1' >" +
                            itemContent +
                        "</item>" + 
                    "</items>" + 
                 "</event>" + 
            "</message>");
        
        Packet message = PacketParserUtils.parseMessage(parser);
        PacketExtension eventExt = message.getExtension(PubSubNamespace.EVENT.getXmlns());
        
        assertTrue(eventExt instanceof EventElement);
        EventElement event = (EventElement) eventExt;
        assertEquals(EventElementType.items, event.getEventType());
        assertEquals(1, event.getExtensions().size());
        assertTrue(event.getExtensions().get(0) instanceof ItemsExtension);
        assertEquals(1, ((ItemsExtension)event.getExtensions().get(0)).items.size());
        
        PacketExtension itemExt = ((ItemsExtension)event.getExtensions().get(0)).items.get(0);
        assertTrue(itemExt instanceof PayloadItem<?>);
        PayloadItem<?> item = (PayloadItem<?>)itemExt;
        
        assertEquals("testid1", item.getId());
        assertTrue(item.getPayload() instanceof SimplePayload);
        
        assertXMLEqual(itemContent, ((SimplePayload)item.getPayload()).toXML().toString());
    }
}