gridpilot.gg/core/racing/domain/entities/season/Season.test.ts

import { describe, it, expect } from 'vitest';

import {
  RacingDomainInvariantError,
  RacingDomainValidationError,
} from '@core/racing/domain/errors/RacingDomainError';

import { SeasonScoringConfig } from '@core/racing/domain/value-objects/SeasonScoringConfig';
import {
  SeasonDropPolicy,
} from '@core/racing/domain/value-objects/SeasonDropPolicy';
import { SeasonStewardingConfig } from '@core/racing/domain/value-objects/SeasonStewardingConfig';
import { Season, SeasonStatus } from '@core/racing/domain/entities/season/Season';

const createMinimalSeason = (overrides?: { status?: SeasonStatus }) =>
  Season.create({
    id: 'season-1',
    leagueId: 'league-1',
    gameId: 'iracing',
    name: 'Test Season',
    status: overrides?.status ?? 'planned',
  });

const createBaseSeason = () =>
  Season.create({
    id: 'season-1',
    leagueId: 'league-1',
    gameId: 'iracing',
    name: 'Config Season',
    status: 'planned',
    startDate: new Date('2025-01-01T00:00:00Z'),
    endDate: undefined,
    maxDrivers: 24,
  });

describe('Season aggregate lifecycle', () => {
  it('transitions Planned → Active → Completed → Archived with timestamps', () => {
    const planned = createMinimalSeason({ status: 'planned' });

    const activated = planned.activate();
    expect(activated.status.toString()).toBe('active');
    expect(activated.startDate).toBeInstanceOf(Date);
    expect(activated.endDate).toBeUndefined();

    const completed = activated.complete();
    expect(completed.status.toString()).toBe('completed');
    expect(completed.startDate).toEqual(activated.startDate);
    expect(completed.endDate).toBeInstanceOf(Date);

    const archived = completed.archive();
    expect(archived.status.toString()).toBe('archived');
    expect(archived.startDate).toEqual(completed.startDate);
    expect(archived.endDate).toEqual(completed.endDate);
  });

  it('throws when activating a non-planned season', () => {
    const active = createMinimalSeason({ status: 'active' });
    const completed = createMinimalSeason({ status: 'completed' });
    const archived = createMinimalSeason({ status: 'archived' });
    const cancelled = createMinimalSeason({ status: 'cancelled' });

    expect(() => active.activate()).toThrow(RacingDomainInvariantError);
    expect(() => completed.activate()).toThrow(RacingDomainInvariantError);
    expect(() => archived.activate()).toThrow(RacingDomainInvariantError);
    expect(() => cancelled.activate()).toThrow(RacingDomainInvariantError);
  });

  it('throws when completing a non-active season', () => {
    const planned = createMinimalSeason({ status: 'planned' });
    const completed = createMinimalSeason({ status: 'completed' });
    const archived = createMinimalSeason({ status: 'archived' });
    const cancelled = createMinimalSeason({ status: 'cancelled' });

    expect(() => planned.complete()).toThrow(RacingDomainInvariantError);
    expect(() => completed.complete()).toThrow(RacingDomainInvariantError);
    expect(() => archived.complete()).toThrow(RacingDomainInvariantError);
    expect(() => cancelled.complete()).toThrow(RacingDomainInvariantError);
  });

  it('throws when archiving a non-completed season', () => {
    const planned = createMinimalSeason({ status: 'planned' });
    const active = createMinimalSeason({ status: 'active' });
    const archived = createMinimalSeason({ status: 'archived' });
    const cancelled = createMinimalSeason({ status: 'cancelled' });

    expect(() => planned.archive()).toThrow(RacingDomainInvariantError);
    expect(() => active.archive()).toThrow(RacingDomainInvariantError);
    expect(() => archived.archive()).toThrow(RacingDomainInvariantError);
    expect(() => cancelled.archive()).toThrow(RacingDomainInvariantError);
  });

  it('allows cancelling planned or active seasons and rejects completed/archived', () => {
    const planned = createMinimalSeason({ status: 'planned' });
    const active = createMinimalSeason({ status: 'active' });
    const completed = createMinimalSeason({ status: 'completed' });
    const archived = createMinimalSeason({ status: 'archived' });

    const cancelledFromPlanned = planned.cancel();
    expect(cancelledFromPlanned.status.toString()).toBe('cancelled');
    expect(cancelledFromPlanned.startDate).toBe(planned.startDate);
    expect(cancelledFromPlanned.endDate).toBeInstanceOf(Date);

    const cancelledFromActive = active.cancel();
    expect(cancelledFromActive.status.toString()).toBe('cancelled');
    expect(cancelledFromActive.startDate).toBe(active.startDate);
    expect(cancelledFromActive.endDate).toBeInstanceOf(Date);

    expect(() => completed.cancel()).toThrow(RacingDomainInvariantError);
    expect(() => archived.cancel()).toThrow(RacingDomainInvariantError);
  });

  it('cancel is idempotent for already cancelled seasons', () => {
    const planned = createMinimalSeason({ status: 'planned' });
    const cancelled = planned.cancel();

    const cancelledAgain = cancelled.cancel();
    expect(cancelledAgain).toBe(cancelled);
  });

  it('canWithdrawFromWallet only when completed', () => {
    const planned = createMinimalSeason({ status: 'planned' });
    const active = createMinimalSeason({ status: 'active' });
    const completed = createMinimalSeason({ status: 'completed' });
    const archived = createMinimalSeason({ status: 'archived' });
    const cancelled = createMinimalSeason({ status: 'cancelled' });

    expect(planned.canWithdrawFromWallet()).toBe(false);
    expect(active.canWithdrawFromWallet()).toBe(false);
    expect(completed.canWithdrawFromWallet()).toBe(true);
    expect(archived.canWithdrawFromWallet()).toBe(false);
    expect(cancelled.canWithdrawFromWallet()).toBe(false);
  });
});

describe('Season configuration updates', () => {

  it('withScoringConfig returns a new Season with updated scoringConfig only', () => {
    const season = createBaseSeason();
    const scoringConfig = new SeasonScoringConfig({
      scoringPresetId: 'sprint-main-driver',
      customScoringEnabled: true,
    });

    const updated = season.withScoringConfig(scoringConfig);

    expect(updated).not.toBe(season);
    expect(updated.scoringConfig).toBe(scoringConfig);
    expect(updated.schedule).toBe(season.schedule);
    expect(updated.dropPolicy).toBe(season.dropPolicy);
    expect(updated.stewardingConfig).toBe(season.stewardingConfig);
    expect(updated.maxDrivers).toBe(season.maxDrivers);
  });

  it('withDropPolicy returns a new Season with updated dropPolicy only', () => {
    const season = createBaseSeason();
    const dropPolicy = new SeasonDropPolicy({
      strategy: 'bestNResults',
      n: 3,
    });

    const updated = season.withDropPolicy(dropPolicy);

    expect(updated).not.toBe(season);
    expect(updated.dropPolicy).toBe(dropPolicy);
    expect(updated.scoringConfig).toBe(season.scoringConfig);
    expect(updated.schedule).toBe(season.schedule);
    expect(updated.stewardingConfig).toBe(season.stewardingConfig);
    expect(updated.maxDrivers).toBe(season.maxDrivers);
  });

  it('withStewardingConfig returns a new Season with updated stewardingConfig only', () => {
    const season = createBaseSeason();
    const stewardingConfig = new SeasonStewardingConfig({
      decisionMode: 'steward_vote',
      requiredVotes: 3,
      requireDefense: true,
      defenseTimeLimit: 24,
      voteTimeLimit: 24,
      protestDeadlineHours: 48,
      stewardingClosesHours: 72,
      notifyAccusedOnProtest: true,
      notifyOnVoteRequired: true,
    });

    const updated = season.withStewardingConfig(stewardingConfig);

    expect(updated).not.toBe(season);
    expect(updated.stewardingConfig).toBe(stewardingConfig);
    expect(updated.scoringConfig).toBe(season.scoringConfig);
    expect(updated.schedule).toBe(season.schedule);
    expect(updated.dropPolicy).toBe(season.dropPolicy);
    expect(updated.maxDrivers).toBe(season.maxDrivers);
  });

  it('withMaxDrivers updates maxDrivers when positive', () => {
    const season = createBaseSeason();

    const updated = season.withMaxDrivers(30);

    expect(updated.maxDrivers).toBe(30);
    expect(updated.id).toBe(season.id);
    expect(updated.leagueId).toBe(season.leagueId);
    expect(updated.gameId).toBe(season.gameId);
  });

  it('withMaxDrivers allows undefined to clear value', () => {
    const season = createBaseSeason();

    const updated = season.withMaxDrivers(undefined);

    expect(updated.maxDrivers).toBeUndefined();
  });

  it('withMaxDrivers rejects non-positive values', () => {
    const season = createBaseSeason();

    expect(() => season.withMaxDrivers(0)).toThrow(
      RacingDomainValidationError,
    );
    expect(() => season.withMaxDrivers(-5)).toThrow(
      RacingDomainValidationError,
    );
  });
});