TriggerManagerDeadlockTest.java

/*
 * 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 azkaban.executor.AlerterHolder;
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.Ignore;
import org.junit.Test;

import azkaban.executor.ExecutorLoader;
import azkaban.executor.ExecutorManager;
import azkaban.executor.ExecutorManagerException;
import azkaban.executor.MockExecutorLoader;
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();
    ExecutorManager executorManager = new ExecutorManager(props, execLoader, new AlerterHolder(props));
    triggerManager = new TriggerManager(props, loader, executorManager);
  }

  @After
  public void tearDown() {

  }

  // TODO kunkun-tang: This test has problems. Will fix
  @Ignore @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 static 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.TriggerBuilder("azkaban",
            "azkabanTest",
            triggerCond,
            expireCond,
            actions).build();

    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.TriggerBuilder("azkaban",
                                   "azkabanTest",
                                   triggerCond,
                                   expireCond,
                                   actions).build();

    return t;
  }

}