diff --git a/src/models/actions/update/UpdateTrackScan.ts b/src/models/actions/update/UpdateTrackScan.ts index 6bdc83e..0e0a2e5 100644 --- a/src/models/actions/update/UpdateTrackScan.ts +++ b/src/models/actions/update/UpdateTrackScan.ts @@ -1,9 +1,9 @@ import { IsBoolean, IsInt, IsOptional, IsPositive } from 'class-validator'; import { getConnection } from 'typeorm'; import { RunnerNotFoundError } from '../../../errors/RunnerErrors'; -import { ScanStationNotFoundError } from '../../../errors/ScanStationErrors'; +import { TrackNotFoundError } from '../../../errors/TrackErrors'; import { Runner } from '../../entities/Runner'; -import { ScanStation } from '../../entities/ScanStation'; +import { Track } from '../../entities/Track'; import { TrackScan } from '../../entities/TrackScan'; /** @@ -38,7 +38,7 @@ export abstract class UpdateTrackScan { */ @IsInt() @IsPositive() - public station: number; + public track: number; /** * Update a TrackScan entity based on this. @@ -47,8 +47,7 @@ export abstract class UpdateTrackScan { public async update(scan: TrackScan): Promise { scan.valid = this.valid; scan.runner = await this.getRunner(); - scan.station = await this.getStation(); - scan.track = scan.station.track; + scan.track = await this.getTrack(); return scan; } @@ -67,11 +66,11 @@ export abstract class UpdateTrackScan { /** * Gets a runner based on the runner id provided via this.runner. */ - public async getStation(): Promise { - const station = await getConnection().getRepository(ScanStation).findOne({ id: this.station }, { relations: ['track'] }); - if (!station) { - throw new ScanStationNotFoundError(); + public async getTrack(): Promise { + const track = await getConnection().getRepository(Track).findOne({ id: this.track }); + if (!track) { + throw new TrackNotFoundError(); } - return station; + return track; } } \ No newline at end of file