mirror of
https://github.com/public-transport/db-vendo-client.git
synced 2025-02-23 15:19:35 +02:00
479 lines
12 KiB
JavaScript
479 lines
12 KiB
JavaScript
'use strict'
|
||
|
||
const tap = require('tap')
|
||
const isRoughlyEqual = require('is-roughly-equal')
|
||
const maxBy = require('lodash/maxBy')
|
||
const flatMap = require('lodash/flatMap')
|
||
|
||
const {createWhen} = require('./lib/util')
|
||
const createClient = require('../..')
|
||
const dbProfile = require('../../p/db')
|
||
const products = require('../../p/db/products')
|
||
const {
|
||
station: createValidateStation,
|
||
trip: createValidateTrip
|
||
} = require('./lib/validators')
|
||
const createValidate = require('./lib/validate-fptf-with')
|
||
const testJourneysStationToStation = require('./lib/journeys-station-to-station')
|
||
const testJourneysStationToAddress = require('./lib/journeys-station-to-address')
|
||
const testJourneysStationToPoi = require('./lib/journeys-station-to-poi')
|
||
const testEarlierLaterJourneys = require('./lib/earlier-later-journeys')
|
||
const testLegCycleAlternatives = require('./lib/leg-cycle-alternatives')
|
||
const testRefreshJourney = require('./lib/refresh-journey')
|
||
const journeysFailsWithNoProduct = require('./lib/journeys-fails-with-no-product')
|
||
const testDepartures = require('./lib/departures')
|
||
const testDeparturesInDirection = require('./lib/departures-in-direction')
|
||
const testArrivals = require('./lib/arrivals')
|
||
const testJourneysWithDetour = require('./lib/journeys-with-detour')
|
||
const testReachableFrom = require('./lib/reachable-from')
|
||
const testServerInfo = require('./lib/server-info')
|
||
|
||
const isObj = o => o !== null && 'object' === typeof o && !Array.isArray(o)
|
||
const minute = 60 * 1000
|
||
|
||
const when = createWhen('Europe/Berlin', 'de-DE')
|
||
|
||
const cfg = {
|
||
when,
|
||
stationCoordsOptional: false,
|
||
products,
|
||
minLatitude: 46.673100,
|
||
maxLatitude: 55.030671,
|
||
minLongitude: 6.896517,
|
||
maxLongitude: 16.180237
|
||
}
|
||
|
||
const validateStation = createValidateStation(cfg)
|
||
|
||
const validate = createValidate(cfg, {
|
||
station: validateStation
|
||
})
|
||
|
||
const assertValidPrice = (t, p) => {
|
||
t.ok(p)
|
||
if (p.amount !== null) {
|
||
t.equal(typeof p.amount, 'number')
|
||
t.ok(p.amount > 0)
|
||
}
|
||
if (p.hint !== null) {
|
||
t.equal(typeof p.hint, 'string')
|
||
t.ok(p.hint)
|
||
}
|
||
}
|
||
|
||
const client = createClient(dbProfile, 'public-transport/hafas-client:test')
|
||
|
||
const berlinHbf = '8011160'
|
||
const münchenHbf = '8000261'
|
||
const jungfernheide = '8011167'
|
||
const blnSchwedterStr = '732652'
|
||
const westhafen = '8089116'
|
||
const wedding = '8089131'
|
||
const württembergallee = '731084'
|
||
const regensburgHbf = '8000309'
|
||
const blnOstbahnhof = '8010255'
|
||
const blnTiergarten = '8089091'
|
||
const blnJannowitzbrücke = '8089019'
|
||
const potsdamHbf = '8012666'
|
||
const berlinSüdkreuz = '8011113'
|
||
const kölnHbf = '8000207'
|
||
|
||
tap.test('journeys – Berlin Schwedter Str. to München Hbf', async (t) => {
|
||
const res = await client.journeys(blnSchwedterStr, münchenHbf, {
|
||
results: 4,
|
||
departure: when,
|
||
stopovers: true
|
||
})
|
||
|
||
await testJourneysStationToStation({
|
||
test: t,
|
||
res,
|
||
validate,
|
||
fromId: blnSchwedterStr,
|
||
toId: münchenHbf
|
||
})
|
||
// todo: find a journey where there pricing info is always available
|
||
for (let journey of res.journeys) {
|
||
if (journey.price) assertValidPrice(t, journey.price)
|
||
}
|
||
|
||
t.end()
|
||
})
|
||
|
||
// todo: journeys, only one product
|
||
|
||
tap.test('journeys – fails with no product', (t) => {
|
||
journeysFailsWithNoProduct({
|
||
test: t,
|
||
fetchJourneys: client.journeys,
|
||
fromId: blnSchwedterStr,
|
||
toId: münchenHbf,
|
||
when,
|
||
products
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
tap.test('Berlin Schwedter Str. to Torfstraße 17', async (t) => {
|
||
const torfstr = {
|
||
type: 'location',
|
||
address: 'Torfstraße 17',
|
||
latitude: 52.5416823,
|
||
longitude: 13.3491223
|
||
}
|
||
const res = await client.journeys(blnSchwedterStr, torfstr, {
|
||
results: 3,
|
||
departure: when
|
||
})
|
||
|
||
await testJourneysStationToAddress({
|
||
test: t,
|
||
res,
|
||
validate,
|
||
fromId: blnSchwedterStr,
|
||
to: torfstr
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
tap.test('Berlin Schwedter Str. to ATZE Musiktheater', async (t) => {
|
||
const atze = {
|
||
type: 'location',
|
||
id: '991598902',
|
||
poi: true,
|
||
name: 'ATZE Musiktheater',
|
||
latitude: 52.542417,
|
||
longitude: 13.350437
|
||
}
|
||
const res = await client.journeys(blnSchwedterStr, atze, {
|
||
results: 3,
|
||
departure: when
|
||
})
|
||
|
||
await testJourneysStationToPoi({
|
||
test: t,
|
||
res,
|
||
validate,
|
||
fromId: blnSchwedterStr,
|
||
to: atze
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
tap.test('journeys: via works – with detour', async (t) => {
|
||
// Going from Westhafen to Wedding via Württembergalle without detour
|
||
// is currently impossible. We check if the routing engine computes a detour.
|
||
const res = await client.journeys(westhafen, wedding, {
|
||
via: württembergallee,
|
||
results: 1,
|
||
departure: when,
|
||
stopovers: true
|
||
})
|
||
|
||
await testJourneysWithDetour({
|
||
test: t,
|
||
res,
|
||
validate,
|
||
detourIds: [württembergallee]
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
// todo: walkingSpeed "Berlin - Charlottenburg, Hallerstraße" -> jungfernheide
|
||
// todo: without detour
|
||
|
||
tap.test('earlier/later journeys, Jungfernheide -> München Hbf', async (t) => {
|
||
await testEarlierLaterJourneys({
|
||
test: t,
|
||
fetchJourneys: client.journeys,
|
||
validate,
|
||
fromId: jungfernheide,
|
||
toId: münchenHbf,
|
||
when
|
||
})
|
||
|
||
t.end()
|
||
})
|
||
|
||
tap.skip('journeys – leg cycle & alternatives', async (t) => {
|
||
await testLegCycleAlternatives({
|
||
test: t,
|
||
fetchJourneys: client.journeys,
|
||
fromId: blnTiergarten,
|
||
toId: blnJannowitzbrücke
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
tap.test('refreshJourney', async (t) => {
|
||
await testRefreshJourney({
|
||
test: t,
|
||
fetchJourneys: client.journeys,
|
||
refreshJourney: client.refreshJourney,
|
||
validate,
|
||
fromId: jungfernheide,
|
||
toId: münchenHbf,
|
||
when
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
tap.test('journeysFromTrip – U Mehringdamm to U Naturkundemuseum, reroute to Spittelmarkt.', async (t) => {
|
||
const blnMehringdamm = '730939'
|
||
const blnStadtmitte = '732541'
|
||
const blnNaturkundemuseum = '732539'
|
||
const blnSpittelmarkt = '732543'
|
||
|
||
const isU6Leg = leg => (
|
||
leg.line && leg.line.name
|
||
&& leg.line.name.toUpperCase().replace(/\s+/g, '') === 'U6'
|
||
)
|
||
const sameStopOrStation = (stopA) => (stopB) => {
|
||
if (stopA.id && stopB.id && stopA.id === stopB.id) return true
|
||
const statA = stopA.stat && stopA.stat.id || NaN
|
||
const statB = stopB.stat && stopB.stat.id || NaN
|
||
return (statA === statB || stopA.id === statB || stopB.id === statA)
|
||
}
|
||
const departureOf = st => +new Date(st.departure || st.scheduledDeparture)
|
||
const arrivalOf = st => +new Date(st.arrival || st.scheduledArrival)
|
||
|
||
// `journeysFromTrip` only supports queries *right now*, so we can't use `when` as in all
|
||
// other tests. To make the test less brittle, we pick a connection that is served all night. 🙄
|
||
const when = new Date()
|
||
const validate = createValidate({...cfg, when})
|
||
|
||
const findTripBetween = async (stopAId, stopBId, products = {}) => {
|
||
const {journeys} = await client.journeys(stopAId, stopBId, {
|
||
departure: new Date(when - 10 * minute),
|
||
transfers: 0, products,
|
||
results: 8, stopovers: false, remarks: false,
|
||
})
|
||
for (const j of journeys) {
|
||
const l = j.legs.find(isU6Leg)
|
||
if (!l) continue
|
||
const t = await client.trip(l.tripId, l.line && l.line.name, {
|
||
stopovers: true, remarks: false
|
||
})
|
||
|
||
const hasStoppedAtA = t.stopovers
|
||
.filter(st => departureOf(st) < Date.now()) // todo: <= ?
|
||
.find(sameStopOrStation({id: stopAId}))
|
||
const willStopAtB = t.stopovers
|
||
.filter(st => arrivalOf(st) > Date.now()) // todo: >= ?
|
||
.find(sameStopOrStation({id: stopBId}))
|
||
|
||
if (hasStoppedAtA && willStopAtB) {
|
||
const prevStopover = maxBy(pastStopovers, departureOf)
|
||
return {trip: t, prevStopover}
|
||
}
|
||
}
|
||
return {trip: null, prevStopover: null}
|
||
}
|
||
|
||
// Find a vehicle from U Mehringdamm to U Stadtmitte (to the north) that is currently
|
||
// between these two stations.
|
||
const {trip, prevStopover} = await findTripBetween(blnMehringdamm, blnStadtmitte, {
|
||
regionalExp: false, regional: false, suburban: false
|
||
})
|
||
t.ok(trip, 'precondition failed: trip not found')
|
||
t.ok(prevStopover, 'precondition failed: previous stopover missing')
|
||
|
||
// todo: "Error: Suche aus dem Zug: Vor Abfahrt des Zuges"
|
||
const newJourneys = await client.journeysFromTrip(trip.id, prevStopover, blnSpittelmarkt, {
|
||
results: 3, stopovers: true, remarks: false
|
||
})
|
||
|
||
// Validate with fake prices.
|
||
const withFakePrice = (j) => {
|
||
const clone = Object.assign({}, j)
|
||
clone.price = {amount: 123, currency: 'EUR'}
|
||
return clone
|
||
}
|
||
validate(t, newJourneys.map(withFakePrice), 'newJourneys', 'journeysFromTrip')
|
||
|
||
for (let i = 0; i < newJourneys.length; i++) {
|
||
const j = newJourneys[i]
|
||
const n = `newJourneys[${i}]`
|
||
|
||
const legOnTrip = j.legs.find(l => l.tripId === trip.id)
|
||
t.ok(legOnTrip, n + ': leg with trip ID not found')
|
||
t.equal(last(legOnTrip.stopovers).stop.id, blnStadtmitte)
|
||
}
|
||
})
|
||
|
||
tap.test('trip details', async (t) => {
|
||
const res = await client.journeys(berlinHbf, münchenHbf, {
|
||
results: 1, departure: when
|
||
})
|
||
|
||
const p = res.journeys[0].legs.find(l => !l.walking)
|
||
t.ok(p.tripId, 'precondition failed')
|
||
t.ok(p.line.name, 'precondition failed')
|
||
const trip = await client.trip(p.tripId, p.line.name, {when})
|
||
|
||
const validateTrip = createValidateTrip(cfg)
|
||
const validate = createValidate(cfg, {
|
||
trip: (validate, trip, name) => {
|
||
trip = Object.assign({}, trip)
|
||
if (!trip.direction) trip.direction = 'foo' // todo, see #49
|
||
validateTrip(validate, trip, name)
|
||
}
|
||
})
|
||
validate(t, trip, 'trip', 'trip')
|
||
|
||
t.end()
|
||
})
|
||
|
||
tap.test('departures at Berlin Schwedter Str.', async (t) => {
|
||
const departures = await client.departures(blnSchwedterStr, {
|
||
duration: 5, when,
|
||
})
|
||
|
||
await testDepartures({
|
||
test: t,
|
||
departures,
|
||
validate,
|
||
id: blnSchwedterStr
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
tap.test('departures with station object', async (t) => {
|
||
const deps = await client.departures({
|
||
type: 'station',
|
||
id: jungfernheide,
|
||
name: 'Berlin Jungfernheide',
|
||
location: {
|
||
type: 'location',
|
||
latitude: 1.23,
|
||
longitude: 2.34
|
||
}
|
||
}, {when})
|
||
|
||
validate(t, deps, 'departures', 'departures')
|
||
t.end()
|
||
})
|
||
|
||
tap.test('departures at Berlin Hbf in direction of Berlin Ostbahnhof', async (t) => {
|
||
await testDeparturesInDirection({
|
||
test: t,
|
||
fetchDepartures: client.departures,
|
||
fetchTrip: client.trip,
|
||
id: berlinHbf,
|
||
directionIds: [blnOstbahnhof, '8089185', '732676'],
|
||
when,
|
||
validate
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
tap.test('arrivals at Berlin Schwedter Str.', async (t) => {
|
||
const arrivals = await client.arrivals(blnSchwedterStr, {
|
||
duration: 5, when,
|
||
})
|
||
|
||
await testArrivals({
|
||
test: t,
|
||
arrivals,
|
||
validate,
|
||
id: blnSchwedterStr
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
tap.test('nearby Berlin Jungfernheide', async (t) => {
|
||
const nearby = await client.nearby({
|
||
type: 'location',
|
||
latitude: 52.530273,
|
||
longitude: 13.299433
|
||
}, {
|
||
results: 2, distance: 400
|
||
})
|
||
|
||
validate(t, nearby, 'locations', 'nearby')
|
||
|
||
t.equal(nearby.length, 2)
|
||
|
||
const s0 = nearby[0]
|
||
t.equal(s0.id, jungfernheide)
|
||
t.equal(s0.name, 'Berlin Jungfernheide')
|
||
t.ok(isRoughlyEqual(.0005, s0.location.latitude, 52.530408))
|
||
t.ok(isRoughlyEqual(.0005, s0.location.longitude, 13.299424))
|
||
t.ok(s0.distance >= 0)
|
||
t.ok(s0.distance <= 100)
|
||
|
||
t.end()
|
||
})
|
||
|
||
tap.test('locations named Jungfernheide', async (t) => {
|
||
const locations = await client.locations('Jungfernheide', {
|
||
results: 10
|
||
})
|
||
|
||
validate(t, locations, 'locations', 'locations')
|
||
t.ok(locations.length <= 10)
|
||
t.ok(locations.some((l) => {
|
||
return l.station && l.station.id === jungfernheide || l.id === jungfernheide
|
||
}), 'Jungfernheide not found')
|
||
|
||
t.end()
|
||
})
|
||
|
||
tap.test('stop', async (t) => {
|
||
const s = await client.stop(regensburgHbf)
|
||
|
||
validate(t, s, ['stop', 'station'], 'stop')
|
||
t.equal(s.id, regensburgHbf)
|
||
|
||
t.end()
|
||
})
|
||
|
||
tap.test('line with additionalName', async (t) => {
|
||
const departures = await client.departures(potsdamHbf, {
|
||
when,
|
||
duration: 12 * 60, // 12 minutes
|
||
products: {bus: false, suburban: false, tram: false}
|
||
})
|
||
t.ok(departures.some(d => d.line && d.line.additionalName))
|
||
t.end()
|
||
})
|
||
|
||
tap.test('radar', async (t) => {
|
||
const vehicles = await client.radar({
|
||
north: 52.52411,
|
||
west: 13.41002,
|
||
south: 52.51942,
|
||
east: 13.41709
|
||
}, {
|
||
duration: 5 * 60, when
|
||
})
|
||
|
||
validate(t, vehicles, 'movements', 'vehicles')
|
||
t.end()
|
||
})
|
||
|
||
tap.test('reachableFrom', async (t) => {
|
||
const torfstr17 = {
|
||
type: 'location',
|
||
address: 'Torfstraße 17',
|
||
latitude: 52.5416823,
|
||
longitude: 13.3491223
|
||
}
|
||
|
||
await testReachableFrom({
|
||
test: t,
|
||
reachableFrom: client.reachableFrom,
|
||
address: torfstr17,
|
||
when,
|
||
maxDuration: 15,
|
||
validate
|
||
})
|
||
t.end()
|
||
})
|
||
|
||
tap.test('serverInfo works', async (t) => {
|
||
await testServerInfo({
|
||
test: t,
|
||
fetchServerInfo: client.serverInfo,
|
||
})
|
||
})
|