package adql.query;

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

import java.util.ArrayList;
import java.util.Iterator;

import org.junit.Before;
import org.junit.Test;

import adql.parser.ADQLParser;
import adql.parser.ParseException;
import adql.query.ADQLObject;
import adql.query.ADQLQuery;
import adql.query.TextPosition;
import adql.query.constraint.Comparison;
import adql.query.from.ADQLJoin;
import adql.query.from.ADQLTable;
import adql.query.operand.ADQLColumn;
import adql.query.operand.ADQLOperand;
import adql.query.operand.function.ADQLFunction;
import adql.search.SimpleSearchHandler;

public class TestADQLObjectPosition {

	private ADQLParser parser = new ADQLParser();

	@Before
	public void setUp(){

	}

	@Test
	public void testPositionInAllClauses(){
		try{
			ADQLQuery query = parser.parseQuery("SELECT truc, bidule.machin, toto(truc, chose) AS \"super\" FROM foo JOIN bidule USING(id) WHERE truc > 12.5 AND bidule.machin < 5 GROUP BY chose HAVING try > 0 ORDER BY chouetteAlors");

			Iterator<ADQLObject> results = query.search(new SimpleSearchHandler(true){
				@Override
				protected boolean match(ADQLObject obj){
					return obj.getPosition() == null;
				}
			});
			if (results.hasNext()){
				System.err.println("OBJECT WITH NO DEFINED POSITION:");
				while(results.hasNext())
					System.err.println("    * " + results.next().toADQL());
				fail("At least one item of the generated ADQL tree does not have a position information! (see System.err for more details)");
			}
		}catch(ParseException pe){
			pe.printStackTrace();
			fail("No error should have occured here: the ADQL query is syntactically correct!");
		}
	}

	private void assertEquality(final TextPosition expected, final TextPosition realPos){
		assertEquals(expected.beginLine, realPos.beginLine);
		assertEquals(expected.beginColumn, realPos.beginColumn);
		assertEquals(expected.endLine, realPos.endLine);
		assertEquals(expected.endColumn, realPos.endColumn);
	}

	@Test
	public void testPositionAccuracy(){
		try{
			ADQLQuery query = parser.parseQuery("SELECT TOP 1000 oid FROM foo JOIN bar USING(oid)\nWHERE foo || toto = 'truc'\n      AND 2 > 1+0 GROUP BY oid HAVING COUNT(oid) > 10\n\tORDER BY 1 DESC");
			// Test SELECT
			assertEquality(new TextPosition(1, 1, 1, 20), query.getSelect().getPosition());
			// Test ADQLColumn (here: "oid")
			assertEquality(new TextPosition(1, 17, 1, 20), query.getSelect().get(0).getPosition());
			// Test FROM & ADQLJoin
			/* NB: The clause FROM is the only one which is not a list but a single item of type FromContent (JOIN or table).
			 *     That's why, it is not possible to get its exact starting position ('FROM') ; the starting position is
			 *     the one of the first table of the clause FROM. */
			assertEquality(new TextPosition(1, 26, 1, 49), query.getFrom().getPosition());
			// Test ADQLTable
			ArrayList<ADQLTable> tables = query.getFrom().getTables();
			assertEquality(new TextPosition(1, 26, 1, 29), tables.get(0).getPosition());
			assertEquality(new TextPosition(1, 35, 1, 38), tables.get(1).getPosition());
			// Test the join condition:
			Iterator<ADQLColumn> itCol = ((ADQLJoin)query.getFrom()).getJoinedColumns();
			assertEquality(new TextPosition(1, 45, 1, 48), itCol.next().getPosition());
			// Test WHERE
			assertEquality(new TextPosition(2, 1, 3, 18), query.getWhere().getPosition());
			// Test COMPARISON = CONSTRAINT
			Comparison comp = (Comparison)(query.getWhere().get(0));
			assertEquality(new TextPosition(2, 7, 2, 27), comp.getPosition());
			// Test left operand = concatenation:
			ADQLOperand operand = comp.getLeftOperand();
			assertEquality(new TextPosition(2, 7, 2, 18), operand.getPosition());
			Iterator<ADQLObject> itObj = operand.adqlIterator();
			// foo
			assertEquality(new TextPosition(2, 7, 2, 10), itObj.next().getPosition());
			// toto
			assertEquality(new TextPosition(2, 14, 2, 18), itObj.next().getPosition());
			// Test right operand = string:
			operand = comp.getRightOperand();
			assertEquality(new TextPosition(2, 21, 2, 27), operand.getPosition());
			// Test COMPARISON > CONSTRAINT:
			comp = (Comparison)(query.getWhere().get(1));
			assertEquality(new TextPosition(3, 11, 3, 18), comp.getPosition());
			// Test left operand = numeric:
			operand = comp.getLeftOperand();
			assertEquality(new TextPosition(3, 11, 3, 12), operand.getPosition());
			// Test right operand = operation:
			operand = comp.getRightOperand();
			assertEquality(new TextPosition(3, 15, 3, 18), operand.getPosition());
			itObj = operand.adqlIterator();
			// 1
			assertEquality(new TextPosition(3, 15, 3, 16), itObj.next().getPosition());
			// 0
			assertEquality(new TextPosition(3, 17, 3, 18), itObj.next().getPosition());
			// Test GROUP BY
			assertEquality(new TextPosition(3, 19, 3, 31), query.getGroupBy().getPosition());
			// oid
			assertEquality(new TextPosition(3, 28, 3, 31), query.getGroupBy().get(0).getPosition());
			// Test HAVING
			assertEquality(new TextPosition(3, 32, 3, 54), query.getHaving().getPosition());
			// Test COMPARISON > CONSTRAINT:
			comp = (Comparison)(query.getHaving().get(0));
			assertEquality(new TextPosition(3, 39, 3, 54), comp.getPosition());
			// Test left operand = COUNT function:
			operand = comp.getLeftOperand();
			assertEquality(new TextPosition(3, 39, 3, 49), operand.getPosition());
			// Test parameter = ADQLColumn oid:
			assertEquality(new TextPosition(3, 45, 3, 48), ((ADQLFunction)operand).getParameter(0).getPosition());
			// Test right operand = operation:
			operand = comp.getRightOperand();
			assertEquality(new TextPosition(3, 52, 3, 54), operand.getPosition());
			// Test ORDER BY
			assertEquality(new TextPosition(4, 9, 4, 24), query.getOrderBy().getPosition());
			// Test column index:
			assertEquality(new TextPosition(4, 18, 4, 19), query.getOrderBy().get(0).getPosition());

		}catch(ParseException pe){
			System.err.println("ERROR IN THE ADQL QUERY AT " + pe.getPosition());
			pe.printStackTrace();
			fail("No error should have occured here: the ADQL query is syntactically correct!");
		}
	}

}