/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * 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.
 * ============LICENSE_END=========================================================
 */
package org.onap.aai.rest.dsl;

import static org.junit.Assert.assertEquals;

import org.junit.Ignore;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
import org.onap.aai.AAISetup;
import org.onap.aai.exceptions.AAIException;
import org.onap.aai.rest.enums.QueryVersion;
import org.springframework.beans.factory.annotation.Autowired;

/**
 * The Class DslMain.
 */
public class DslQueryProcessorV2Test extends AAISetup {

    @Autowired
    V2DslQueryProcessor dslQueryProcessor;

    @Rule
    public ExpectedException expectedEx = ExpectedException.none();

    @Test
    public void dbAliasTest() throws AAIException {
        String aaiQuery = "logical-link* ('model-invariant-id','invid')";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'logical-link').getVerticesByProperty('model-invariant-id-local','invid')"
                + ".store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void apostropheTest() throws AAIException {
        String aaiQuery = "logical-link*('link-id','dsl\\'link')";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'logical-link').getVerticesByProperty('link-id','dsl\\'link')"
                + ".store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void cloudRegionFromVnf() throws AAIException {
        String aaiQuery =
            "generic-vnf*('vnf-name','xyz')  [> vnfc* > vserver*  [>pserver*, > tenant* > cloud-region*], "
                + "> vserver*  [> pserver*, >tenant* > cloud-region*] ]";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-name','xyz').store('x').union(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vnfc').store('x')."
                + "createEdgeTraversal(EdgeType.COUSIN, 'vnfc','vserver').store('x').union(builder.newInstance().createEdgeTraversal"
                + "(EdgeType.COUSIN, 'vserver','pserver').store('x'),builder.newInstance().createEdgeTraversal(EdgeType.TREE, 'vserver','tenant')"
                + ".store('x').createEdgeTraversal(EdgeType.TREE, 'tenant','cloud-region').store('x')),builder.newInstance().createEdgeTraversal"
                + "(EdgeType.COUSIN, 'generic-vnf','vserver').store('x').union(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'vserver'"
                + ",'pserver').store('x'),builder.newInstance().createEdgeTraversal(EdgeType.TREE, 'vserver','tenant').store('x').createEdgeTraversal"
                + "(EdgeType.TREE, 'tenant','cloud-region').store('x'))).cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void cloudRegionFromVnfWithDirection() throws AAIException {
        String aaiQuery =
            "generic-vnf*('vnf-name','xyz')  [>> vnfc* >> vserver*  [>>pserver*, >> tenant* >> cloud-region*], "
                + ">> vserver*  [>> pserver*, >>tenant* >> cloud-region*] ]";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-name','xyz').store('x').union(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vnfc').store('x')."
                + "createEdgeTraversal(EdgeType.COUSIN, 'vnfc','vserver').store('x').union(builder.newInstance().createEdgeTraversal"
                + "(EdgeType.COUSIN, 'vserver','pserver').store('x'),builder.newInstance().createEdgeTraversal(EdgeType.TREE, 'vserver','tenant')"
                + ".store('x').createEdgeTraversal(EdgeType.TREE, 'tenant','cloud-region').store('x')),builder.newInstance().createEdgeTraversal"
                + "(EdgeType.COUSIN, 'generic-vnf','vserver').store('x').union(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'vserver'"
                + ",'pserver').store('x'),builder.newInstance().createEdgeTraversal(EdgeType.TREE, 'vserver','tenant').store('x').createEdgeTraversal"
                + "(EdgeType.TREE, 'tenant','cloud-region').store('x'))).cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void cloudRegionSites() throws AAIException {
        String aaiQuery = "cloud-region*('cloud-owner','xyz') > complex*";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-owner'"
                + ",'xyz').store('x').createEdgeTraversal(EdgeType.COUSIN, 'cloud-region','complex').store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void pserverWithNoComplexTest() throws AAIException {
        String aaiQuery = "pserver*('hostname','xyz')!(> complex)";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'pserver').getVerticesByProperty('hostname','xyz').where(builder.newInstance().not("
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'pserver','complex'))).store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void pserverWhereNotTest() throws AAIException {
        String aaiQuery = "pserver('hostname','xyz')>vserver*!(> vnfc > configuration))";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'pserver').getVerticesByProperty('hostname','xyz').createEdgeTraversal"
                + "(EdgeType.COUSIN, 'pserver','vserver').where(builder.newInstance().not(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN,"
                + " 'vserver','vnfc').createEdgeTraversal(EdgeType.COUSIN, 'vnfc','configuration'))).store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void singleNode1() throws AAIException {
        String aaiQuery = "cloud-region* !('cloud-owner','coid')";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesExcludeByProperty('cloud-owner','coid')"
                + ".store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void singleNodeLimit() throws AAIException {
        String aaiQuery = "cloud-region* !('cloud-owner','coid') LIMIT 10";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesExcludeByProperty('cloud-owner','coid')"
                + ".store('x').cap('x').unfold().dedup().limit(10)";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void specialCharacterTest() throws AAIException {
        String aaiQuery =
            "cloud-region* !('cloud-owner','coidhello:?_-)(!@#$%^&*+={}[]|/.<,') LIMIT 10";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesExcludeByProperty('cloud-owner','coidhello:?_-)(!@#$%^&*+={}[]|/.<,')"
                + ".store('x').cap('x').unfold().dedup().limit(10)";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void singleNodeLimitBlah() throws AAIException {
        String aaiQuery = "cloud-region* !('cloud-owner','coid') LIMIT blah";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesExcludeByProperty('cloud-owner','coid')"
                + ".getVerticesByProperty('cloud-region-id','cr id').store('x').cap('x').unfold().dedup().limit(10)";

        expectedEx.expect(AAIException.class);
        expectedEx.expectMessage("DSL Syntax Error while processing the query");
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
    }

    @Test
    public void singleNodeLimitNull() throws AAIException {
        String aaiQuery = "cloud-region* !('cloud-owner','coid') LIMIT ";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesExcludeByProperty('cloud-owner','coid')"
                + ".getVerticesByProperty('cloud-region-id','cr id').store('x').cap('x').unfold().dedup().limit(10)";

        expectedEx.expect(AAIException.class);
        expectedEx.expectMessage("DSL Syntax Error while processing the query");

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
    }

    @Test
    public void cloudRegion1Test() throws AAIException {
        String aaiQuery =
            "cloud-region* !('cloud-owner','coid')('cloud-region-id','cr id')  LIMIT 10";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesExcludeByProperty('cloud-owner','coid')"
                + ".getVerticesByProperty('cloud-region-id','cr id').store('x').cap('x').unfold().dedup().limit(10)";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void cloudRegion_entitlementTest() throws AAIException {

        /*
         * A store within a where makes no sense
         */
        String aaiQuery =
            "generic-vnf('vnf-id','vnfId') ( > vserver > tenant > cloud-region('cloud-region-id','One')) > entitlement*";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-id','vnfId').where("
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vserver').createEdgeTraversal(EdgeType.TREE, 'vserver','tenant').createEdgeTraversal(EdgeType.TREE, 'tenant','cloud-region')"
                + ".getVerticesByProperty('cloud-region-id','One'))"
                + ".createEdgeTraversal(EdgeType.TREE, 'generic-vnf','entitlement').store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void cloudRegion_entitlementTestWithLabels() throws AAIException {

        String aaiQuery =
            "generic-vnf('vnf-id','vnfId') (> ('tosca.relationships.HostedOn') vserver > ('org.onap.relationships.inventory.BelongsTo') tenant > ('org.onap.relationships.inventory.BelongsTo') cloud-region('cloud-region-id','One')) > ('org.onap.relationships.inventory.ComposedOf')service-instance*";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-id','vnfId').where("
                + "builder.newInstance().createEdgeTraversalWithLabels( 'generic-vnf','vserver', new ArrayList<>(Arrays.asList('tosca.relationships.HostedOn'))).createEdgeTraversalWithLabels( 'vserver','tenant', new ArrayList<>(Arrays.asList('org.onap.relationships.inventory.BelongsTo'))).createEdgeTraversalWithLabels( 'tenant','cloud-region', new ArrayList<>(Arrays.asList('org.onap.relationships.inventory.BelongsTo')))"
                + ".getVerticesByProperty('cloud-region-id','One'))"
                + ".createEdgeTraversalWithLabels( 'generic-vnf','service-instance', new ArrayList<>(Arrays.asList('org.onap.relationships.inventory.ComposedOf'))).store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void complex_az_fromComplexTest() throws AAIException {

        String aaiQuery =
            "cloud-region('cloud-owner','coid')('cloud-region-id','crid')   [>  availability-zone* ,>  complex*]";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        String dslQuery = "builder.getVerticesByProperty('aai-node-type', 'cloud-region')"
            + ".getVerticesByProperty('cloud-owner','coid').getVerticesByProperty('cloud-region-id','crid')"
            + ".union(builder.newInstance().createEdgeTraversal(EdgeType.TREE, 'cloud-region','availability-zone').store('x')"
            + ",builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'cloud-region','complex').store('x')).cap('x').unfold().dedup()";

        assertEquals(dslQuery, query);
    }

    @Test
    public void complex_az_fromComplexTestWithLabels() throws AAIException {

        String aaiQuery =
            "cloud-region('cloud-owner','coid')('cloud-region-id','crid')   [>  ('org.onap.relationships.inventory.BelongsTo')availability-zone* ,>  ('org.onap.relationships.inventory.LocatedIn')complex*]";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        String dslQuery = "builder.getVerticesByProperty('aai-node-type', 'cloud-region')"
            + ".getVerticesByProperty('cloud-owner','coid').getVerticesByProperty('cloud-region-id','crid')"
            + ".union(builder.newInstance().createEdgeTraversalWithLabels( 'cloud-region','availability-zone', new ArrayList<>(Arrays.asList('org.onap.relationships.inventory.BelongsTo'))).store('x')"
            + ",builder.newInstance().createEdgeTraversalWithLabels( 'cloud-region','complex', new ArrayList<>(Arrays.asList('org.onap.relationships.inventory.LocatedIn'))).store('x')).cap('x').unfold().dedup()";

        assertEquals(dslQuery, query);
    }

    @Test
    public void cloudRegion_fromComplex1Test() throws AAIException {

        String builderQuery =
            "builder.getVerticesByProperty('aai-node-type', 'complex').getVerticesByProperty('data-center-code','data-center-code-name')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'complex','cloud-region').store('x').cap('x').unfold().dedup()";
        String aaiQuery = "complex('data-center-code','data-center-code-name') >  cloud-region*";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void cloudRegion_fromComplex2Test() throws AAIException {

        String builderQuery =
            "builder.getVerticesByProperty('aai-node-type', 'complex').getVerticesByProperty('data-center-code','data-center-code-name')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'complex','cloud-region').getVerticesByProperty('cloud-region-version','crv')"
                + ".store('x').cap('x').unfold().dedup()";
        String aaiQuery =
            "complex('data-center-code','data-center-code-name') >  cloud-region*('cloud-region-version','crv')";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void cloudRegion_fromNfTypeTest() throws AAIException {

        String builderQuery =
            "builder.getVerticesByProperty('aai-node-type', 'image').getVerticesByProperty('application-vendor','F5')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'image','vserver')"
                + ".where(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'vserver','generic-vnf').getVerticesByProperty('vnf-name','ZALL1MMSC03'))"
                + ".createEdgeTraversal(EdgeType.TREE, 'vserver','tenant').createEdgeTraversal(EdgeType.TREE, 'tenant','cloud-region')"
                + ".store('x').cap('x').unfold().dedup()";

        String aaiQuery =
            "image('application-vendor','F5') > vserver (> generic-vnf('vnf-name','ZALL1MMSC03')) > tenant > cloud-region*";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(builderQuery, query);
    }

    @Test
    public void cloudRegion_fromNfTypeVendorVersionTest() throws AAIException {

        String builderQuery =
            "builder.getVerticesByProperty('aai-node-type', 'image').getVerticesByProperty('application-vendor','vendor')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'image','vserver').where("
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'vserver','generic-vnf').getVerticesByProperty('nf-type','nfType')"
                + ").createEdgeTraversal(EdgeType.TREE, 'vserver','tenant').createEdgeTraversal(EdgeType.TREE, 'tenant','cloud-region')"
                + ".store('x').cap('x').unfold().dedup()";

        String aaiQuery =
            "image('application-vendor','vendor') >  vserver(> generic-vnf('nf-type', 'nfType') ) > tenant > cloud-region*";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void cloud_region_fromVnfTest() throws AAIException {

        String builderQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-id','vnfId')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vnfc').store('x')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'vnfc','vserver').store('x')"
                + ".createEdgeTraversal(EdgeType.TREE, 'vserver','tenant').store('x')"
                + ".createEdgeTraversal(EdgeType.TREE, 'tenant','cloud-region').store('x')"
                + ".cap('x').unfold().dedup()";

        String aaiQuery =
            "generic-vnf('vnf-id','vnfId')  > vnfc* > vserver* > tenant* > cloud-region*";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void cloud_region_sitesTest() throws AAIException {

        String builderQuery = "builder.getVerticesByProperty('aai-node-type', 'cloud-region')."
            + "getVerticesByProperty('cloud-owner','co').store('x').createEdgeTraversal(EdgeType.COUSIN, "
            + "'cloud-region','complex').store('x').cap('x').unfold().dedup()";

        String aaiQuery = "cloud-region*('cloud-owner','co') > complex*";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void complex_fromVnf2Test() throws AAIException {

        String builderQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-id','vnfId').store('x').union("
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','pserver').store('x')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'pserver','complex').store('x'),"
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vserver')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'vserver','pserver').store('x')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'pserver','complex').store('x')"
                + ").cap('x').unfold().dedup()";

        String aaiQuery = "generic-vnf*('vnf-id','vnfId')  [ > pserver* > complex*, "
            + "  >vserver > pserver* > complex* " + "]";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void complex_fromVnfTest2() throws AAIException {

        String builderQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-id','vnfId').store('x').union("
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','pserver').store('x'),"
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vserver')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'vserver','pserver').store('x'))"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'pserver','complex').store('x')"
                + ".cap('x').unfold().dedup()";

        String aaiQuery = "generic-vnf*('vnf-id','vnfId')    [>  pserver* , "
            + "  > vserver > pserver*  ] > complex*";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void fn_topology1Test() throws AAIException {

        String builderQuery = "builder.getVerticesByProperty('aai-node-type', 'customer')"
            + ".getVerticesByProperty('global-customer-id','a9a77d5a-123e-4ca2-9eb9-0b015d2ee0fb')"
            + ".createEdgeTraversal(EdgeType.TREE, 'customer','service-subscription').getVerticesByProperty('service-subscription-id','Nimbus')"
            + ".createEdgeTraversal(EdgeType.TREE, 'service-subscription','service-instance').getVerticesByProperty('service-instance-id','sid')"
            + ".createEdgeTraversal(EdgeType.COUSIN, 'service-instance','generic-vnf').store('x')"
            + ".union(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vnfc').store('x'),"
            + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vserver').store('x'),"
            + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','pserver').store('x'),"
            + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','pnf').store('x')).cap('x').unfold().dedup()";

        String aaiQuery =
            "customer('global-customer-id', 'a9a77d5a-123e-4ca2-9eb9-0b015d2ee0fb') > service-subscription('service-subscription-id', 'Nimbus') "
                + " > service-instance('service-instance-id','sid') > generic-vnf* "
                + "   [>  vnfc* ,>  vserver*,>  pserver* ,>  pnf* ]";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void vnf_Dsl() throws AAIException {

        String builderQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-id','vnfId').where(builder.newInstance().union(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','pserver')"
                + ".getVerticesByProperty('hostname','hostname1'),builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vserver')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'vserver','pserver').getVerticesByProperty('hostname','hostname1'))).store('x').cap('x').unfold().dedup()";

        String aaiQuery = "generic-vnf*('vnf-id','vnfId') ( [>  pserver('hostname','hostname1'), "
            + " > vserver > pserver('hostname','hostname1')])";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void hasPropertyTest() throws AAIException {
        String aaiQuery = "cloud-region* ('cloud-owner')";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-owner').store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void hasPropertyValuesTest() throws AAIException {
        String aaiQuery = "cloud-region* ('cloud-owner','cloud-owner1','cloud-owner2')";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-owner', new ArrayList<>(Arrays.asList('cloud-owner1','cloud-owner2'))).store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void hasNotPropertyValuesTest() throws AAIException {
        String aaiQuery = "cloud-region* !('cloud-owner','cloud-owner1','cloud-owner2')";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesExcludeByProperty('cloud-owner', new ArrayList<>(Arrays.asList('cloud-owner1','cloud-owner2'))).store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void hasNotPropertyNullValuesTest() throws AAIException {
        String aaiQuery = "cloud-region* !('cloud-owner',' ',' null ')";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesExcludeByProperty('cloud-owner', new ArrayList<>(Arrays.asList(' ',' null '))).store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Ignore
    @Test
    public void hasPropertyIntegerTest() throws AAIException {
        String aaiQuery =
            "cloud-region('cloud-owner', 'att-nc')('cloud-region-id', 'MTN61a') > vlan-range > vlan-tag*('vlan-id-inner', 20)";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-owner','att-nc').getVerticesByProperty('cloud-region-id','MTN61a')"
                + ".createEdgeTraversal(EdgeType.TREE, 'cloud-region','vlan-range').createEdgeTraversal(EdgeType.TREE, 'vlan-range','vlan-tag').getVerticesByProperty('vlan-id-inner',20).store('x').cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void directionalityTest() throws AAIException {
        String aaiQuery = "cloud-region('cloud-region-id','abc') >> complex* >> l3-network*";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-region-id','abc').createEdgeTraversal(EdgeType.COUSIN, 'cloud-region','complex').store('x').createEdgeTraversal(EdgeType.COUSIN, 'complex','l3-network').store('x').cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void enterSelectFilterTest() throws AAIException {
        String aaiQuery = "cloud-region*('cloud-region-id','whp3a'){'cloud-region-id'}";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-region-id','whp3a').store('x').cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(query, dslQuery);
    }

    @Ignore
    @Test
    public void returnSpecificPropsAndAllForDifferentVertices() throws AAIException {
        String aaiQuery =
            "cloud-region{'cloud-owner'}('cloud-region-id','new-r111egion-111111') > [ l3-network*, vlan-range > vlan-tag*]";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-region-id','new-r111egion-111111').store('x').union(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'cloud-region','l3-network').store('x'),builder.newInstance()"
                + ".createEdgeTraversal(EdgeType.TREE, 'cloud-region','vlan-range').createEdgeTraversal(EdgeType.TREE, 'vlan-range','vlan-tag').store('x')).cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(query, dslQuery);
    }

    @Test
    public void groupingAttributesTestForAggregate() throws AAIException {
        String aaiQuery =
            "cloud-region{'cloud-owner'}('cloud-owner','att-nc')('cloud-region-id','wah2a') > vip-ipv6-address-list > [subnet{'subnet-name'}, instance-group{'id'}]";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-owner','att-nc').getVerticesByProperty('cloud-region-id','wah2a').store('x').createEdgeTraversal(EdgeType.TREE, 'cloud-region','vip-ipv6-address-list').union(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'vip-ipv6-address-list','subnet').store('x'),builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'vip-ipv6-address-list','instance-group').store('x')).cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(query, dslQuery);
    }

    @Test
    public void multipleEdgeRuleTest() throws AAIException {
        String aaiQuery = "vserver('vserver-id','abc') > l-interface* > lag-interface*";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'vserver').getVerticesByProperty('vserver-id','abc').createEdgeTraversal(EdgeType.TREE, 'vserver','l-interface').store('x').createEdgeTraversal( 'l-interface','lag-interface').store('x').cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void multipleEdgeRuleTestWithLabels() throws AAIException {
        String aaiQuery =
            "vserver('vserver-id','abc') > l-interface* > ('org.onap.relationships.inventory.BelongsTo') lag-interface*";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'vserver').getVerticesByProperty('vserver-id','abc').createEdgeTraversal(EdgeType.TREE, 'vserver','l-interface').store('x').createEdgeTraversalWithLabels( 'l-interface','lag-interface', new ArrayList<>(Arrays.asList('org.onap.relationships.inventory.BelongsTo'))).store('x').cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void hasNotPropertyTest() throws AAIException {
        String aaiQuery = "cloud-region* !('cloud-owner')";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesExcludeByProperty('cloud-owner').store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void overlyNestedQueryTest() throws AAIException {
        // String aaiQuery = "generic-vnf*('vnf-id','vnfId') (> [ pserver('hostname','hostname1'),
        // vserver (> [ pserver('hostname','hostname1'), pserver('hostname','hostname1')])]) >
        // vserver";
        String aaiQuery =
            "generic-vnf*('vnf-id','vnfId') ( [> pserver('hostname','hostname1'), > vserver ( [>  pserver('hostname','hostname1'), > pserver('hostname','hostname1')])]) > vserver";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-id','vnfId').where(builder.newInstance().union"
                + "(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','pserver').getVerticesByProperty('hostname','hostname1'),"
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vserver').where(builder.newInstance().union(builder.newInstance()"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'vserver','pserver').getVerticesByProperty('hostname','hostname1'),builder.newInstance()."
                + "createEdgeTraversal(EdgeType.COUSIN, 'vserver','pserver').getVerticesByProperty('hostname','hostname1'))))).store('x').createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vserver').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void overlyNestedQueryTestWithLabels() throws AAIException {
        String aaiQuery =
            "generic-vnf*('vnf-id','vnfId') ( [> ('tosca.relationships.HostedOn')pserver('hostname','hostname1'),>  ('tosca.relationships.HostedOn')vserver ( [>  ('tosca.relationships.HostedOn')pserver('hostname','hostname1'),>  ('tosca.relationships.HostedOn')pserver('hostname','hostname1')])]) > ('org.onap.relationships.inventory.PartOf')allotted-resource";

        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-id','vnfId').where(builder.newInstance().union"
                + "(builder.newInstance().createEdgeTraversalWithLabels( 'generic-vnf','pserver', new ArrayList<>(Arrays.asList('tosca.relationships.HostedOn'))).getVerticesByProperty('hostname','hostname1'),"
                + "builder.newInstance().createEdgeTraversalWithLabels( 'generic-vnf','vserver', new ArrayList<>(Arrays.asList('tosca.relationships.HostedOn'))).where(builder.newInstance().union(builder.newInstance()"
                + ".createEdgeTraversalWithLabels( 'vserver','pserver', new ArrayList<>(Arrays.asList('tosca.relationships.HostedOn'))).getVerticesByProperty('hostname','hostname1'),builder.newInstance()."
                + "createEdgeTraversalWithLabels( 'vserver','pserver', new ArrayList<>(Arrays.asList('tosca.relationships.HostedOn'))).getVerticesByProperty('hostname','hostname1'))))).store('x').createEdgeTraversalWithLabels( 'generic-vnf','allotted-resource', new ArrayList<>(Arrays.asList('org.onap.relationships.inventory.PartOf'))).cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void getPserverWithAnEdgeToComplexAndCloudRegion() throws AAIException {
        String aaiQuery = "pserver*('prov-status')(> complex)(> cloud-region)";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'pserver').getVerticesByProperty('prov-status').where(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'pserver','complex')).where(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'pserver','cloud-region')).store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void getPserverWithAnEdgeToComplexButNotToCloudRegion() throws AAIException {
        String aaiQuery = "pserver*('prov-status')(> complex)!(> cloud-region)";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'pserver').getVerticesByProperty('prov-status').where(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'pserver','complex')).where(builder.newInstance().not(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'pserver','cloud-region'))).store('x').cap('x').unfold().dedup()";

        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void nestedUnionQueryTest() throws AAIException {

        String builderQuery =
            "builder.getVerticesByProperty('aai-node-type', 'generic-vnf').getVerticesByProperty('vnf-id','vnfId').store('x').union("
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','pserver').store('x'),"
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','vserver')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'vserver','pserver').store('x'))"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'pserver','complex').store('x')"
                + ".union(builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'complex','availability-zone')"
                + ".createEdgeTraversal(EdgeType.TREE, 'availability-zone','cloud-region').store('x'),"
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'complex','cloud-region').store('x'),"
                + "builder.newInstance().createEdgeTraversal(EdgeType.TREE, 'complex','ctag-pool').store('x').union("
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'ctag-pool','availability-zone').store('x')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'availability-zone','complex').store('x'),"
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'ctag-pool','generic-vnf').store('x')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'generic-vnf','availability-zone').store('x')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'availability-zone','complex').store('x'),"
                + "builder.newInstance().createEdgeTraversal(EdgeType.COUSIN, 'ctag-pool','vpls-pe').store('x')"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'vpls-pe','complex').store('x'))"
                + ".createEdgeTraversal(EdgeType.COUSIN, 'complex','cloud-region').store('x'))"
                + ".createEdgeTraversal(EdgeType.TREE, 'cloud-region','tenant').store('x')"
                + ".cap('x').unfold().dedup()";

        String aaiQuery = "generic-vnf*('vnf-id','vnfId')    [>  pserver* , "
            + "  > vserver > pserver*  ] > complex*   [>  availability-zone > cloud-region*,>  cloud-region*,  "
            + " > ctag-pool*   [ > availability-zone* > complex* , >  generic-vnf* > availability-zone* > complex*,>  vpls-pe* > complex*] > cloud-region*] > tenant* ";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();

        assertEquals(builderQuery, query);
    }

    @Test
    public void booleanPropertyValuesTest() throws AAIException {
        String aaiQuery = "cloud-region ('cloud-owner','cloud-owner1') > zone*('in-maint',true)";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-owner','cloud-owner1').createEdgeTraversal(EdgeType.COUSIN, 'cloud-region','zone').getVerticesByBooleanProperty('in-maint',true).store('x').cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void upperCaseBooleanPropertyValuesTest() throws AAIException {
        String aaiQuery = "cloud-region ('cloud-owner','cloud-owner1') > zone*('in-maint',TRUE)";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-owner','cloud-owner1').createEdgeTraversal(EdgeType.COUSIN, 'cloud-region','zone').getVerticesByBooleanProperty('in-maint',true).store('x').cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test
    public void booleanPropertyTest() throws AAIException {
        String aaiQuery = "cloud-region ('cloud-owner','cloud-owner1') > zone*('in-maint')";
        String dslQuery =
            "builder.getVerticesByProperty('aai-node-type', 'cloud-region').getVerticesByProperty('cloud-owner','cloud-owner1').createEdgeTraversal(EdgeType.COUSIN, 'cloud-region','zone').getVerticesByProperty('in-maint').store('x').cap('x').unfold().dedup()";
        String query =
            dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
        assertEquals(dslQuery, query);
    }

    @Test(expected = AAIException.class)
    public void filterOnUnion_vserverFromTentantFromCloudRegion_BadRequest() throws AAIException {
        String aaiQuery =
            "cloud-region('cloud-owner', 'test-aic')>[tenant*('tenant-id', 'tenant1'), tenant*('tenant-id', 'tenant2')]('tenant-name', 'tenant1')>[vserver*]";
        dslQueryProcessor.parseAaiQuery(QueryVersion.V2, aaiQuery).get("query").toString();
    }
}