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

/**
 *
 * 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 static 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(null).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(null).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(null).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(null).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));
    }

}