org.nakedobjects.example.expenses.fixtures.RandomClaimFixture Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of expenses-fixture
The newest version!
package org.nakedobjects.example.expenses.fixtures;

import org.nakedobjects.applib.value.Date;
import java.util.List;
import java.util.Random;

import org.nakedobjects.applib.value.Money;
import org.nakedobjects.example.expenses.claims.Claim;
import org.nakedobjects.example.expenses.claims.ClaimRepository;
import org.nakedobjects.example.expenses.claims.ExpenseItem;
import org.nakedobjects.example.expenses.claims.ExpenseType;
import org.nakedobjects.example.expenses.claims.ProjectCode;
import org.nakedobjects.example.expenses.claims.items.Airfare;
import org.nakedobjects.example.expenses.claims.items.CarRental;
import org.nakedobjects.example.expenses.claims.items.Hotel;
import org.nakedobjects.example.expenses.claims.items.Journey;
import org.nakedobjects.example.expenses.claims.items.PrivateCarJourney;
import org.nakedobjects.example.expenses.employee.Employee;
import org.nakedobjects.example.expenses.employee.EmployeeRepository;



public class RandomClaimFixture extends AbstractClaimFixture {
	
	
    // {{ Injected Services
    /*
     * This region contains references to the services (Repositories, 
     * Factories or other Services) used by this domain object.  The 
     * references are injected by the application container.
     */
     
	// {{ Injected: ClaimRepository
    private ClaimRepository claimRepository;
    /**
     * This field is not persisted, nor displayed to the user.
     */
    protected ClaimRepository getClaimRepository() {
        return this.claimRepository;
    }
    /**
     * Injected by the application container.
     */
    public void setClaimRepository(ClaimRepository claimRepository) {
        this.claimRepository = claimRepository;
    }
	// }}
    
		
	// {{ Injected: EmployeeRepository
    private EmployeeRepository employeeRepository;
    /**
     * This field is not persisted, nor displayed to the user.
     */
    protected EmployeeRepository getEmployeeRepository() {
        return this.employeeRepository;
    }
    /**
     * Injected by the application container.
     */
    public void setEmployeeRepository(EmployeeRepository employeeRepository) {
        this.employeeRepository = employeeRepository;
    }
	// }}

		
    // }}
	

    private Random random = new Random();

    private ExpenseType[] expenseTypes;

    private ProjectCode[] codes;
    private int numberOfCodes;

    private int claimCount = 4;

    public void install() {
    	loadExpenseTypes();
        loadProjectCodes();

        Employee claimant = (Employee) employeeRepository.findEmployeeByName("Sven Bloggs").get(0);
        createClaims(claimant);

    }
    
    private void loadExpenseTypes() {
    	List types = allInstances(ExpenseType.class, false);
    	expenseTypes =  (ExpenseType[]) types.toArray(new ExpenseType[types.size()]);
    }

 
	private void createClaims(Employee employee) {
        for (int i = claimCount - 1; i >= 0; i--) {
            addRandomExpenseItems(createClaim(employee));
        }
    }

    private Claim createClaim(Employee employee) {
        Date rDate = new Date(random.nextInt(7) + 2000, random.nextInt(12) + 1, random.nextInt(28) + 1);
        Claim claim = getClaimFactory().createNewClaim(employee, rDate.toString());
        claim.setDateCreated(rDate);
        return claim;
    }

    private void addRandomExpenseItems(final Claim claim) {
        for (int count = random.nextInt(10); count >= 0; count--) {
            addRandomExpenseItem(claim);
        }
    }

    private void addRandomExpenseItem(final Claim claim) {
        ExpenseItem item = claim.createNewExpenseItem(expenseTypes[random.nextInt(8)]);
        populate(item);
        makePersistent(item);
    }

    private void populate(ExpenseItem item) {
        populateGeneral((ExpenseItem) item);

        if (item instanceof Journey) {
            populateJourney((Journey) item);
        }
        if (item instanceof Airfare) {
            populateAirfare((Airfare) item);
        }
        if (item instanceof CarRental) {
            populateCarRental((CarRental) item);
        }
        if (item instanceof PrivateCarJourney) {
            populatePrivateCarJourney((PrivateCarJourney) item);
        }

        if (item instanceof Hotel) {
            populateHotel((Hotel) item);
        }

    }

    private Money getRandomAmount() {
        return new Money(random.nextDouble() * 1000, "GBP");
    }

    private void populateJourney(Journey journey) {
        journey.setOrigin((new String[] { "London", "New York", "Tokyo" })[random.nextInt(3)]);
        journey.setDestination((new String[] { "Chicago", "Sydney", "Berlin" })[random.nextInt(3)]);
        journey.setAmount(getRandomAmount());
        journey.setReturnJourney(new Boolean(random.nextBoolean()));
    }

    private void populateAirfare(Airfare airfare) {
        airfare.setAirlineAndFlight((new String[] { "BA", "Air France", "RyanAir" })[random.nextInt(3)]);
    }

    private void populateCarRental(CarRental rental) {
        rental.setRentalCompany((new String[] { "Avis", "EasyCar", "Hertz" })[random.nextInt(3)]);
        rental.setNumberOfDays(random.nextInt(14) + 1);
    }

    private void populatePrivateCarJourney(PrivateCarJourney car) {
        car.setMileageRate((random.nextFloat() + .01) * 100.0);
        car.setTotalMiles(random.nextInt(1000) + 1);
    }

    private void populateGeneral(ExpenseItem item) {
        Date rDate = new Date(random.nextInt(7) + 2000, random.nextInt(12) + 1, random.nextInt(28) + 1);
        item.modifyAmount(getRandomAmount());
        item.modifyDescription(item.getClass().getSimpleName() + " " + rDate.toString());
        item.modifyDateIncurred(rDate);
        item.modifyProjectCode(getRandomProjectCode());
    }

    private void populateHotel(Hotel item) {
        item.setHotelURL((new String[] { "The Grand", "The Ritz", "Albert at Bay" })[random.nextInt(3)]);
        item.setAccommodation(getRandomAmount());
        item.setFood(getRandomAmount());
        item.setOther(getRandomAmount());
        item.setNumberOfNights(random.nextInt(14) + 1);
    }

   
    private void loadProjectCodes() {
    	List allCodes = allInstances(ProjectCode.class, false);
    	codes =  (ProjectCode[]) allCodes.toArray(new ProjectCode[allCodes.size()]);
    	numberOfCodes = allCodes.size();
    }
    
    private ProjectCode getRandomProjectCode() {
    	return codes[random.nextInt(numberOfCodes)];
    }
}