diff --git a/src/core/game/PlayerImpl.ts b/src/core/game/PlayerImpl.ts index 7d762efc2..e2d3d14ee 100644 --- a/src/core/game/PlayerImpl.ts +++ b/src/core/game/PlayerImpl.ts @@ -1317,7 +1317,7 @@ export class PlayerImpl implements Player { canUpgrade, cost, overlappingRailroads: buildNew - ? rail.overlappingRailroads(canBuild as TileRef) + ? rail.overlappingRailroads(u, canBuild as TileRef) : [], ghostRailPaths: buildNew ? rail.computeGhostRailPaths(u, canBuild as TileRef) diff --git a/src/core/game/RailNetwork.ts b/src/core/game/RailNetwork.ts index 420e59c29..c86921218 100644 --- a/src/core/game/RailNetwork.ts +++ b/src/core/game/RailNetwork.ts @@ -8,7 +8,7 @@ export interface RailNetwork { removeStation(unit: Unit): void; findStationsPath(from: TrainStation, to: TrainStation): TrainStation[]; stationManager(): StationManager; - overlappingRailroads(tile: TileRef): TileRef[]; + overlappingRailroads(unitType: UnitType, tile: TileRef): TileRef[]; computeGhostRailPaths(unitType: UnitType, tile: TileRef): TileRef[][]; recomputeClusters(): void; } diff --git a/src/core/game/RailNetworkImpl.ts b/src/core/game/RailNetworkImpl.ts index 9144734a7..e955790d2 100644 --- a/src/core/game/RailNetworkImpl.ts +++ b/src/core/game/RailNetworkImpl.ts @@ -223,7 +223,10 @@ export class RailNetworkImpl implements RailNetwork { return editedClusters.size !== 0; } - overlappingRailroads(tile: TileRef): TileRef[] { + overlappingRailroads(unitType: UnitType, tile: TileRef): TileRef[] { + if (![UnitType.City, UnitType.Port, UnitType.Factory].includes(unitType)) { + return []; + } const tiles = new Set(); for (const railroad of this.railGrid.query(tile, this.stationRadius)) { for (const t of railroad.tiles) { diff --git a/tests/core/game/RailNetwork.test.ts b/tests/core/game/RailNetwork.test.ts index e63fadef2..f0ed13fc3 100644 --- a/tests/core/game/RailNetwork.test.ts +++ b/tests/core/game/RailNetwork.test.ts @@ -178,7 +178,7 @@ describe("RailNetworkImpl", () => { }; (network as any).railGrid = railGridMock; - const result = network.overlappingRailroads(tile); + const result = network.overlappingRailroads(UnitType.City, tile); expect(railGridMock.query).toHaveBeenCalledWith(tile, 3); expect(result).toEqual([42, 45, 50, 60]); // Deduplicated and sorted @@ -189,10 +189,30 @@ describe("RailNetworkImpl", () => { const railGridMock = { query: vi.fn(() => new Set()) }; (network as any).railGrid = railGridMock; - const result = network.overlappingRailroads(tile); + const result = network.overlappingRailroads(UnitType.City, tile); expect(result).toEqual([]); }); + + test.each([ + UnitType.MissileSilo, + UnitType.DefensePost, + UnitType.SAMLauncher, + ])( + "returns empty array for %s which cannot snap to railroads", + (unitType) => { + const tile = 42 as any; + const railGridMock = { + query: vi.fn(() => new Set([{ tiles: [50, 42, 60] }])), + }; + (network as any).railGrid = railGridMock; + + const result = network.overlappingRailroads(unitType, tile); + + expect(result).toEqual([]); + expect(railGridMock.query).not.toHaveBeenCalled(); + }, + ); }); describe("computeGhostRailPaths", () => {