TriggerManagerDeadlockTest.java

package azkaban.test.trigger;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

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

import azkaban.alert.Alerter;
import azkaban.executor.ExecutorLoader;
import azkaban.executor.ExecutorManager;
import azkaban.executor.ExecutorManagerException;
import azkaban.test.execapp.MockExecutorLoader;
import azkaban.trigger.Condition;
import azkaban.trigger.ConditionChecker;
import azkaban.trigger.Trigger;
import azkaban.trigger.TriggerAction;
import azkaban.trigger.TriggerLoader;
import azkaban.trigger.TriggerLoaderException;
import azkaban.trigger.TriggerManager;
import azkaban.trigger.TriggerManagerException;
import azkaban.trigger.builtin.CreateTriggerAction;
import azkaban.utils.Props;

public class TriggerManagerDeadlockTest {

  TriggerLoader loader;
  TriggerManager triggerManager;
  ExecutorLoader execLoader;

  @Before
  public void setup() throws ExecutorManagerException, TriggerManagerException {
    loader = new MockTriggerLoader();
    Props props = new Props();
    props.put("trigger.scan.interval", 1000);
    props.put("executor.port", 12321);
    execLoader = new MockExecutorLoader();
    Map<String, Alerter> alerters = new HashMap<String, Alerter>();
    ExecutorManager executorManager =
        new ExecutorManager(props, execLoader, alerters);
    triggerManager = new TriggerManager(props, loader, executorManager);
  }

  @After
  public void tearDown() {

  }

  @Test
  public void deadlockTest() throws TriggerLoaderException,
      TriggerManagerException {
    // this should well saturate it
    for (int i = 0; i < 1000; i++) {
      Trigger t = createSelfRegenTrigger();
      loader.addTrigger(t);
    }
    // keep going and add more
    for (int i = 0; i < 10000; i++) {
      Trigger d = createDummyTrigger();
      triggerManager.insertTrigger(d);
      triggerManager.removeTrigger(d);
    }

    System.out.println("No dead lock.");
  }

  public class AlwaysOnChecker implements ConditionChecker {

    public static final String type = "AlwaysOnChecker";

    private final String id;
    private final Boolean alwaysOn;

    public AlwaysOnChecker(String id, Boolean alwaysOn) {
      this.id = id;
      this.alwaysOn = alwaysOn;
    }

    @Override
    public Object eval() {
      // TODO Auto-generated method stub
      return alwaysOn;
    }

    @Override
    public Object getNum() {
      // TODO Auto-generated method stub
      return null;
    }

    @Override
    public void reset() {
      // TODO Auto-generated method stub

    }

    @Override
    public String getId() {
      return id;
    }

    @Override
    public String getType() {
      // TODO Auto-generated method stub
      return type;
    }

    @Override
    public ConditionChecker fromJson(Object obj) throws Exception {
      // TODO Auto-generated method stub
      return null;
    }

    @Override
    public Object toJson() {
      // TODO Auto-generated method stub
      return null;
    }

    @Override
    public void stopChecker() {
      // TODO Auto-generated method stub

    }

    @Override
    public void setContext(Map<String, Object> context) {
      // TODO Auto-generated method stub

    }

    @Override
    public long getNextCheckTime() {
      // TODO Auto-generated method stub
      return 0;
    }

  }

  private Trigger createSelfRegenTrigger() {
    ConditionChecker alwaysOnChecker =
        new AlwaysOnChecker("alwaysOn", Boolean.TRUE);
    String triggerExpr = alwaysOnChecker.getId() + ".eval()";
    Map<String, ConditionChecker> triggerCheckers =
        new HashMap<String, ConditionChecker>();
    triggerCheckers.put(alwaysOnChecker.getId(), alwaysOnChecker);
    Condition triggerCond = new Condition(triggerCheckers, triggerExpr);

    TriggerAction triggerAct =
        new CreateTriggerAction("dummyTrigger", createDummyTrigger());
    List<TriggerAction> actions = new ArrayList<TriggerAction>();
    actions.add(triggerAct);

    ConditionChecker alwaysOffChecker =
        new AlwaysOnChecker("alwaysOff", Boolean.FALSE);
    String expireExpr = alwaysOffChecker.getId() + ".eval()";
    Map<String, ConditionChecker> expireCheckers =
        new HashMap<String, ConditionChecker>();
    expireCheckers.put(alwaysOffChecker.getId(), alwaysOffChecker);
    Condition expireCond = new Condition(expireCheckers, expireExpr);

    Trigger t =
        new Trigger("azkaban", "azkabanTest", triggerCond, expireCond, actions);
    return t;
  }

  private Trigger createDummyTrigger() {
    ConditionChecker alwaysOnChecker =
        new AlwaysOnChecker("alwaysOn", Boolean.TRUE);
    String triggerExpr = alwaysOnChecker.getId() + ".eval()";
    Map<String, ConditionChecker> triggerCheckers =
        new HashMap<String, ConditionChecker>();
    triggerCheckers.put(alwaysOnChecker.getId(), alwaysOnChecker);
    Condition triggerCond = new Condition(triggerCheckers, triggerExpr);

    TriggerAction triggerAct = new DummyTriggerAction("howdy!");
    List<TriggerAction> actions = new ArrayList<TriggerAction>();
    actions.add(triggerAct);

    ConditionChecker alwaysOffChecker =
        new AlwaysOnChecker("alwaysOff", Boolean.FALSE);
    String expireExpr = alwaysOffChecker.getId() + ".eval()";
    Map<String, ConditionChecker> expireCheckers =
        new HashMap<String, ConditionChecker>();
    expireCheckers.put(alwaysOffChecker.getId(), alwaysOffChecker);
    Condition expireCond = new Condition(expireCheckers, expireExpr);

    Trigger t =
        new Trigger("azkaban", "azkabanTest", triggerCond, expireCond, actions);
    return t;
  }

}