/**
* 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.rule;
import com.foundationdb.ais.model.AkibanInformationSchema;
import com.foundationdb.server.rowdata.SchemaFactory;
import org.yaml.snakeyaml.Yaml;
import java.util.ArrayList;
import java.util.List;
import java.io.File;
import java.io.FileInputStream;
public class RulesTestHelper
{
private RulesTestHelper() {
}
public static List<BaseRule> loadRules(File file) throws Exception {
Yaml yaml = new Yaml();
FileInputStream istr = new FileInputStream(file);
List list = yaml.loadAs(istr, List.class );
istr.close();
return parseRules(list);
}
public static List<BaseRule> parseRules(String str) throws Exception {
Yaml yaml = new Yaml();
List list = yaml.loadAs(str, List.class);
return parseRules(list);
}
public static List<BaseRule> parseRules(List list) throws Exception {
List<BaseRule> result = new ArrayList<>();
for (Object obj : list) {
if (obj instanceof String) {
String cname = (String)obj;
if (cname.indexOf('.') < 0)
cname = RulesTestHelper.class.getPackage().getName() + '.' + cname;
result.add((BaseRule)Class.forName(cname).newInstance());
}
else {
// TODO: Someday parse options from hash, etc.
throw new Exception("Don't know what to do with " + obj);
}
}
return result;
}
// Make fake row def cache to keep TableRowType constructor
// and Index.getAllColumns() from getting NPE.
public static void ensureRowDefs(AkibanInformationSchema ais) {
new SchemaFactory().buildRowDefs(ais);
}
}