/*
 * Copyright (c) 2011, 2021 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0,
 * or the Eclipse Distribution License v. 1.0 which is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause
 */

// Contributors:
//     David McCann - Aug.15, 2012 - 2.4.1 - Initial implementation
package dbws.testing.mtom;

import java.sql.SQLException;
import java.sql.Statement;
import java.util.Iterator;

import javax.xml.namespace.QName;
import jakarta.xml.soap.AttachmentPart;

import jakarta.xml.soap.SOAPElement;
import jakarta.xml.soap.SOAPFactory;
import jakarta.xml.soap.SOAPMessage;
import jakarta.xml.ws.Dispatch;
import jakarta.xml.ws.Service;
import jakarta.xml.ws.soap.SOAPBinding;

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;

import dbws.testing.DBWSTestSuite;
import static dbws.testing.mtom.MTOMBuilderTestSuite.CREATE_TABLE;
import static dbws.testing.mtom.MTOMBuilderTestSuite.DROP_TABLE;
import static dbws.testing.mtom.MTOMBuilderTestSuite.POPULATE_TABLE;

import static org.junit.Assert.fail;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertEquals;

/**
 * Tests MTOM.
 *
 */
public class MTOMServiceTestSuite extends DBWSTestSuite {
    static final String SOAP_FINDBYPK_REQUEST =
        "<env:Envelope xmlns:env=\"http://schemas.xmlsoap.org/soap/envelope/\">" +
          "<env:Body>" +
            "<findByPrimaryKey_MtomType xmlns=\"urn:mtomService\" xmlns:urn=\"urn:mtom\">" +
              "<id>3</id>" +
            "</findByPrimaryKey_MtomType>" +
          "</env:Body>" +
        "</env:Envelope>";
    static final String SOAP_FINDALL_REQUEST =
        "<env:Envelope xmlns:env=\"http://schemas.xmlsoap.org/soap/envelope/\">" +
          "<env:Body>" +
            "<findAll_MtomType xmlns=\"urn:mtomService\" xmlns:urn=\"urn:mtom\"/>" +
          "</env:Body>" +
        "</env:Envelope>";

    @BeforeClass
    public static void setUp() {
        if (conn == null) {
            try {
                conn = buildConnection();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        if (ddlCreate) {
            runDdl(conn, CREATE_TABLE, ddlDebug);
            try {
                Statement stmt = conn.createStatement();
                for (int i = 0; i < POPULATE_TABLE.length; i++) {
                    stmt.addBatch(POPULATE_TABLE[i]);
                }
                stmt.executeBatch();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }

    @AfterClass
    public static void tearDown() {
        if (ddlDrop) {
            runDdl(conn, DROP_TABLE, ddlDebug);
        }
    }

    @Test
    public void testService() {
        try {
            QName qname = new QName("urn:mtomService", "mtomServicePort");
            Service service = Service.create(new QName("urn:mtom", "mtomService"));
            service.addPort(qname, SOAPBinding.SOAP11HTTP_MTOM_BINDING, "http://" + host + ":" + port + "/mtom/mtom");
            Dispatch<SOAPMessage> sourceDispatch = service.createDispatch(qname, SOAPMessage.class, Service.Mode.MESSAGE);

            // TEST FINDALL
            // we expect 3 attachments
            SOAPMessage request = createSOAPMessage(SOAP_FINDALL_REQUEST);
            SOAPMessage response = sourceDispatch.invoke(request);
            assertNotNull("findAll failed:  response is null.", response);
            assertTrue("findAll failed:  wrong number of attachments - expected [3] but was [" + response.countAttachments() + "]", response.countAttachments() == 3);

            // verify MTOM format, i.e.
            // <xop:Include xmlns:xop="http://www.w3.org/2004/08/xop/include" href="cid:c060bfb1-82cb-4820-9d87-4f2422b50915"/>
            for (@SuppressWarnings("unchecked")
                 Iterator<AttachmentPart> attachmentsIterator = response.getAttachments(); attachmentsIterator.hasNext();) {
                AttachmentPart ap = attachmentsIterator.next();

                SOAPElement elt = SOAPFactory.newInstance().createElement(new QName("http://www.w3.org/2004/08/xop/include", "Include", "xop"));
                // content id will be wrapped in angled brackets - need to remove them
                String contentId = ap.getContentId();
                contentId = contentId.replaceFirst("<", "");
                contentId = contentId.replaceFirst(">", "");
                elt.addAttribute(new QName("href"), "cid:" + contentId);

                ap = response.getAttachment(elt);
                assertNotNull("findAll failed:  no attachment for [cid:" + contentId + "]", ap);

                byte[] rawBytes = ap.getRawContentBytes() ;
                assertTrue("findAll failed:  wrong number of bytes returned - expected [15] but was [" + rawBytes.length + "]", rawBytes.length == 15);

                // no order is guaranteed, so need to check which attachment we are dealing with
                byte b = rawBytes[0];
                assertTrue("findAll failed:  wrong byte value returned - expected [1], [2] or [3] but was [" + b + "]", (b==1 || b==2 || b==3));
                compareBytes(b, rawBytes, "findAll");
                elt.removeContents();
            }
            // TEST FINDBYPK
            // we expect 1 attachment
            request = createSOAPMessage(SOAP_FINDBYPK_REQUEST);
            response = sourceDispatch.invoke(request);
            assertTrue("findByPk failed:  wrong number of attachments - expected [1] but was [" + response.countAttachments() + "]", response.countAttachments() == 1);

            @SuppressWarnings("rawtypes")
            AttachmentPart ap = (AttachmentPart)((Iterator)response.getAttachments()).next();
            SOAPElement elt = SOAPFactory.newInstance().createElement(new QName("http://www.w3.org/2004/08/xop/include", "Include", "xop"));
            // content id will be wrapped in angled brackets - need to remove them
            String contentId = ap.getContentId();
            contentId = contentId.replaceFirst("<", "");
            contentId = contentId.replaceFirst(">", "");
            elt.addAttribute(new QName("href"), "cid:" + contentId);

            ap = response.getAttachment(elt);
            assertNotNull("findByPk failed:  no attachment for [cid:" + contentId + "]", ap);

            byte[] rawBytes = ap.getRawContentBytes() ;
            assertTrue("findByPk failed:  wrong number of bytes returned - expected [15] but was [" + rawBytes.length + "]", rawBytes.length == 15);
            compareBytes(3, rawBytes, "findByPk");
        } catch (Exception x) {
            fail("Service test failed: " + x.getMessage());
        }
    }

    static void compareBytes(int intVal, byte[] bytes, String testName) {
        for (int i = 0; i < bytes.length; i++) {
            assertEquals(testName + " failed:  wrong byte value returned - expected [" + intVal + "] but was [" + bytes[i] + "]", bytes[i], intVal);
        }
    }
}