hiremanagetest.java

比较简单的vechilesystem代码还没经过优化

import junit.framework.TestCase;
import vehicleinfo.*;

import java.util.ArrayList;

import report.HireReport;

/**
 * Created by IntelliJ IDEA.
 * User: Ray
 * Date: 2008-4-8
 * Time: 13:48:18
 * To change this template use File | Settings | File Templates.
 */
public class HireManageTest extends TestCase {
    private HireManage hireManage;
    Vehicle vehicle1;
    public void setUp(){
        hireManage=new HireManage();
        vehicle1=new Vehicle("car","C8888","red",0);
        hireManage.vehicleRegister(vehicle1);
    }
    public void testVehicleRegister(){
        assertEquals(1,hireManage.getNumberOfVehicle());
        Vehicle vehicle2=new Vehicle("motor","C0000","blue",0);
        hireManage.vehicleRegister(vehicle2);
        assertEquals(2,hireManage.getNumberOfVehicle());
        hireManage.deleteVehicle(vehicle2);
        assertEquals(1,hireManage.getNumberOfVehicle());
    }
    public void testVehiclesSort(){
        Vehicle vehicle2=new Vehicle("bus","C0000","blue",0);
        hireManage.vehicleRegister(vehicle1);
        hireManage.vehicleRegister(vehicle2);
        hireManage.vehiclesSort();
        ArrayList<Vehicle> vehicles=new ArrayList<Vehicle>();
        vehicles.add(vehicle2);
        vehicles.add(vehicle1);
        assertEquals(true,assertVehicles(vehicles,hireManage.getAllVehicles()));
    }
    public void testHireAndRevertVehicle(){
        hireManage.hireVehicle("C8888");
        ArrayList<Vehicle> vehicles=hireManage.getAllVehicles();
        assertTrue(vehicles.get(0).isHired());
        assertEquals(HireReport.HIRE_REPORT_HEADER+"Type:car"+HireReport.NEWLINE+
                "Registration:C8888"+HireReport.NEWLINE+
                "Colour:red"+HireReport.NEWLINE+
                "Before be hired:0.0"+HireReport.NEWLINE+
                "Now:10.2"+HireReport.NEWLINE+
                HireReport.HIRE_REPORT_FOOTER+"459.00Yuan"+
                    HireReport.NEWLINE,hireManage.revertVehicle("C8888",10.2));
    }
    private boolean assertVehicles(ArrayList<Vehicle> vehicles, ArrayList<Vehicle> allVehicles) {
        for(int i=0;i<vehicles.size();i++){
            if(vehicles.get(i).compareTo(allVehicles.get(i))!=0)return false;
        }
        return true;
    }
}