/*
 *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.org) All Rights Reserved.
 *
 *  WSO2 LLC. licenses this file to you 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.apache.synapse.util.synapse.expression;

import com.google.gson.GsonBuilder;
import com.jayway.jsonpath.Configuration;
import com.jayway.jsonpath.Option;
import com.jayway.jsonpath.spi.json.GsonJsonProvider;
import com.jayway.jsonpath.spi.json.JsonProvider;
import com.jayway.jsonpath.spi.mapper.GsonMappingProvider;
import com.jayway.jsonpath.spi.mapper.MappingProvider;
import org.apache.synapse.commons.json.JsonUtil;
import org.apache.synapse.core.axis2.Axis2MessageContext;
import org.apache.synapse.mediators.TestUtils;
import org.apache.synapse.util.xpath.SynapseExpression;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import java.util.EnumSet;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * Test class for header and property access.
 */
public class HeaderAndPropertyAccessTest {

    private Axis2MessageContext synCtx;

    @Before
    public void setUp() throws Exception {
        Configuration.setDefaults(new Configuration.Defaults() {
            private final JsonProvider jsonProvider = new GsonJsonProvider(new GsonBuilder().serializeNulls().create());
            private final MappingProvider mappingProvider = new GsonMappingProvider();

            public JsonProvider jsonProvider() {
                return jsonProvider;
            }

            public MappingProvider mappingProvider() {
                return mappingProvider;
            }

            public Set<Option> options() {
                return EnumSet.noneOf(Option.class);
            }
        });
        synCtx = TestUtils.getAxis2MessageContext("<test/>", null);
        Map<String, String> headersMap = new HashMap<>();
        headersMap.put("test", "Hello World");
        headersMap.put("toUpper", "HELLO WORLD");
        headersMap.put("te st", "HELLO");
        headersMap.put("numerical", "400");
        headersMap.put("price", "10");
        headersMap.put("payload", "Hello World");
        synCtx.getAxis2MessageContext().setProperty(
                org.apache.axis2.context.MessageContext.TRANSPORT_HEADERS, headersMap);
        synCtx.getAxis2MessageContext().setProperty("test", "Sic Mundus");
        synCtx.getAxis2MessageContext().setProperty("test2", "Creatus Est");
        synCtx.getAxis2MessageContext().setProperty("empty", "Thus the world was created");
        synCtx.getAxis2MessageContext().setProperty("category", "biography");
        synCtx.getAxis2MessageContext().setProperty("payload", "Hello");
        synCtx.setProperty("phrase", "Now I Am Become Death, the Destroyer of Worlds");
        synCtx.setProperty("selected book", "Animal Farm");
        synCtx.setProperty("null", null);
        synCtx.setProperty("empty", "");
        JsonUtil.getNewJsonPayload(synCtx.getAxis2MessageContext(), "{\n" +
                "  \"store\": {\n" +
                "    \"book\": [\n" +
                "      {\n" +
                "        \"category\": \"reference\",\n" +
                "        \"author\": \"Nigel Rees\",\n" +
                "        \"title\": \"Sayings of the Century\",\n" +
                "        \"price\": 8.95\n" +
                "      },\n" +
                "      {\n" +
                "        \"category\": \"fiction\",\n" +
                "        \"author\": \"Herman Melville\",\n" +
                "        \"title\": \"Moby Dick\",\n" +
                "        \"isbn\": \"0-553-21311-3\",\n" +
                "        \"price\": 8.99\n" +
                "      },\n" +
                "      {\n" +
                "        \"category\": \"fiction\",\n" +
                "        \"author\": \"J.R.R. Tolkien\",\n" +
                "        \"title\": \"The Lord of the Rings\",\n" +
                "        \"isbn\": \"0-395-19395-8\",\n" +
                "        \"price\": 22.99\n" +
                "      },\n" +
                "      {\n" +
                "        \"category\": \"fiction\",\n" +
                "        \"author\": \"Harper Lee\",\n" +
                "        \"title\": \"To Kill a Mockingbird\",\n" +
                "        \"price\": 10.99\n" +
                "      },\n" +
                "      {\n" +
                "        \"category\": \"fiction\",\n" +
                "        \"author\": \"George Orwell\",\n" +
                "        \"title\": \"Animal Farm\",\n" +
                "        \"price\": 7.99\n" +
                "      },\n" +
                "      {\n" +
                "        \"category\": \"biography\",\n" +
                "        \"author\": \"Anne Frank\",\n" +
                "        \"title\": \"The Diary of a Young Girl\",\n" +
                "        \"price\": 6.99\n" +
                "      }\n" +
                "    ],\n" +
                "    \"bicycle\": {\n" +
                "      \"color\": \"red\",\n" +
                "      \"price\": 19.95\n" +
                "    }\n" +
                "  },\n" +
                "  \"expensive\": 10,\n" +
                "  \"selectedCategory\": \"biography\"\n" +
                "}\n", true, true);
    }

    @Test
    public void testTransportHeader() throws Exception {
        SynapseExpression testPath = new SynapseExpression("headers.test");
        Assert.assertEquals("Hello World", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testTransportHeaderWithReservedName() throws Exception {
        SynapseExpression testPath = new SynapseExpression("toLower(headers.[\"toUpper\"])");
        Assert.assertEquals("hello world", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testTransportHeaderInFilter() throws Exception {
        SynapseExpression testPath = new SynapseExpression("$..book[?(@.price > integer(headers.price))].title");
        Assert.assertEquals("[\"The Lord of the Rings\",\"To Kill a Mockingbird\"]", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testTransportHeaderWithSpecialChar() throws Exception {
        SynapseExpression testPath = new SynapseExpression("toLower(headers.[\"te st\"]) + \" World\"");
        Assert.assertEquals("hello World", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testTransportHeaderNotExist() throws Exception {
        SynapseExpression testPath = new SynapseExpression("headers[\"toUpper2\"]");
        Assert.assertEquals("", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testTransportHeaderNumerical() throws Exception {
        SynapseExpression testPath = new SynapseExpression("integer(headers.numerical)+ 4");
        Assert.assertEquals("404", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testAxis2Header() throws Exception {
        SynapseExpression testPath = new SynapseExpression("props.axis2.test");
        Assert.assertEquals("Sic Mundus", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testAxis2HeaderWithExpression() throws Exception {
        SynapseExpression testPath = new SynapseExpression("properties.axis2.test + \" \" " +
                "+ props.axis2.test2");
        Assert.assertEquals("Sic Mundus Creatus Est", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testAxis2HeaderInFilter() throws Exception {
        SynapseExpression testPath = new SynapseExpression("$..book[?(@.category == props.axis2.category)].title");
        Assert.assertEquals("[\"The Diary of a Young Girl\"]", testPath.stringValueOf(synCtx));
    }
    @Test
    public void testAxis2HeaderWithReservedName() throws Exception {
        SynapseExpression testPath = new SynapseExpression("props.axis2.[\"empty\"]");
        Assert.assertEquals("Thus the world was created", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testSynapseHeader() throws Exception {
        SynapseExpression testPath = new SynapseExpression("props.synapse.phrase");
        Assert.assertEquals("Now I Am Become Death, the Destroyer of Worlds", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testSynapseHeaderWithFunction() throws Exception {
        SynapseExpression testPath = new SynapseExpression("length(split(properties.synapse.phrase, \" \"))");
        Assert.assertEquals("9", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testSynapseHeaderInFilter() throws Exception {
        SynapseExpression testPath = new SynapseExpression("$..book[?(@.title == props.synapse.[\"selected book\"])].price");
        Assert.assertEquals("[7.99]", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testNonExistingEmptyAndNull() throws Exception {
        SynapseExpression testPath = new SynapseExpression("props.synapse.nonExisting");
        Assert.assertEquals("", testPath.stringValueOf(synCtx));
        testPath = new SynapseExpression("props.synapse.[\"null\"]");
        Assert.assertEquals("", testPath.stringValueOf(synCtx));
        testPath = new SynapseExpression("props.synapse[\"empty\"]");
        Assert.assertEquals("", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testHeaderWithKeyword() throws Exception {
        SynapseExpression testPath = new SynapseExpression("headers.payload");
        Assert.assertEquals("Hello World", testPath.stringValueOf(synCtx));
    }

    @Test
    public void testAxis2PropertyWithKeyword() throws Exception {
        SynapseExpression testPath = new SynapseExpression("props.axis2.payload");
        Assert.assertEquals("Hello", testPath.stringValueOf(synCtx));
    }
}
