DataSourceMockup.java

package dk.statsbiblioteket.medieplatform.autonomous.processmonitor.datasources;

import dk.statsbiblioteket.medieplatform.autonomous.Batch;
import dk.statsbiblioteket.medieplatform.autonomous.Event;
import dk.statsbiblioteket.medieplatform.autonomous.NotFoundException;

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

/**
 * Created with IntelliJ IDEA.
 * User: abr
 * Date: 9/17/13
 * Time: 10:32 AM
 * To change this template use File | Settings | File Templates.
 */
public class DataSourceMockup implements DataSource {

    private List<Batch> dummyBatches;

    private String username;
    private String password;

    public DataSourceMockup() {

/*
        System.out.println("Username: " + username);
        System.out.println("Password: " + password);
*/

        Event e1 = new Event();
        e1.setEventID("Shipped_to_supplier");
        e1.setSuccess(true);

        Event e2 = new Event();
        e2.setEventID("Data_Received");
        e2.setSuccess(false);

        Event e3 = new Event();
        e3.setEventID("Data_Archived");
        e3.setSuccess(true);
        List<Event> b1Events = new ArrayList<>();
        b1Events.add(e1);
        b1Events.add(e2);
        b1Events.add(e3);

        Batch b1 = new Batch("3001");
        b1.setEventList(b1Events);

        Event e4 = new Event();
        e4.setEventID("Shipped_to_supplier");
        e4.setSuccess(true);

        Event e5 = new Event();
        e5.setEventID("Data_Received");
        e5.setSuccess(false);

        Event e6 = new Event();
        e6.setEventID("Data_Archived");
        e6.setSuccess(false);

        List<Event> b2Events = new ArrayList<>();
        b2Events.add(e4);
        b2Events.add(e5);
        b2Events.add(e6);

        Batch b2 = new Batch("3002");
        b2.setEventList(b2Events);

        dummyBatches = new ArrayList<>();
        dummyBatches.add(b1);
        dummyBatches.add(b2);
    }


    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    public String getPassword() {
        return password;
    }

    public void setPassword(String password) {
        this.password = password;
    }


    @Override
    public List<Batch> getBatches(boolean includeDetails, Map<String, String> filters) {
        return dummyBatches;
    }

    @Override
    public Batch getBatch(String batchID, Integer roundTripNumber, boolean includeDetails) throws NotFoundException {
        Batch batch = null;
        for (Batch b : dummyBatches) {
            if (b.getBatchID().equals(batchID)) {
                batch = b;
            }
        }
        if (batch == null) {
            throw new NotFoundException("Batch not found" + batchID);
        }
        return batch;
    }

    @Override
    public Event getBatchEvent(String batchID, Integer roundTripNumber, String eventID, boolean includeDetails) throws
                                                                                                                NotFoundException {
        Event event = null;
        for (Batch b : dummyBatches) {
            if (b.getBatchID().equals(batchID)) {
                for (Event e : b.getEventList()) {
                    if (e.getEventID().equals(eventID)) {
                        event = e;
                    }
                }
            }
        }
        if (event == null) {
            throw new NotFoundException("Event not found");
        }
        return event;
    }
}