package org.apache.rya.indexing.IndexPlanValidator;

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF 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.
 */

import java.util.Iterator;
import java.util.List;
import java.util.Set;

import org.eclipse.rdf4j.query.MalformedQueryException;
import org.eclipse.rdf4j.query.algebra.TupleExpr;
import org.eclipse.rdf4j.query.parser.ParsedQuery;
import org.eclipse.rdf4j.query.parser.sparql.SPARQLParser;
import org.junit.Assert;
import org.junit.Test;

import com.google.common.collect.Lists;
import com.google.common.collect.Sets;

public class TupleExecutionPlanGeneratorTest {


    private String q1 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  ?s a ?t ."//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?u <uri:talksTo> ?s . "//
            + "}";//


    private String q2 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?s a ?t ."//
            + "  ?u <uri:talksTo> ?s . "//
            + "}";//


    private String q3 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  ?u <uri:talksTo> ?s . "//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?s a ?t ."//
            + "}";//


    private String q4 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  ?s a ?t ."//
            + "  ?u <uri:talksTo> ?s . "//
             + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "}";//


    private String q5 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?u <uri:talksTo> ?s . "//
            + "  ?s a ?t ."//
            + "}";//


    private String q6 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  ?u <uri:talksTo> ?s . "//
            + "  ?s a ?t ."//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "}";//


    private String q7 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  ?s a ?t ."//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "}";//


    private String q8 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?s a ?t ."//
            + "}";//


    private String q9 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  Filter(?t > 2). "//
            + "  Filter(?s > 1). "//
            + "  ?s a ?t ."//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?u <uri:talksTo> ?s . "//
            + "}";//


    private String q10 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  Filter(?t > 2). "//
            + "  Filter(?s > 1). "//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?s a ?t ."//
            + "  ?u <uri:talksTo> ?s . "//
            + "}";//


    private String q11 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  Filter(?t > 2). "//
            + "  Filter(?s > 1). "//
            + "  ?u <uri:talksTo> ?s . "//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?s a ?t ."//
            + "}";//


    private String q12 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  Filter(?t > 2). "//
            + "  Filter(?s > 1). "//
            + "  ?s a ?t ."//
            + "  ?u <uri:talksTo> ?s . "//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "}";//


    private String q13 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  Filter(?t > 2). "//
            + "  Filter(?s > 1). "//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?u <uri:talksTo> ?s . "//
            + "  ?s a ?t ."//
            + "}";//


    private String q14 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  Filter(?t > 2). "//
            + "  Filter(?s > 1). "//
            + "  ?u <uri:talksTo> ?s . "//
            + "  ?s a ?t ."//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "}";//


    private String q15 = ""//
            + "SELECT ?s ?t ?u " //
            + "{" //
            + "  Filter(?s > 1). "//
            + "  Filter(?t > 2). "//
            + "  ?s a ?t ."//
            + "  ?t <http://www.w3.org/2000/01/rdf-schema#label> ?u ."//
            + "  ?u <uri:talksTo> ?s . "//
            + "}";//


    @Test
    public void testTwoNodeOrder() {

        SPARQLParser parser = new SPARQLParser();

        ParsedQuery pq1 = null;
        ParsedQuery pq2 = null;

        try {
            pq1 = parser.parseQuery(q7, null);
            pq2 = parser.parseQuery(q8, null);
        } catch (MalformedQueryException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        Set<TupleExpr> tupSet = Sets.newHashSet();
        tupSet.add(pq1.getTupleExpr());

        TupleExecutionPlanGenerator tep = new TupleExecutionPlanGenerator();
        Iterator<TupleExpr> processedTups = tep.getPlans(tupSet.iterator());

        List<TupleExpr> processedTupList = Lists.newArrayList();

        int size = 0;

        while(processedTups.hasNext()) {
            Assert.assertTrue(processedTups.hasNext());
            processedTupList.add(processedTups.next());
            size++;
        }

        Assert.assertEquals(2, size);

        Assert.assertEquals(pq1.getTupleExpr(), processedTupList.get(0));
        Assert.assertEquals(pq2.getTupleExpr(), processedTupList.get(1));

    }


    @Test
    public void testThreeNodeOrder() {

        SPARQLParser parser = new SPARQLParser();

        ParsedQuery pq1 = null;
        ParsedQuery pq2 = null;
        ParsedQuery pq3 = null;
        ParsedQuery pq4 = null;
        ParsedQuery pq5 = null;
        ParsedQuery pq6 = null;

        try {
            pq1 = parser.parseQuery(q1, null);
            pq2 = parser.parseQuery(q2, null);
            pq3 = parser.parseQuery(q3, null);
            pq4 = parser.parseQuery(q4, null);
            pq5 = parser.parseQuery(q5, null);
            pq6 = parser.parseQuery(q6, null);
        } catch (MalformedQueryException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        Set<TupleExpr> tupSet = Sets.newHashSet();
        tupSet.add(pq1.getTupleExpr());

        TupleExecutionPlanGenerator tep = new TupleExecutionPlanGenerator();
        Iterator<TupleExpr> processedTups= tep.getPlans(tupSet.iterator());

        List<TupleExpr> processedTupList = Lists.newArrayList();

        int size = 0;

        while(processedTups.hasNext()) {
            Assert.assertTrue(processedTups.hasNext());
            processedTupList.add(processedTups.next());
            size++;
        }

        Assert.assertTrue(!processedTups.hasNext());
        Assert.assertEquals(6, size);

        Assert.assertEquals(pq1.getTupleExpr(), processedTupList.get(1));
        Assert.assertEquals(pq2.getTupleExpr(), processedTupList.get(2));
        Assert.assertEquals(pq3.getTupleExpr(), processedTupList.get(4));
        Assert.assertEquals(pq4.getTupleExpr(), processedTupList.get(0));
        Assert.assertEquals(pq5.getTupleExpr(), processedTupList.get(3));
        Assert.assertEquals(pq6.getTupleExpr(), processedTupList.get(5));

    }


    @Test
    public void testThreeNodeOrderFilter() {

        SPARQLParser parser = new SPARQLParser();

        ParsedQuery pq1 = null;
        ParsedQuery pq2 = null;
        ParsedQuery pq3 = null;
        ParsedQuery pq4 = null;
        ParsedQuery pq5 = null;
        ParsedQuery pq6 = null;
        ParsedQuery pq7 = null;

        try {
            pq1 = parser.parseQuery(q9, null);
            pq2 = parser.parseQuery(q10, null);
            pq3 = parser.parseQuery(q11, null);
            pq4 = parser.parseQuery(q12, null);
            pq5 = parser.parseQuery(q13, null);
            pq6 = parser.parseQuery(q14, null);
            pq7 = parser.parseQuery(q15, null);
        } catch (MalformedQueryException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        System.out.println(pq1.getTupleExpr());

        Set<TupleExpr> tupSet = Sets.newHashSet();
        tupSet.add(pq7.getTupleExpr());

        TupleExecutionPlanGenerator tep = new TupleExecutionPlanGenerator();
        Iterator<TupleExpr> processedTups= tep.getPlans(tupSet.iterator());

        List<TupleExpr> processedTupList = Lists.newArrayList();

        int size = 0;

        while(processedTups.hasNext()) {

            Assert.assertTrue(processedTups.hasNext());
            TupleExpr te = processedTups.next();
            processedTupList.add(te);
            System.out.println("Processed tups are " + te);
            size++;
        }

        Assert.assertTrue(!processedTups.hasNext());
        Assert.assertEquals(6, size);

        Assert.assertEquals(pq1.getTupleExpr(), processedTupList.get(1));
        Assert.assertEquals(pq2.getTupleExpr(), processedTupList.get(2));
        Assert.assertEquals(pq3.getTupleExpr(), processedTupList.get(4));
        Assert.assertEquals(pq4.getTupleExpr(), processedTupList.get(0));
        Assert.assertEquals(pq5.getTupleExpr(), processedTupList.get(3));
        Assert.assertEquals(pq6.getTupleExpr(), processedTupList.get(5));

    }


}