MockTriggerLoader.java

package azkaban.test.trigger;

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

import azkaban.trigger.Trigger;
import azkaban.trigger.TriggerLoader;
import azkaban.trigger.TriggerLoaderException;

public class MockTriggerLoader implements TriggerLoader {

	Map<Integer, Trigger> triggers = new HashMap<Integer, Trigger>();
	int triggerCount = 0;
	
	@Override
	public synchronized void addTrigger(Trigger t) throws TriggerLoaderException {
		t.setTriggerId(triggerCount);
		t.setLastModifyTime(System.currentTimeMillis());
		triggers.put(t.getTriggerId(), t);
		triggerCount++;
	}

	@Override
	public synchronized void removeTrigger(Trigger s) throws TriggerLoaderException {
		triggers.remove(s);
	}

	@Override
	public synchronized void updateTrigger(Trigger t) throws TriggerLoaderException {
		t.setLastModifyTime(System.currentTimeMillis());
		triggers.put(t.getTriggerId(), t);
	}

	@Override
	public synchronized List<Trigger> loadTriggers() throws TriggerLoaderException {
		return new ArrayList<Trigger>(triggers.values());
	}

	@Override
	public synchronized Trigger loadTrigger(int triggerId) throws TriggerLoaderException {
		return triggers.get(triggerId);
	}

	@Override
	public List<Trigger> getUpdatedTriggers(long lastUpdateTime)
			throws TriggerLoaderException {
		// TODO Auto-generated method stub
		return null;
	}

}