MultipleFareServiceTest.java

/* This program is free software: you can redistribute it and/or
 modify it under the terms of the GNU Lesser General Public License
 as published by the Free Software Foundation, either version 3 of
 the License, or (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>. */

package org.opentripplanner.routing.fares;

import java.util.ArrayList;
import java.util.Arrays;

import junit.framework.TestCase;

import org.opentripplanner.routing.core.Fare;
import org.opentripplanner.routing.core.Fare.FareType;
import org.opentripplanner.routing.core.WrappedCurrency;
import org.opentripplanner.routing.services.FareService;
import org.opentripplanner.routing.spt.GraphPath;

/**
 * 
 * @author laurent
 */
public class MultipleFareServiceTest extends TestCase {

    private class SimpleFareService implements FareService {

        private Fare fare;

        private SimpleFareService(Fare fare) {
            this.fare = fare;
        }

        @Override
        public Fare getCost(GraphPath path) {
            return fare;
        }
    }

    public void testAddingMultipleFareService() {

        Fare fare1 = new Fare();
        fare1.addFare(FareType.regular, new WrappedCurrency("EUR"), 100);
        FareService fs1 = new SimpleFareService(fare1);

        Fare fare2 = new Fare();
        fare2.addFare(FareType.regular, new WrappedCurrency("EUR"), 140);
        fare2.addFare(FareType.student, new WrappedCurrency("EUR"), 120);
        FareService fs2 = new SimpleFareService(fare2);

        /*
         * Note: this fare is not very representative, as you should probably always compute a
         * "regular" fare in case you want to add bike and transit fares.
         */
        Fare fare3 = new Fare();
        fare3.addFare(FareType.student, new WrappedCurrency("EUR"), 80);
        FareService fs3 = new SimpleFareService(fare3);

        AddingMultipleFareService mfs = new AddingMultipleFareService(new ArrayList<FareService>());
        Fare fare = mfs.getCost(null);
        assertNull(fare);

        mfs = new AddingMultipleFareService(Arrays.asList(fs1));
        fare = mfs.getCost(null);
        assertEquals(100, fare.getFare(FareType.regular).getCents());
        assertEquals(null, fare.getFare(FareType.student));

        mfs = new AddingMultipleFareService(Arrays.asList(fs2));
        fare = mfs.getCost(null);
        assertEquals(140, fare.getFare(FareType.regular).getCents());
        assertEquals(120, fare.getFare(FareType.student).getCents());

        mfs = new AddingMultipleFareService(Arrays.asList(fs1, fs2));
        fare = mfs.getCost(null);
        assertEquals(240, fare.getFare(FareType.regular).getCents());
        assertEquals(220, fare.getFare(FareType.student).getCents());

        mfs = new AddingMultipleFareService(Arrays.asList(fs2, fs1));
        fare = mfs.getCost(null);
        assertEquals(240, fare.getFare(FareType.regular).getCents());
        assertEquals(220, fare.getFare(FareType.student).getCents());

        mfs = new AddingMultipleFareService(Arrays.asList(fs1, fs3));
        fare = mfs.getCost(null);
        assertEquals(100, fare.getFare(FareType.regular).getCents());
        assertEquals(180, fare.getFare(FareType.student).getCents());

        mfs = new AddingMultipleFareService(Arrays.asList(fs3, fs1));
        fare = mfs.getCost(null);
        assertEquals(100, fare.getFare(FareType.regular).getCents());
        assertEquals(180, fare.getFare(FareType.student).getCents());

        mfs = new AddingMultipleFareService(Arrays.asList(fs1, fs2, fs3));
        fare = mfs.getCost(null);
        assertEquals(240, fare.getFare(FareType.regular).getCents());
        assertEquals(300, fare.getFare(FareType.student).getCents());
    }
}