TravelModeWrapperTest.java

package org.opentripplanner.api.thrift.util;

import static org.junit.Assert.assertEquals;

import java.util.HashMap;
import java.util.Map;

import org.junit.Test;
import org.opentripplanner.api.thrift.definition.TravelMode;
import org.opentripplanner.routing.core.TraverseMode;

public class TravelModeWrapperTest {

    @Test
    public void testToTraverseMode() {
        Map<TravelMode, TraverseMode> modeMap = new HashMap<TravelMode, TraverseMode>();
        modeMap.put(TravelMode.BICYCLE, TraverseMode.BICYCLE);
        modeMap.put(TravelMode.CAR, TraverseMode.CAR);
        modeMap.put(TravelMode.CUSTOM_MOTOR_VEHICLE, TraverseMode.CUSTOM_MOTOR_VEHICLE);
        modeMap.put(TravelMode.WALK, TraverseMode.WALK);
        modeMap.put(TravelMode.ANY_TRAIN, TraverseMode.TRAINISH);
        modeMap.put(TravelMode.ANY_TRANSIT, TraverseMode.TRANSIT);

        for (TravelMode travelMode : modeMap.keySet()) {
            TraverseMode expectedTraverseMode = modeMap.get(travelMode);
            TraverseMode actualTraverseMode = (new TravelModeWrapper(travelMode)).toTraverseMode();
            assertEquals(expectedTraverseMode, actualTraverseMode);
        }
    }
}