package com.tesora.dve.sql.transform;

/*
 * #%L
 * Tesora Inc.
 * Database Virtualization Engine
 * %%
 * Copyright (C) 2011 - 2014 Tesora Inc.
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 * #L%
 */

import org.junit.Test;

import com.tesora.dve.sql.schema.SchemaContext;
import com.tesora.dve.sql.statement.dml.SelectStatement;
import com.tesora.dve.sql.util.TestName;

public class SessionRewriteTransformTest extends TransformTest {

	public SessionRewriteTransformTest() {
		super("SessionRewrite");
		// TODO Auto-generated constructor stub
	}

	// our tiny little schema
	private static final String[] schema = new String[] {
		"create table A (`id` integer unsigned not null, `desc` varchar(50), flags tinyint, `slug` varchar(128)) static distribute on (`id`);",
		"create table B (`id` integer unsigned not null, `desc` varchar(50), flags tinyint, `slug` varchar(128)) broadcast distribute;",
		"create table R (`id` integer unsigned not null, `desc` varchar(50), flags tinyint, `slug` varchar(128)) random distribute;",
	};

	@Test
	public void testSimpleDatabase() throws Throwable {
		SchemaContext db = buildSchema(TestName.MULTI,schema);
		stmtTest(db, "select database()",
				SelectStatement.class,
				bes(new AdhocResultsSessionExpectedStep()));
	}
	
	@Test
	public void testPE1675() throws Throwable {
		SchemaContext db = buildSchema(TestName.MULTI,schema);
		stmtTest(db, "select schema()",
				SelectStatement.class,
				bes(new AdhocResultsSessionExpectedStep()));
	}
	
	@Test
	public void testComplexTransientDatabase() throws Throwable {
		SchemaContext db = buildSchema(TestName.MULTI,schema);
		stmtTest(db, "select concat('my',database())",
				SelectStatement.class,
				bes(new ProjectingExpectedStep("SELECT concat( 'my','mydb' )  AS func", null)));	
	}
	
	@Test
	public void testPE351() throws Throwable {
		SchemaContext db = buildSchema(TestName.MULTI);
		String sql = "select case 1.0 when 0.1 then 'a' when 1.0 then 'b' else 'c' end";
		stmtTest(db,sql,
				SelectStatement.class,
				bes(
						new ProjectingExpectedStep(
						"SELECT CASE 1.0 WHEN 0.1 THEN 'a' WHEN 1.0 THEN 'b' ELSE 'c' END AS casex",
							null)
					));
	}

	@Test
	public void testPE730() throws Throwable {
		SchemaContext db = buildSchema(TestName.MULTI,schema);
		stmtTest(db,"select @@version_comment limit 1",
				SelectStatement.class,
				bes(new AdhocResultsSessionExpectedStep()));
	}
	

}