diff --git a/src/de/schildbach/pte/NasaProvider.java b/src/de/schildbach/pte/NasaProvider.java index 2d718d7d..a85869be 100644 --- a/src/de/schildbach/pte/NasaProvider.java +++ b/src/de/schildbach/pte/NasaProvider.java @@ -21,17 +21,14 @@ import java.io.IOException; import java.util.ArrayList; import java.util.Calendar; import java.util.Collections; -import java.util.Date; import java.util.GregorianCalendar; import java.util.List; import java.util.regex.Matcher; import java.util.regex.Pattern; import de.schildbach.pte.dto.Departure; -import de.schildbach.pte.dto.GetConnectionDetailsResult; import de.schildbach.pte.dto.Location; import de.schildbach.pte.dto.LocationType; -import de.schildbach.pte.dto.QueryConnectionsResult; import de.schildbach.pte.dto.QueryDeparturesResult; import de.schildbach.pte.dto.QueryDeparturesResult.Status; import de.schildbach.pte.dto.StationDepartures; @@ -61,7 +58,7 @@ public class NasaProvider extends AbstractHafasProvider public boolean hasCapabilities(final Capability... capabilities) { for (final Capability capability : capabilities) - if (capability == Capability.DEPARTURES) + if (capability == Capability.AUTOCOMPLETE_ONE_LINE || capability == Capability.DEPARTURES || capability == Capability.CONNECTIONS) return true; return false; @@ -262,6 +259,8 @@ public class NasaProvider extends AbstractHafasProvider if (ucType.equals("STB")) return 'T'; + if ("STRB".equals(ucType)) + return 'T'; if (ucType.equals("BSV")) return 'B'; @@ -275,23 +274,4 @@ public class NasaProvider extends AbstractHafasProvider return 0; } - - @Override - public QueryConnectionsResult queryConnections(final Location from, final Location via, final Location to, final Date date, final boolean dep, - final String products, final WalkSpeed walkSpeed) throws IOException - { - throw new UnsupportedOperationException(); - } - - @Override - public QueryConnectionsResult queryMoreConnections(String uri) throws IOException - { - throw new UnsupportedOperationException(); - } - - @Override - public GetConnectionDetailsResult getConnectionDetails(String connectionUri) throws IOException - { - throw new UnsupportedOperationException(); - } } diff --git a/test/de/schildbach/pte/live/NasaProviderLiveTest.java b/test/de/schildbach/pte/live/NasaProviderLiveTest.java index 32ac4ebd..fc8f4b50 100644 --- a/test/de/schildbach/pte/live/NasaProviderLiveTest.java +++ b/test/de/schildbach/pte/live/NasaProviderLiveTest.java @@ -17,13 +17,17 @@ package de.schildbach.pte.live; +import java.util.Date; import java.util.List; import org.junit.Test; import de.schildbach.pte.NasaProvider; +import de.schildbach.pte.NetworkProvider.WalkSpeed; import de.schildbach.pte.dto.Location; +import de.schildbach.pte.dto.LocationType; import de.schildbach.pte.dto.NearbyStationsResult; +import de.schildbach.pte.dto.QueryConnectionsResult; import de.schildbach.pte.dto.QueryDeparturesResult; /** @@ -32,6 +36,7 @@ import de.schildbach.pte.dto.QueryDeparturesResult; public class NasaProviderLiveTest { private final NasaProvider provider = new NasaProvider(); + private static final String ALL_PRODUCTS = "IRSUTBFC"; @Test public void autocomplete() throws Exception @@ -64,4 +69,14 @@ public class NasaProviderLiveTest System.out.println(result.stationDepartures); } + + @Test + public void shortConnection() throws Exception + { + final QueryConnectionsResult result = provider.queryConnections(new Location(LocationType.STATION, 13002, null, "Leipzig, Augustusplatz"), + null, new Location(LocationType.STATION, 8010205, null, "Leipzig Hbf"), new Date(), true, ALL_PRODUCTS, WalkSpeed.NORMAL); + System.out.println(result); + final QueryConnectionsResult moreResult = provider.queryMoreConnections(result.context); + System.out.println(moreResult); + } }