/**
 *
 * Copyright 2017 Paul Schaub
 *
 * 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.jingle_filetransfer;

import static junit.framework.TestCase.assertEquals;
import static junit.framework.TestCase.assertNull;
import static org.custommonkey.xmlunit.XMLAssert.assertXMLEqual;

import java.util.Collections;
import java.util.Date;

import org.jivesoftware.smack.DummyConnection;
import org.jivesoftware.smack.XMPPConnection;
import org.jivesoftware.smack.test.util.SmackTestSuite;
import org.jivesoftware.smack.test.util.TestUtils;
import org.jivesoftware.smackx.hashes.HashManager;
import org.jivesoftware.smackx.hashes.element.HashElement;
import org.jivesoftware.smackx.jingle.JingleManager;
import org.jivesoftware.smackx.jingle.JingleUtil;
import org.jivesoftware.smackx.jingle.JingleUtilTest;
import org.jivesoftware.smackx.jingle3.element.JingleContentElement;
import org.jivesoftware.smackx.jingle3.element.JingleElement;
import org.jivesoftware.smackx.jingle3.element.JingleAction;
import org.jivesoftware.smackx.jingle3.element.JingleContentDescriptionChildElement;
import org.jivesoftware.smackx.jingle.transports.jingle_ibb.element.JingleIBBTransport;
import org.jivesoftware.smackx.jingle_filetransfer.element.JingleFileTransfer;
import org.jivesoftware.smackx.jingle_filetransfer.element.JingleFileTransferChild;
import org.jivesoftware.smackx.jingle_filetransfer.provider.JingleFileTransferProvider;

import org.junit.Before;
import org.junit.Test;
import org.jxmpp.jid.FullJid;
import org.jxmpp.jid.impl.JidCreate;
import org.jxmpp.stringprep.XmppStringprepException;
import org.jxmpp.util.XmppDateTime;

/**
 * Created by vanitas on 12.07.17.
 */
public class JingleUtilFileTransferTest extends SmackTestSuite {
    private XMPPConnection connection;
    private JingleUtil jutil;

    private FullJid romeo;
    private FullJid juliet;

    @Before
    public void setup() throws XmppStringprepException {
        connection = new DummyConnection(
                DummyConnection.getDummyConfigurationBuilder()
                        .setUsernameAndPassword("romeo@montague.lit",
                                "iluvJulibabe13").build());
        JingleManager jm = JingleManager.getInstanceFor(connection);
        jutil = new JingleUtil(connection);
        romeo = connection.getUser().asFullJidOrThrow();
        juliet = JidCreate.fullFrom("juliet@capulet.lit/balcony");
    }

    @Test
    public void createSessionInitiateTest() throws Exception {
        JingleIBBTransport transport = new JingleIBBTransport("transid");
        Date date = new Date();
        HashElement hash = new HashElement(HashManager.ALGORITHM.SHA_256, "f4OxZX/x/FO5LcGBSKHWXfwtSx+j1ncoSt3SABJtkGk=");
        JingleFileTransferChild file = new JingleFileTransferChild(date, "desc", hash, "application/octet-stream", "name", 1337, null);
        JingleFileTransfer description = new JingleFileTransfer(Collections.<JingleContentDescriptionChildElement>singletonList(file));

        String contentName = "content";

        JingleElement initiate = jutil.createSessionInitiate(juliet, "letsstart", JingleContentElement.Creator.initiator, contentName, JingleContentElement.Senders.initiator, description, transport);
        JingleElement accept = jutil.createSessionAccept(juliet, "acceptID", JingleContentElement.Creator.initiator, contentName, JingleContentElement.Senders.initiator, description, transport);
        JingleElement fileOffer = jutil.createSessionInitiateFileOffer(juliet, "fileOffer", JingleContentElement.Creator.initiator, contentName, description, transport);

        assertEquals(JingleAction.session_initiate, initiate.getAction());
        assertEquals(JingleAction.session_accept, accept.getAction());

        assertEquals(romeo, initiate.getInitiator());
        assertEquals(romeo, accept.getResponder());
        //Must be null
        assertNull(initiate.getResponder());
        assertNull(accept.getInitiator());

        assertEquals("letsstart", initiate.getSid());
        assertEquals("acceptID", accept.getSid());

        assertEquals(1, initiate.getContents().size());
        assertEquals(1, accept.getContents().size());

        JingleContentElement content = initiate.getContents().get(0);
        assertEquals(content.toXML().toString(), initiate.getContents().get(0).toXML().toString());
        assertEquals(content.toXML().toString(), accept.getContents().get(0).toXML().toString());

        assertEquals("content", content.getName());
        assertEquals(JingleContentElement.Creator.initiator, content.getCreator());
        assertEquals(JingleContentElement.Senders.initiator, content.getSenders());

        assertEquals(1, description.getJingleContentDescriptionChildren().size());
        assertEquals(file, description.getJingleContentDescriptionChildren().get(0));
        assertEquals(JingleFileTransferChild.ELEMENT, file.getElementName());
        assertEquals(JingleFileTransfer.NAMESPACE_V5, description.getNamespace());
        assertEquals(date, file.getDate());
        assertEquals(hash, file.getHash());
        assertEquals("application/octet-stream", file.getMediaType());
        assertEquals("name", file.getName());
        assertEquals(1337, file.getSize());
        assertNull(file.getRange());

        assertEquals(transport, content.getTransport());
        assertEquals("transid", transport.getSessionId());
        assertEquals(JingleIBBTransport.DEFAULT_BLOCK_SIZE, transport.getBlockSize());

        String transportXML =
                "<transport xmlns='urn:xmpp:jingle:transports:ibb:1' " +
                "block-size='4096' " +
                "sid='transid'/>";
        assertXMLEqual(transportXML, transport.toXML().toString());

        String descriptionXML =
                "<description xmlns='urn:xmpp:jingle:apps:file-transfer:5'>" +
                "<file>" +
                "<date>" + XmppDateTime.formatXEP0082Date(date) + "</date>" +
                "<desc>desc</desc>" +
                "<media-type>application/octet-stream</media-type>" +
                "<name>name</name>" +
                //"<range/>" + TODO: insert empty element when null?
                "<size>1337</size>" +
                "<hash xmlns='urn:xmpp:hashes:2' " +
                "algo='sha-256'>f4OxZX/x/FO5LcGBSKHWXfwtSx+j1ncoSt3SABJtkGk=</hash>" +
                "</file>" +
                "</description>";
        assertXMLEqual(descriptionXML, description.toXML().toString());

        JingleFileTransfer parsed = new JingleFileTransferProvider().parse(TestUtils.getParser(descriptionXML));
        assertEquals(1, parsed.getJingleContentDescriptionChildren().size());
        assertEquals(file.toXML().toString(), parsed.getJingleContentDescriptionChildren().get(0).toXML().toString());

        String contentXML = "<content creator='initiator' name='content' senders='initiator'>" +
                descriptionXML +
                transportXML +
                "</content>";
        assertXMLEqual(contentXML, content.toXML().toString());

        String initiateXML =
                "<jingle xmlns='urn:xmpp:jingle:1' " +
                        "action='session-initiate' " +
                        "initiator='" + romeo + "' " +
                        "sid='letsstart'>" +
                        contentXML +
                        "</jingle>";
        String xml = JingleUtilTest.getIQXML(romeo, juliet, initiate.getStanzaId(), initiateXML);
        assertXMLEqual(xml, initiate.toXML().toString());

        String acceptXML =
                "<jingle xmlns='urn:xmpp:jingle:1' " +
                "action='session-accept' " +
                "responder='" + romeo + "' " +
                "sid='acceptID'>" +
                contentXML +
                "</jingle>";
        xml = JingleUtilTest.getIQXML(romeo, juliet, accept.getStanzaId(), acceptXML);
        assertXMLEqual(xml, accept.toXML().toString());

        String fileOfferXML =
                "<jingle xmlns='urn:xmpp:jingle:1' " +
                        "action='session-initiate' " +
                        "initiator='" + romeo + "' " +
                        "sid='fileOffer'>" +
                        contentXML +
                        "</jingle>";
        xml = JingleUtilTest.getIQXML(romeo, juliet, fileOffer.getStanzaId(), fileOfferXML);
        assertXMLEqual(xml, fileOffer.toXML().toString());
    }
}