aris/packages/aris-source-tfl/src/tfl-source.test.ts

import { Context } from "@aris/core"
import { LocationKey, type Location } from "@aris/source-location"
import { describe, expect, test } from "bun:test"

import type {
	ITflApi,
	StationLocation,
	TflAlertSeverity,
	TflLineId,
	TflLineStatus,
} from "./types.ts"

import fixtures from "../fixtures/tfl-responses.json"
import { TflSource } from "./tfl-source.ts"

// Mock API that returns fixture data
class FixtureTflApi implements ITflApi {
	async fetchLineStatuses(_lines?: TflLineId[]): Promise<TflLineStatus[]> {
		const statuses: TflLineStatus[] = []

		for (const line of fixtures.lineStatuses as Record<string, unknown>[]) {
			for (const status of line.lineStatuses as Record<string, unknown>[]) {
				const severityCode = status.statusSeverity as number
				const severity = this.mapSeverity(severityCode)
				if (severity) {
					statuses.push({
						lineId: line.id as TflLineId,
						lineName: line.name as string,
						severity,
						description: (status.reason as string) ?? (status.statusSeverityDescription as string),
					})
				}
			}
		}

		return statuses
	}

	async fetchStations(): Promise<StationLocation[]> {
		const stationMap = new Map<string, StationLocation>()

		for (const [lineId, stops] of Object.entries(fixtures.stopPoints)) {
			for (const stop of stops as Record<string, unknown>[]) {
				const id = stop.naptanId as string
				const existing = stationMap.get(id)
				if (existing) {
					if (!existing.lines.includes(lineId as TflLineId)) {
						existing.lines.push(lineId as TflLineId)
					}
				} else {
					stationMap.set(id, {
						id,
						name: stop.commonName as string,
						lat: stop.lat as number,
						lng: stop.lon as number,
						lines: [lineId as TflLineId],
					})
				}
			}
		}

		return Array.from(stationMap.values())
	}

	private mapSeverity(code: number): TflAlertSeverity | null {
		const map: Record<number, TflAlertSeverity | null> = {
			1: "closure",
			2: "closure",
			3: "closure",
			4: "closure",
			5: "closure",
			6: "major-delays",
			7: "major-delays",
			8: "major-delays",
			9: "minor-delays",
			10: null,
		}
		return map[code] ?? null
	}
}

function createContext(location?: Location): Context {
	const ctx = new Context(new Date("2026-01-15T12:00:00Z"))
	if (location) {
		ctx.set([[LocationKey, location]])
	}
	return ctx
}

describe("TflSource", () => {
	const api = new FixtureTflApi()

	describe("interface", () => {
		test("has correct id", () => {
			const source = new TflSource({ client: api })
			expect(source.id).toBe("aris.tfl")
		})

		test("depends on location", () => {
			const source = new TflSource({ client: api })
			expect(source.dependencies).toEqual(["aris.location"])
		})

		test("implements fetchItems", () => {
			const source = new TflSource({ client: api })
			expect(source.fetchItems).toBeDefined()
		})

		test("throws if neither client nor apiKey provided", () => {
			expect(() => new TflSource({})).toThrow("Either client or apiKey must be provided")
		})
	})

	describe("setLinesOfInterest", () => {
		const lineFilteringApi: ITflApi = {
			async fetchLineStatuses(lines?: TflLineId[]): Promise<TflLineStatus[]> {
				const all: TflLineStatus[] = [
					{
						lineId: "northern",
						lineName: "Northern",
						severity: "minor-delays",
						description: "Delays",
					},
					{
						lineId: "central",
						lineName: "Central",
						severity: "closure",
						description: "Closed",
					},
				]
				return lines ? all.filter((s) => lines.includes(s.lineId)) : all
			},
			async fetchStations(): Promise<StationLocation[]> {
				return []
			},
		}

		test("changes which lines are fetched", async () => {
			const source = new TflSource({ client: lineFilteringApi })
			const before = await source.fetchItems(createContext())
			expect(before.length).toBe(2)

			source.setLinesOfInterest(["northern"])
			const after = await source.fetchItems(createContext())

			expect(after.length).toBe(1)
			expect(after[0]!.data.line).toBe("northern")
		})

		test("DEFAULT_LINES_OF_INTEREST restores all lines", async () => {
			const source = new TflSource({
				client: lineFilteringApi,
				lines: ["northern"],
			})
			const filtered = await source.fetchItems(createContext())
			expect(filtered.length).toBe(1)

			source.setLinesOfInterest([...TflSource.DEFAULT_LINES_OF_INTEREST])
			const all = await source.fetchItems(createContext())

			expect(all.length).toBe(2)
		})
	})

	describe("fetchItems", () => {
		test("returns feed items array", async () => {
			const source = new TflSource({ client: api })
			const items = await source.fetchItems(createContext())
			expect(Array.isArray(items)).toBe(true)
		})

		test("feed items have correct base structure", async () => {
			const source = new TflSource({ client: api })
			const location: Location = {
				lat: 51.5074,
				lng: -0.1278,
				accuracy: 10,
				timestamp: new Date(),
			}
			const items = await source.fetchItems(createContext(location))

			for (const item of items) {
				expect(typeof item.id).toBe("string")
				expect(item.id).toMatch(/^tfl-alert-/)
				expect(item.type).toBe("tfl-alert")
				expect(item.signals).toBeDefined()
				expect(typeof item.signals!.urgency).toBe("number")
				expect(item.timestamp).toBeInstanceOf(Date)
			}
		})

		test("feed items have correct data structure", async () => {
			const source = new TflSource({ client: api })
			const location: Location = {
				lat: 51.5074,
				lng: -0.1278,
				accuracy: 10,
				timestamp: new Date(),
			}
			const items = await source.fetchItems(createContext(location))

			for (const item of items) {
				expect(typeof item.data.line).toBe("string")
				expect(typeof item.data.lineName).toBe("string")
				expect(["minor-delays", "major-delays", "closure"]).toContain(item.data.severity)
				expect(typeof item.data.description).toBe("string")
				expect(
					item.data.closestStationDistance === null ||
						typeof item.data.closestStationDistance === "number",
				).toBe(true)
			}
		})

		test("feed item ids are unique", async () => {
			const source = new TflSource({ client: api })
			const items = await source.fetchItems(createContext())

			const ids = items.map((item) => item.id)
			const uniqueIds = new Set(ids)
			expect(uniqueIds.size).toBe(ids.length)
		})

		test("feed items are sorted by urgency descending", async () => {
			const source = new TflSource({ client: api })
			const items = await source.fetchItems(createContext())

			for (let i = 1; i < items.length; i++) {
				const prev = items[i - 1]!
				const curr = items[i]!
				expect(prev.signals!.urgency).toBeGreaterThanOrEqual(curr.signals!.urgency!)
			}
		})

		test("urgency values match severity levels", async () => {
			const source = new TflSource({ client: api })
			const items = await source.fetchItems(createContext())

			const severityUrgency: Record<string, number> = {
				closure: 1.0,
				"major-delays": 0.8,
				"minor-delays": 0.6,
			}

			for (const item of items) {
				expect(item.signals!.urgency).toBe(severityUrgency[item.data.severity]!)
			}
		})

		test("closestStationDistance is number when location provided", async () => {
			const source = new TflSource({ client: api })
			const location: Location = {
				lat: 51.5074,
				lng: -0.1278,
				accuracy: 10,
				timestamp: new Date(),
			}
			const items = await source.fetchItems(createContext(location))

			for (const item of items) {
				expect(typeof item.data.closestStationDistance).toBe("number")
				expect(item.data.closestStationDistance!).toBeGreaterThan(0)
			}
		})

		test("closestStationDistance is null when no location provided", async () => {
			const source = new TflSource({ client: api })
			const items = await source.fetchItems(createContext())

			for (const item of items) {
				expect(item.data.closestStationDistance).toBeNull()
			}
		})
	})

	describe("actions", () => {
		test("listActions returns set-lines-of-interest", async () => {
			const source = new TflSource({ client: api })
			const actions = await source.listActions()

			expect(actions["set-lines-of-interest"]).toBeDefined()
			expect(actions["set-lines-of-interest"]!.id).toBe("set-lines-of-interest")
		})

		test("executeAction set-lines-of-interest updates lines", async () => {
			const lineFilteringApi: ITflApi = {
				async fetchLineStatuses(lines?: TflLineId[]): Promise<TflLineStatus[]> {
					const all: TflLineStatus[] = [
						{
							lineId: "northern",
							lineName: "Northern",
							severity: "minor-delays",
							description: "Delays",
						},
						{
							lineId: "central",
							lineName: "Central",
							severity: "closure",
							description: "Closed",
						},
					]
					return lines ? all.filter((s) => lines.includes(s.lineId)) : all
				},
				async fetchStations(): Promise<StationLocation[]> {
					return []
				},
			}

			const source = new TflSource({ client: lineFilteringApi })
			await source.executeAction("set-lines-of-interest", ["northern"])

			const items = await source.fetchItems(createContext())
			expect(items.length).toBe(1)
			expect(items[0]!.data.line).toBe("northern")
		})

		test("executeAction throws on invalid input", async () => {
			const source = new TflSource({ client: api })

			await expect(source.executeAction("set-lines-of-interest", "not-an-array")).rejects.toThrow()
		})

		test("executeAction throws for unknown action", async () => {
			const source = new TflSource({ client: api })

			await expect(source.executeAction("nonexistent", {})).rejects.toThrow("Unknown action")
		})
	})
})

describe("TfL Fixture Data Shape", () => {
	test("fixtures have expected structure", () => {
		expect(typeof fixtures.fetchedAt).toBe("string")
		expect(Array.isArray(fixtures.lineStatuses)).toBe(true)
		expect(typeof fixtures.stopPoints).toBe("object")
	})

	test("line statuses have required fields", () => {
		for (const line of fixtures.lineStatuses as Record<string, unknown>[]) {
			expect(typeof line.id).toBe("string")
			expect(typeof line.name).toBe("string")
			expect(Array.isArray(line.lineStatuses)).toBe(true)

			for (const status of line.lineStatuses as Record<string, unknown>[]) {
				expect(typeof status.statusSeverity).toBe("number")
				expect(typeof status.statusSeverityDescription).toBe("string")
			}
		}
	})

	test("stop points have required fields", () => {
		for (const [lineId, stops] of Object.entries(fixtures.stopPoints)) {
			expect(typeof lineId).toBe("string")
			expect(Array.isArray(stops)).toBe(true)

			for (const stop of stops as Record<string, unknown>[]) {
				expect(typeof stop.naptanId).toBe("string")
				expect(typeof stop.commonName).toBe("string")
				expect(typeof stop.lat).toBe("number")
				expect(typeof stop.lon).toBe("number")
			}
		}
	})
})