/*
* Copyright 2014 LinkedIn Corp.
*
* Licensed 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.
*/
package azkaban.trigger;
import java.io.File;
import java.util.HashMap;
import java.util.Map;
import org.joda.time.DateTime;
import org.junit.Test;
import org.junit.Ignore;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertFalse;
import azkaban.trigger.builtin.BasicTimeChecker;
import azkaban.utils.JSONUtils;
import azkaban.utils.Props;
import azkaban.utils.Utils;
public class ConditionTest {
@Test
public void conditionTest() {
Map<String, ConditionChecker> checkers =
new HashMap<String, ConditionChecker>();
ThresholdChecker fake1 = new ThresholdChecker("thresholdchecker1", 10);
ThresholdChecker fake2 = new ThresholdChecker("thresholdchecker2", 20);
ThresholdChecker.setVal(15);
checkers.put(fake1.getId(), fake1);
checkers.put(fake2.getId(), fake2);
String expr1 =
"( " + fake1.getId() + ".eval()" + " && " + fake2.getId() + ".eval()"
+ " )" + " || " + "( " + fake1.getId() + ".eval()" + " && " + "!"
+ fake2.getId() + ".eval()" + " )";
String expr2 =
"( " + fake1.getId() + ".eval()" + " && " + fake2.getId() + ".eval()"
+ " )" + " || " + "( " + fake1.getId() + ".eval()" + " && "
+ fake2.getId() + ".eval()" + " )";
Condition cond = new Condition(checkers, expr1);
System.out.println("Setting expression " + expr1);
assertTrue(cond.isMet());
cond.setExpression(expr2);
System.out.println("Setting expression " + expr2);
assertFalse(cond.isMet());
}
@Ignore @Test
public void jsonConversionTest() throws Exception {
CheckerTypeLoader checkerTypeLoader = new CheckerTypeLoader();
checkerTypeLoader.init(new Props());
Condition.setCheckerLoader(checkerTypeLoader);
Map<String, ConditionChecker> checkers =
new HashMap<String, ConditionChecker>();
// get a new timechecker, start from now, repeat every minute. should
// evaluate to false now, and true a minute later.
DateTime now = DateTime.now();
String period = "6s";
// BasicTimeChecker timeChecker = new BasicTimeChecker(now, true, true,
// period);
ConditionChecker timeChecker =
new BasicTimeChecker("BasicTimeChecker_1", now.getMillis(),
now.getZone(), true, true, Utils.parsePeriodString(period), null);
System.out.println("checker id is " + timeChecker.getId());
checkers.put(timeChecker.getId(), timeChecker);
String expr = timeChecker.getId() + ".eval()";
Condition cond = new Condition(checkers, expr);
File temp = File.createTempFile("temptest", "temptest");
temp.deleteOnExit();
Object obj = cond.toJson();
JSONUtils.toJSON(obj, temp);
Condition cond2 = Condition.fromJson(JSONUtils.parseJSONFromFile(temp));
Map<String, ConditionChecker> checkers2 = cond2.getCheckers();
assertTrue(cond.getExpression().equals(cond2.getExpression()));
System.out.println("cond1: " + cond.getExpression());
System.out.println("cond2: " + cond2.getExpression());
assertTrue(checkers2.size() == 1);
ConditionChecker checker2 = checkers2.get(timeChecker.getId());
// assertTrue(checker2.getId().equals(timeChecker.getId()));
System.out.println("checker1: " + timeChecker.getId());
System.out.println("checker2: " + checker2.getId());
assertTrue(timeChecker.getId().equals(checker2.getId()));
}
}