Implement Message Archive Management (MAM) XEP-0313

Fixes SMACK-625

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/FiltersTest.java

@@ -0,0 +1,160 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import java.lang.reflect.Method;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+
+import org.jivesoftware.smackx.mam.element.MamElements;
+import org.jivesoftware.smackx.xdata.packet.DataForm;
+import org.junit.Assert;
+import org.junit.Test;
+import org.jxmpp.jid.Jid;
+import org.jxmpp.jid.impl.JidCreate;
+import org.jxmpp.util.XmppDateTime;
+
+public class FiltersTest extends MamTest {
+
+    private String getMamXMemberWith(List<String> fieldsNames, List<String> fieldsValues) {
+        String xml = "<x xmlns='jabber:x:data' type='submit'>" + "<field var='FORM_TYPE' type='hidden'>" + "<value>"
+                + MamElements.NAMESPACE + "</value>" + "</field>";
+
+        for (int i = 0; i < fieldsNames.size() && i < fieldsValues.size(); i++) {
+            xml += "<field var='" + fieldsNames.get(i) + "'>" + "<value>" + fieldsValues.get(i) + "</value>"
+                    + "</field>";
+        }
+
+        xml += "</x>";
+        return xml;
+    }
+
+    @Test
+    public void checkStartDateFilter() throws Exception {
+        Method methodAddStartDateFilter = MamManager.class.getDeclaredMethod("addStart", Date.class, DataForm.class);
+        methodAddStartDateFilter.setAccessible(true);
+
+        Date date = new Date();
+        DataForm dataForm = getNewMamForm();
+        methodAddStartDateFilter.invoke(mamManager, date, dataForm);
+
+        List<String> fields = new ArrayList<>();
+        fields.add("start");
+        List<String> values = new ArrayList<>();
+        values.add(XmppDateTime.formatXEP0082Date(date));
+
+        Assert.assertEquals(dataForm.toXML().toString(), getMamXMemberWith(fields, values));
+    }
+
+    @Test
+    public void checkEndDateFilter() throws Exception {
+        Method methodAddEndDateFilter = MamManager.class.getDeclaredMethod("addEnd", Date.class, DataForm.class);
+        methodAddEndDateFilter.setAccessible(true);
+
+        Date date = new Date();
+        DataForm dataForm = getNewMamForm();
+        methodAddEndDateFilter.invoke(mamManager, date, dataForm);
+
+        List<String> fields = new ArrayList<>();
+        fields.add("end");
+        List<String> values = new ArrayList<>();
+        values.add(XmppDateTime.formatXEP0082Date(date));
+
+        Assert.assertEquals(dataForm.toXML().toString(), getMamXMemberWith(fields, values));
+    }
+
+    @Test
+    public void checkWithJidFilter() throws Exception {
+        Method methodAddJidFilter = MamManager.class.getDeclaredMethod("addWithJid", Jid.class, DataForm.class);
+        methodAddJidFilter.setAccessible(true);
+
+        String jid = "test@jid.com";
+        DataForm dataForm = getNewMamForm();
+        methodAddJidFilter.invoke(mamManager, JidCreate.from(jid), dataForm);
+
+        List<String> fields = new ArrayList<>();
+        fields.add("with");
+        List<String> values = new ArrayList<>();
+        values.add(jid);
+
+        Assert.assertEquals(dataForm.toXML().toString(), getMamXMemberWith(fields, values));
+    }
+
+    @Test
+    public void checkMultipleFilters() throws Exception {
+        Method methodAddStartDateFilter = MamManager.class.getDeclaredMethod("addStart", Date.class, DataForm.class);
+        methodAddStartDateFilter.setAccessible(true);
+        Method methodAddEndDateFilter = MamManager.class.getDeclaredMethod("addEnd", Date.class, DataForm.class);
+        methodAddEndDateFilter.setAccessible(true);
+        Method methodAddJidFilter = MamManager.class.getDeclaredMethod("addWithJid", Jid.class, DataForm.class);
+        methodAddJidFilter.setAccessible(true);
+
+        DataForm dataForm = getNewMamForm();
+        Date date = new Date();
+        String dateString = XmppDateTime.formatXEP0082Date(date);
+        String jid = "test@jid.com";
+
+        methodAddStartDateFilter.invoke(mamManager, date, dataForm);
+        methodAddEndDateFilter.invoke(mamManager, date, dataForm);
+        methodAddJidFilter.invoke(mamManager, JidCreate.from(jid), dataForm);
+        String dataFormResult = dataForm.toXML().toString();
+
+        List<String> fields = new ArrayList<>();
+        List<String> values = new ArrayList<>();
+
+        fields.add("start");
+        values.add(dateString);
+        Assert.assertNotEquals(dataFormResult, getMamXMemberWith(fields, values));
+
+        fields.add("end");
+        values.add(dateString);
+        Assert.assertNotEquals(dataFormResult, getMamXMemberWith(fields, values));
+
+        fields.clear();
+        values.clear();
+
+        fields.add("start");
+        values.add(dateString);
+        fields.add("with");
+        values.add(jid);
+        Assert.assertNotEquals(dataFormResult, getMamXMemberWith(fields, values));
+
+        fields.clear();
+        values.clear();
+
+        fields.add("end");
+        values.add(dateString);
+        fields.add("with");
+        values.add(jid);
+        fields.add("start");
+        values.add(dateString);
+        Assert.assertNotEquals(dataFormResult, getMamXMemberWith(fields, values));
+
+        fields.clear();
+        values.clear();
+
+        fields.add("start");
+        values.add(dateString);
+        fields.add("end");
+        values.add(dateString);
+        fields.add("with");
+        values.add(jid);
+        Assert.assertEquals(dataFormResult, getMamXMemberWith(fields, values));
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/MamFinProviderTest.java

@@ -0,0 +1,69 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import org.jivesoftware.smack.packet.IQ;
+import org.jivesoftware.smack.packet.IQ.Type;
+import org.jivesoftware.smack.util.PacketParserUtils;
+import org.jivesoftware.smackx.mam.element.MamFinIQ;
+import org.jivesoftware.smackx.mam.provider.MamFinIQProvider;
+import org.jivesoftware.smackx.rsm.packet.RSMSet;
+import org.junit.Assert;
+import org.junit.Test;
+import org.xmlpull.v1.XmlPullParser;
+
+public class MamFinProviderTest extends MamTest {
+
+    String exmapleMamFinXml = "<fin xmlns='urn:xmpp:mam:1' stable='true'>"
+            + "<set xmlns='http://jabber.org/protocol/rsm'>" + "<max>10</max>" + "<after>09af3-cc343-b409f</after>"
+            + "</set>" + "</fin>";
+
+    private String getIQLimitedResultsExample() {
+        return "<iq type='result' id='u29303'>" + "<fin xmlns='urn:xmpp:mam:1' complete='true'>"
+                + "<set xmlns='http://jabber.org/protocol/rsm'>" + "<first index='0'>23452-4534-1</first>"
+                + "<last>390-2342-22</last>" + "<count>16</count>" + "</set>" + "</fin>" + "</iq>";
+    }
+
+    @Test
+    public void checkMamFinProvider() throws Exception {
+        XmlPullParser parser = PacketParserUtils.getParserFor(exmapleMamFinXml);
+        MamFinIQ mamFinIQ = new MamFinIQProvider().parse(parser);
+
+        Assert.assertFalse(mamFinIQ.isComplete());
+        Assert.assertTrue(mamFinIQ.isStable());
+        Assert.assertNull(mamFinIQ.getQueryId());
+
+        RSMSet rsmSet = mamFinIQ.getRSMSet();
+        Assert.assertEquals(rsmSet.getAfter(), "09af3-cc343-b409f");
+        Assert.assertEquals(rsmSet.getMax(), 10);
+    }
+
+    @Test
+    public void checkQueryLimitedResults() throws Exception {
+        IQ iq = (IQ) PacketParserUtils.parseStanza(getIQLimitedResultsExample());
+
+        MamFinIQ mamFinIQ = (MamFinIQ) iq;
+        Assert.assertEquals(mamFinIQ.getType(), Type.result);
+
+        Assert.assertTrue(mamFinIQ.isComplete());
+        Assert.assertEquals(mamFinIQ.getRSMSet().getCount(), 16);
+        Assert.assertEquals(mamFinIQ.getRSMSet().getFirst(), "23452-4534-1");
+        Assert.assertEquals(mamFinIQ.getRSMSet().getFirstIndex(), 0);
+        Assert.assertEquals(mamFinIQ.getRSMSet().getLast(), "390-2342-22");
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/MamPrefIQProviderTest.java

@@ -0,0 +1,84 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import java.util.List;
+
+import org.jivesoftware.smack.packet.IQ;
+import org.jivesoftware.smack.util.PacketParserUtils;
+import org.jivesoftware.smackx.mam.element.MamPrefsIQ;
+import org.jivesoftware.smackx.mam.provider.MamPrefsIQProvider;
+import org.junit.Assert;
+import org.junit.Test;
+import org.jxmpp.jid.Jid;
+import org.xmlpull.v1.XmlPullParser;
+
+public class MamPrefIQProviderTest extends MamTest {
+
+    String exampleMamPrefsIQ1 = "<iq type='set' id='juliet3'>" + "<prefs xmlns='urn:xmpp:mam:1' default='roster'>"
+            + "<always>" + "<jid>romeo@montague.lit</jid>" + "</always>" + "<never>"
+            + "<jid>montague@montague.lit</jid>" + "</never>" + "</prefs>" + "</iq>";
+
+    String exampleMamPrefsIQ2 = "<iq type='set' id='juliet3'>" + "<prefs xmlns='urn:xmpp:mam:1' default='roster'>"
+            + "<always>" + "<jid>romeo@montague.lit</jid>" + "<jid>montague@montague.lit</jid>" + "</always>"
+            + "<never>" + "</never>" + "</prefs>" + "</iq>";
+
+    String exampleMamPrefsIQ3 = "<iq type='get' id='juliet3'>" + "<prefs xmlns='urn:xmpp:mam:1'>" + "</prefs>"
+            + "</iq>";
+
+    String exampleMamPrefsResultIQ = "<iq type='result' id='juliet3'>"
+            + "<prefs xmlns='urn:xmpp:mam:1' default='roster'>" + "<always>" + "<jid>romeo@montague.lit</jid>"
+            + "</always>" + "<never>" + "<jid>sarasa@montague.lit</jid>" + "<jid>montague@montague.lit</jid>"
+            + "</never>" + "</prefs>" + "</iq>";
+
+    @Test
+    public void checkMamPrefsIQProvider() throws Exception {
+        XmlPullParser parser1 = PacketParserUtils.getParserFor(exampleMamPrefsIQ1);
+        MamPrefsIQ mamPrefIQ1 = new MamPrefsIQProvider().parse(parser1);
+
+        Assert.assertTrue(mamPrefIQ1.isUpdate());
+        Assert.assertEquals(mamPrefIQ1.getAlwaysJids().get(0), "romeo@montague.lit");
+        Assert.assertEquals(mamPrefIQ1.getNeverJids().get(0), "montague@montague.lit");
+
+        XmlPullParser parser2 = PacketParserUtils.getParserFor(exampleMamPrefsIQ2);
+        MamPrefsIQ mamPrefIQ2 = new MamPrefsIQProvider().parse(parser2);
+        Assert.assertTrue(mamPrefIQ2.isUpdate());
+        Assert.assertEquals(mamPrefIQ2.getAlwaysJids().get(0), "romeo@montague.lit");
+        Assert.assertEquals(mamPrefIQ2.getAlwaysJids().get(1), "montague@montague.lit");
+        Assert.assertTrue(mamPrefIQ2.getNeverJids().isEmpty());
+
+        XmlPullParser parser3 = PacketParserUtils.getParserFor(exampleMamPrefsIQ3);
+        MamPrefsIQ mamPrefIQ3 = new MamPrefsIQProvider().parse(parser3);
+        Assert.assertFalse(mamPrefIQ3.isUpdate());
+    }
+
+    @Test
+    public void checkMamPrefResult() throws Exception {
+        IQ iq = (IQ) PacketParserUtils.parseStanza(exampleMamPrefsResultIQ);
+
+        MamPrefsIQ mamPrefsIQ = (MamPrefsIQ) iq;
+
+        List<Jid> alwaysJids = mamPrefsIQ.getAlwaysJids();
+        List<Jid> neverJids = mamPrefsIQ.getNeverJids();
+
+        Assert.assertEquals(alwaysJids.size(), 1);
+        Assert.assertEquals(neverJids.size(), 2);
+        Assert.assertEquals(alwaysJids.get(0).toString(), "romeo@montague.lit");
+        Assert.assertEquals(neverJids.get(1).toString(), "montague@montague.lit");
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/MamQueryIQProviderTest.java

@@ -0,0 +1,86 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.jivesoftware.smack.packet.IQ;
+import org.jivesoftware.smack.packet.IQ.Type;
+import org.jivesoftware.smack.util.PacketParserUtils;
+import org.jivesoftware.smackx.mam.element.MamQueryIQ;
+import org.jivesoftware.smackx.xdata.FormField;
+import org.jivesoftware.smackx.xdata.packet.DataForm;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class MamQueryIQProviderTest {
+
+    String exampleMamQueryIQ1 = "<iq type='set' id='query4'>" + "<query xmlns='urn:xmpp:mam:1' queryid='test'>"
+            + "<x xmlns='jabber:x:data' type='submit'>" + "<field type='hidden' var='FORM_TYPE'>"
+            + "<value>urn:xmpp:mam:1</value>" + "</field>"
+            + "<field type='text-single' var='urn:example:xmpp:free-text-search'>"
+            + "<value>Where arth thou, my Juliet?</value>" + "</field>"
+            + "<field type='text-single' var='urn:example:xmpp:stanza-content'>"
+            + "<value>{http://jabber.org/protocol/mood}mood/lonely</value>" + "</field>" + "</x>" + "</query>"
+            + "</iq>";
+
+    String exampleMamQueryIQ2 = "<iq type='result' id='form1'>" + "<query xmlns='urn:xmpp:mam:1'>"
+            + "<x xmlns='jabber:x:data' type='form'>" + "<field type='hidden' var='FORM_TYPE'>"
+            + "<value>urn:xmpp:mam:1</value>" + "</field>" + "<field type='jid-single' var='with'/>"
+            + "<field type='text-single' var='start'/>" + "<field type='text-single' var='end'/>"
+            + "<field type='text-single' var='urn:example:xmpp:free-text-search'/>"
+            + "<field type='text-single' var='urn:example:xmpp:stanza-content'/>" + "</x>" + "</query>" + "</iq>";
+
+    @Test
+    public void checkMamQueryIQProvider() throws Exception {
+        // example 1
+        IQ iq1 = (IQ) PacketParserUtils.parseStanza(exampleMamQueryIQ1);
+        MamQueryIQ mamQueryIQ1 = (MamQueryIQ) iq1;
+
+        Assert.assertEquals(mamQueryIQ1.getType(), Type.set);
+        Assert.assertEquals(mamQueryIQ1.getQueryId(), "test");
+
+        DataForm dataForm1 = (DataForm) mamQueryIQ1.getExtension(DataForm.NAMESPACE);
+        Assert.assertEquals(dataForm1.getType(), DataForm.Type.submit);
+
+        List<FormField> fields1 = dataForm1.getFields();
+        Assert.assertEquals(fields1.get(0).getType(), FormField.Type.hidden);
+        Assert.assertEquals(fields1.get(1).getType(), FormField.Type.text_single);
+        Assert.assertEquals(fields1.get(1).getValues().get(0), "Where arth thou, my Juliet?");
+        Assert.assertEquals(fields1.get(2).getValues().get(0), "{http://jabber.org/protocol/mood}mood/lonely");
+
+        // example2
+        IQ iq2 = (IQ) PacketParserUtils.parseStanza(exampleMamQueryIQ2);
+        MamQueryIQ mamQueryIQ2 = (MamQueryIQ) iq2;
+
+        Assert.assertEquals(mamQueryIQ2.getType(), Type.result);
+        Assert.assertNull(mamQueryIQ2.getQueryId());
+
+        DataForm dataForm2 = (DataForm) mamQueryIQ2.getExtension(DataForm.NAMESPACE);
+        Assert.assertEquals(dataForm2.getType(), DataForm.Type.form);
+
+        List<FormField> fields2 = dataForm2.getFields();
+        Assert.assertEquals(fields2.get(0).getValues().get(0), "urn:xmpp:mam:1");
+        Assert.assertTrue(fields2.get(0).getValues().size() == 1);
+        Assert.assertEquals(fields2.get(1).getType(), FormField.Type.jid_single);
+        Assert.assertEquals(fields2.get(2).getType(), FormField.Type.text_single);
+        Assert.assertEquals(fields2.get(2).getValues(), new ArrayList<>());
+        Assert.assertEquals(fields2.get(4).getVariable(), "urn:example:xmpp:free-text-search");
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/MamResultProviderTest.java

@@ -0,0 +1,90 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import java.util.Date;
+import java.util.GregorianCalendar;
+import java.util.TimeZone;
+
+import org.jivesoftware.smack.packet.Message;
+import org.jivesoftware.smack.util.PacketParserUtils;
+import org.jivesoftware.smackx.forward.packet.Forwarded;
+import org.jivesoftware.smackx.mam.element.MamElements.MamResultExtension;
+import org.jivesoftware.smackx.mam.provider.MamResultProvider;
+import org.junit.Assert;
+import org.junit.Test;
+import org.xmlpull.v1.XmlPullParser;
+
+public class MamResultProviderTest {
+
+    String exampleMamResultXml = "<result xmlns='urn:xmpp:mam:1' queryid='f27' id='28482-98726-73623'>"
+            + "<forwarded xmlns='urn:xmpp:forward:0'>" + "<delay xmlns='urn:xmpp:delay' stamp='2010-07-10T23:08:25Z'/>"
+            + "<message xmlns='jabber:client'" + "to='juliet@capulet.lit/balcony'" + "from='romeo@montague.lit/orchard'"
+            + "type='chat'>"
+            + "<body>Call me but love, and I'll be new baptized; Henceforth I never will be Romeo.</body>"
+            + "</message>" + "</forwarded>" + "</result>";
+
+    String exampleResultMessage = "<message id='aeb213' to='juliet@capulet.lit/chamber'>"
+            + "<result xmlns='urn:xmpp:mam:1' queryid='f27' id='28482-98726-73623'>"
+            + "<forwarded xmlns='urn:xmpp:forward:0'>" + "<delay xmlns='urn:xmpp:delay' stamp='2010-07-10T23:08:25Z'/>"
+            + "<message xmlns='jabber:client' from='witch@shakespeare.lit' to='macbeth@shakespeare.lit'>"
+            + "<body>Hail to thee</body>" + "</message>" + "</forwarded>" + "</result>" + "</message>";
+
+    @Test
+    public void checkMamResultProvider() throws Exception {
+        XmlPullParser parser = PacketParserUtils.getParserFor(exampleMamResultXml);
+        MamResultExtension mamResultExtension = new MamResultProvider().parse(parser);
+
+        Assert.assertEquals(mamResultExtension.getQueryId(), "f27");
+        Assert.assertEquals(mamResultExtension.getId(), "28482-98726-73623");
+
+        GregorianCalendar calendar = new GregorianCalendar(2010, 7 - 1, 10, 23, 8, 25);
+        calendar.setTimeZone(TimeZone.getTimeZone("UTC"));
+        Date date = calendar.getTime();
+
+        Forwarded forwarded = mamResultExtension.getForwarded();
+        Assert.assertEquals(forwarded.getDelayInformation().getStamp(), date);
+
+        Message message = (Message) forwarded.getForwardedStanza();
+        Assert.assertEquals(message.getFrom(), "romeo@montague.lit/orchard");
+        Assert.assertEquals(message.getTo(), "juliet@capulet.lit/balcony");
+        Assert.assertEquals(message.getBody(),
+                "Call me but love, and I'll be new baptized; Henceforth I never will be Romeo.");
+    }
+
+    @Test
+    public void checkResultsParse() throws Exception {
+        Message message = (Message) PacketParserUtils.parseStanza(exampleResultMessage);
+        MamResultExtension mamResultExtension = MamResultExtension.from(message);
+
+        Assert.assertEquals(mamResultExtension.getQueryId(), "f27");
+        Assert.assertEquals(mamResultExtension.getId(), "28482-98726-73623");
+
+        GregorianCalendar calendar = new GregorianCalendar(2010, 7 - 1, 10, 23, 8, 25);
+        calendar.setTimeZone(TimeZone.getTimeZone("UTC"));
+        Date date = calendar.getTime();
+
+        Forwarded forwarded = mamResultExtension.getForwarded();
+        Assert.assertEquals(forwarded.getDelayInformation().getStamp(), date);
+
+        Message forwardedMessage = (Message) forwarded.getForwardedStanza();
+        Assert.assertEquals(forwardedMessage.getFrom(), "witch@shakespeare.lit");
+        Assert.assertEquals(forwardedMessage.getTo(), "macbeth@shakespeare.lit");
+        Assert.assertEquals(forwardedMessage.getBody(), "Hail to thee");
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/MamTest.java

@@ -0,0 +1,54 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import static org.mockito.Mockito.mock;
+
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+
+import org.jivesoftware.smack.XMPPConnection;
+import org.jivesoftware.smackx.ExperimentalInitializerTest;
+import org.jivesoftware.smackx.xdata.packet.DataForm;
+import org.junit.Before;
+
+public class MamTest extends ExperimentalInitializerTest {
+
+    protected XMPPConnection connection;
+    protected String queryId;
+    protected MamManager mamManager;
+
+    @Before
+    public void setup() {
+        // mock connection
+        connection = mock(XMPPConnection.class);
+
+        // test query id
+        queryId = "testid";
+
+        // MamManager instance
+        mamManager = MamManager.getInstanceFor(connection);
+    }
+
+    protected DataForm getNewMamForm() throws NoSuchMethodException, SecurityException, IllegalAccessException,
+            IllegalArgumentException, InvocationTargetException {
+        Method methodGetNewMamForm = MamManager.class.getDeclaredMethod("getNewMamForm");
+        methodGetNewMamForm.setAccessible(true);
+        return (DataForm) methodGetNewMamForm.invoke(mamManager);
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/PagingTest.java

@@ -0,0 +1,55 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import java.lang.reflect.Method;
+
+import org.jivesoftware.smackx.mam.element.MamQueryIQ;
+import org.jivesoftware.smackx.rsm.packet.RSMSet;
+import org.jivesoftware.smackx.xdata.packet.DataForm;
+import org.junit.Test;
+
+import org.junit.Assert;
+
+public class PagingTest extends MamTest {
+
+    String pagingStanza = "<iq id='sarasa' type='set'>" + "<query xmlns='urn:xmpp:mam:1' queryid='testid'>"
+            + "<x xmlns='jabber:x:data' type='submit'>" + "<field var='FORM_TYPE' type='hidden'>"
+            + "<value>urn:xmpp:mam:1</value>" + "</field>" + "</x>" + "<set xmlns='http://jabber.org/protocol/rsm'>"
+            + "<max>10</max>" + "</set>" + "</query>" + "</iq>";
+
+    @Test
+    public void checkPageQueryStanza() throws Exception {
+        Method methodPreparePageQuery = MamManager.class.getDeclaredMethod("preparePageQuery", MamQueryIQ.class,
+                RSMSet.class);
+        methodPreparePageQuery.setAccessible(true);
+
+        DataForm dataForm = getNewMamForm();
+        int max = 10;
+        RSMSet rsmSet = new RSMSet(max);
+
+        MamQueryIQ mamQueryIQ = new MamQueryIQ(queryId, dataForm);
+        mamQueryIQ.setStanzaId("sarasa");
+
+        methodPreparePageQuery.invoke(mamManager, mamQueryIQ, rsmSet);
+
+        Assert.assertEquals(mamQueryIQ.getDataForm(), dataForm);
+        Assert.assertEquals(mamQueryIQ.getDataForm().getFields().get(0).getValues().get(0), "urn:xmpp:mam:1");
+        Assert.assertEquals(mamQueryIQ.toXML().toString(), pagingStanza);
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/PreferencesTest.java

@@ -0,0 +1,68 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import java.lang.reflect.Method;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.junit.Test;
+import org.jxmpp.jid.Jid;
+import org.jxmpp.jid.impl.JidCreate;
+import org.jivesoftware.smackx.mam.element.MamElements;
+import org.jivesoftware.smackx.mam.element.MamPrefsIQ;
+import org.junit.Assert;
+
+public class PreferencesTest extends MamTest {
+
+    String retrievePrefsStanzaExample = "<iq id='sarasa' type='get'>" + "<prefs xmlns='" + MamElements.NAMESPACE
+            + "'></prefs>" + "</iq>";
+
+    String updatePrefsStanzaExample = "<iq id='sarasa' type='set'>" + "<prefs xmlns='" + MamElements.NAMESPACE
+            + "' default='roster'>" + "<always>" + "<jid>romeo@montague.lit</jid>" + "<jid>other@montague.lit</jid>"
+            + "</always>" + "<never>" + "<jid>montague@montague.lit</jid>" + "</never>" + "</prefs>" + "</iq>";
+
+    @Test
+    public void checkRetrievePrefsStanza() throws Exception {
+        Method prepareRetrievePreferencesStanza = MamManager.class
+                .getDeclaredMethod("prepareRetrievePreferencesStanza");
+        prepareRetrievePreferencesStanza.setAccessible(true);
+
+        MamPrefsIQ mamPrefIQ = (MamPrefsIQ) prepareRetrievePreferencesStanza.invoke(mamManager);
+        mamPrefIQ.setStanzaId("sarasa");
+        Assert.assertEquals(mamPrefIQ.toXML().toString(), retrievePrefsStanzaExample);
+    }
+
+    @Test
+    public void checkUpdatePrefsStanza() throws Exception {
+        Method prepareUpdatePreferencesStanza = MamManager.class.getDeclaredMethod("prepareUpdatePreferencesStanza",
+                List.class, List.class, String.class);
+        prepareUpdatePreferencesStanza.setAccessible(true);
+
+        List<Jid> alwaysJids = new ArrayList<>();
+        alwaysJids.add(JidCreate.from("romeo@montague.lit"));
+        alwaysJids.add(JidCreate.from("other@montague.lit"));
+
+        List<Jid> neverJids = new ArrayList<>();
+        neverJids.add(JidCreate.from("montague@montague.lit"));
+
+        MamPrefsIQ mamPrefIQ = (MamPrefsIQ) prepareUpdatePreferencesStanza.invoke(mamManager, alwaysJids, neverJids, "roster");
+        mamPrefIQ.setStanzaId("sarasa");
+        Assert.assertEquals(mamPrefIQ.toXML().toString(), updatePrefsStanzaExample);
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/QueryArchiveTest.java

@@ -0,0 +1,95 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import java.util.Date;
+import java.util.GregorianCalendar;
+import java.util.TimeZone;
+
+import org.jivesoftware.smack.packet.IQ;
+import org.jivesoftware.smack.packet.Message;
+import org.jivesoftware.smack.packet.Message.Type;
+import org.jivesoftware.smackx.delay.packet.DelayInformation;
+import org.jivesoftware.smackx.forward.packet.Forwarded;
+import org.jivesoftware.smackx.mam.element.MamElements;
+import org.jivesoftware.smackx.mam.element.MamElements.MamResultExtension;
+import org.jivesoftware.smackx.mam.element.MamQueryIQ;
+import org.jivesoftware.smackx.xdata.packet.DataForm;
+import org.junit.Assert;
+import org.junit.Test;
+import org.jxmpp.jid.impl.JidCreate;
+
+public class QueryArchiveTest extends MamTest {
+
+    String mamSimpleQueryIQ = "<iq id='sarasa' type='set'>" + "<query xmlns='urn:xmpp:mam:1' queryid='testid'>"
+            + "<x xmlns='jabber:x:data' type='submit'>" + "<field var='FORM_TYPE' type='hidden'>" + "<value>"
+            + MamElements.NAMESPACE + "</value>" + "</field>" + "</x>" + "</query>" + "</iq>";
+
+    String mamQueryResultExample = "<message to='hag66@shakespeare.lit/pda' from='coven@chat.shakespeare.lit' id='iasd207'>"
+            + "<result xmlns='urn:xmpp:mam:1' queryid='g27' id='34482-21985-73620'>"
+            + "<forwarded xmlns='urn:xmpp:forward:0'>"
+            + "<delay xmlns='urn:xmpp:delay' stamp='2002-10-13T23:58:37.000+00:00'></delay>" + "<message "
+            + "from='coven@chat.shakespeare.lit/firstwitch' " + "id='162BEBB1-F6DB-4D9A-9BD8-CFDCC801A0B2' "
+            + "type='chat'>" + "<body>Thrice the brinded cat hath mew.</body>" + "</message>" + "</forwarded>"
+            + "</result>" + "</message>";
+
+    @Test
+    public void checkMamQueryIQ() throws Exception {
+        DataForm dataForm = getNewMamForm();
+        MamQueryIQ mamQueryIQ = new MamQueryIQ(queryId, dataForm);
+        mamQueryIQ.setType(IQ.Type.set);
+        mamQueryIQ.setStanzaId("sarasa");
+        Assert.assertEquals(mamQueryIQ.toXML().toString(), mamSimpleQueryIQ);
+    }
+
+    @Test
+    public void checkMamQueryResults() throws Exception {
+        Message message = new Message();
+        message.setStanzaId("iasd207");
+        message.setFrom(JidCreate.from("coven@chat.shakespeare.lit"));
+        message.setTo(JidCreate.from("hag66@shakespeare.lit/pda"));
+
+        GregorianCalendar calendar = new GregorianCalendar(2002, 10 - 1, 13, 23, 58, 37);
+        calendar.setTimeZone(TimeZone.getTimeZone("UTC"));
+        Date date = calendar.getTime();
+
+        DelayInformation delay = new DelayInformation(date);
+        Message forwardedMessage = new Message();
+        forwardedMessage.setFrom(JidCreate.from("coven@chat.shakespeare.lit/firstwitch"));
+        forwardedMessage.setStanzaId("162BEBB1-F6DB-4D9A-9BD8-CFDCC801A0B2");
+        forwardedMessage.setType(Type.chat);
+        forwardedMessage.setBody("Thrice the brinded cat hath mew.");
+
+        Forwarded forwarded = new Forwarded(delay, forwardedMessage);
+
+        message.addExtension(new MamResultExtension("g27", "34482-21985-73620", forwarded));
+
+        Assert.assertEquals(message.toXML().toString(), mamQueryResultExample);
+
+        MamResultExtension mamResultExtension = MamResultExtension.from(message);
+
+        Assert.assertEquals(mamResultExtension.getId(), "34482-21985-73620");
+        Assert.assertEquals(mamResultExtension.getForwarded().getDelayInformation().getStamp(), date);
+
+        Message resultMessage = (Message) mamResultExtension.getForwarded().getForwardedStanza();
+        Assert.assertEquals(resultMessage.getFrom(), JidCreate.from("coven@chat.shakespeare.lit/firstwitch"));
+        Assert.assertEquals(resultMessage.getStanzaId(), "162BEBB1-F6DB-4D9A-9BD8-CFDCC801A0B2");
+        Assert.assertEquals(resultMessage.getType(), Type.chat);
+        Assert.assertEquals(resultMessage.getBody(), "Thrice the brinded cat hath mew.");
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/ResultsLimitTest.java

@@ -0,0 +1,50 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import java.lang.reflect.Method;
+
+import org.jivesoftware.smack.packet.IQ;
+import org.jivesoftware.smackx.mam.element.MamElements;
+import org.jivesoftware.smackx.mam.element.MamQueryIQ;
+import org.jivesoftware.smackx.xdata.packet.DataForm;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class ResultsLimitTest extends MamTest {
+
+    String resultsLimitStanza = "<iq id='sarasa' type='set'>" + "<query xmlns='urn:xmpp:mam:1' queryid='testid'>"
+            + "<x xmlns='jabber:x:data' type='submit'>" + "<field var='FORM_TYPE' type='hidden'>" + "<value>"
+            + MamElements.NAMESPACE + "</value>" + "</field>" + "</x>" + "<set xmlns='http://jabber.org/protocol/rsm'>"
+            + "<max>10</max>" + "</set>" + "</query>" + "</iq>";
+
+    @Test
+    public void checkResultsLimit() throws Exception {
+        Method methodAddResultsLimit = MamManager.class.getDeclaredMethod("addResultsLimit", Integer.class,
+                MamQueryIQ.class);
+        methodAddResultsLimit.setAccessible(true);
+
+        DataForm dataForm = getNewMamForm();
+        MamQueryIQ mamQueryIQ = new MamQueryIQ(queryId, dataForm);
+        mamQueryIQ.setType(IQ.Type.set);
+        mamQueryIQ.setStanzaId("sarasa");
+
+        methodAddResultsLimit.invoke(mamManager, 10, mamQueryIQ);
+        Assert.assertEquals(mamQueryIQ.toXML().toString(), resultsLimitStanza);
+    }
+
+}

smack-experimental/src/test/java/org/jivesoftware/smackx/mam/RetrieveFormFieldsTest.java

@@ -0,0 +1,81 @@
+/**
+ *
+ * Copyright 2016 Fernando Ramirez
+ *
+ * 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.mam;
+
+import java.lang.reflect.Method;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.jivesoftware.smackx.mam.element.MamElements;
+import org.jivesoftware.smackx.mam.element.MamQueryIQ;
+import org.jivesoftware.smackx.xdata.FormField;
+import org.jivesoftware.smackx.xdata.packet.DataForm;
+import org.junit.Test;
+
+import org.junit.Assert;
+
+public class RetrieveFormFieldsTest extends MamTest {
+
+    String retrieveFormFieldStanza = "<iq id='sarasa' type='get'>" + "<query xmlns='" + MamElements.NAMESPACE
+            + "' queryid='testid'></query>" + "</iq>";
+
+    String additionalFieldsStanza = "<x xmlns='jabber:x:data' type='submit'>" + "<field var='FORM_TYPE' type='hidden'>"
+            + "<value>" + MamElements.NAMESPACE + "</value>" + "</field>"
+            + "<field var='urn:example:xmpp:free-text-search' type='text-single'>" + "<value>Hi</value>" + "</field>"
+            + "<field var='urn:example:xmpp:stanza-content' type='jid-single'>" + "<value>Hi2</value>" + "</field>"
+            + "</x>";
+
+    @Test
+    public void checkRetrieveFormFieldsStanza() throws Exception {
+        Method methodPrepareMamQueryIQGet = MamManager.class.getDeclaredMethod("prepareMamQueryIQGet", String.class);
+        methodPrepareMamQueryIQGet.setAccessible(true);
+
+        MamQueryIQ mamQueryIQ = (MamQueryIQ) methodPrepareMamQueryIQGet.invoke(mamManager, queryId);
+        mamQueryIQ.setStanzaId("sarasa");
+
+        Assert.assertEquals(mamQueryIQ.toXML().toString(), retrieveFormFieldStanza);
+    }
+
+    @Test
+    public void checkAddAdditionalFieldsStanza() throws Exception {
+        Method methodAddAdditionalFields = MamManager.class.getDeclaredMethod("addAdditionalFields", List.class,
+                DataForm.class);
+        methodAddAdditionalFields.setAccessible(true);
+
+        DataForm dataForm = getNewMamForm();
+
+        List<FormField> additionalFields = new ArrayList<>();
+
+        FormField field1 = new FormField("urn:example:xmpp:free-text-search");
+        field1.setType(FormField.Type.text_single);
+        field1.addValue("Hi");
+
+        FormField field2 = new FormField("urn:example:xmpp:stanza-content");
+        field2.setType(FormField.Type.jid_single);
+        field2.addValue("Hi2");
+
+        additionalFields.add(field1);
+        additionalFields.add(field2);
+
+        methodAddAdditionalFields.invoke(mamManager, additionalFields, dataForm);
+
+        String dataFormResult = dataForm.toXML().toString();
+
+        Assert.assertEquals(dataFormResult, additionalFieldsStanza);
+    }
+
+}