/**
* Copyright (C) 2009-2013 FoundationDB, LLC
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* 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/>.
*/
package com.foundationdb.sql.optimizer;
import com.foundationdb.junit.NamedParameterizedRunner;
import com.foundationdb.junit.Parameterization;
import com.foundationdb.junit.ParameterizationBuilder;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.yaml.snakeyaml.Yaml;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import static org.junit.Assert.assertEquals;
@RunWith(NamedParameterizedRunner.class)
public final class DistinctEliminatorSimpleTest extends DistinctEliminatorTestBase {
private static final File SIMPLE_TEST = new File(RESOURCE_DIR, "simple-distincts.yaml");
@NamedParameterizedRunner.TestParameters
public static Collection<Parameterization> statements() throws Exception {
FileReader fileReader = new FileReader(SIMPLE_TEST);
try {
ParameterizationBuilder pb = new ParameterizationBuilder();
BufferedReader buffered = new BufferedReader(fileReader);
Yaml yaml = new Yaml();
for (Object objRaw : yaml.loadAll(buffered)) {
List<?> asList = (List<?>) objRaw;
for (Object lineRaw : asList) {
Map<?,?> line = (Map<?,?>) lineRaw;
if (line.size() != 1)
throw new RuntimeException("need key-val pair:" + line);
String actionStr = (String) line.keySet().iterator().next();
String sql = (String) line.get(actionStr);
String name = sql;
if (name.startsWith("SELECT DISTINCT"))
name = name.substring("SELECT DISTINCT".length());
KeepOrOptimize action = KeepOrOptimize.valueOf(actionStr.toUpperCase());
pb.create(name, action != KeepOrOptimize.IGNORED, sql, action);
}
}
return pb.asList();
}
finally {
fileReader.close();
}
}
@Test
public void test() throws Exception {
if (!sql.toUpperCase().contains("DISTINCT"))
throw new RuntimeException("original didn't have DISTINCT");
String optimized = optimized();
KeepOrOptimize distinctActualOptimized = optimized.contains("DISTINCT")
? KeepOrOptimize.KEPT
: KeepOrOptimize.OPTIMIZED;
assertEquals(optimized, distinctExpectedOptimized, distinctActualOptimized);
}
public DistinctEliminatorSimpleTest(String sql, KeepOrOptimize distinctExpectedOptimized) {
super(sql, sql, null, null);
this.distinctExpectedOptimized = distinctExpectedOptimized;
}
public final KeepOrOptimize distinctExpectedOptimized;
private enum KeepOrOptimize {
KEPT, OPTIMIZED, IGNORED
}
}