013-stdcm-simulation-sheet.spec.ts

import fs from 'fs';

import pdfParse from 'pdf-parse';

import type { Infra } from 'common/api/osrdEditoastApi';

import { electricRollingStockName } from './assets/project-const';
import simulationSheetDetails from './assets/simulation-sheet-const';
import test from './logging-fixture';
import STDCMPage from './pages/stdcm-page-model';
import { waitForInfraStateToBeCached } from './utils';
import { getInfra } from './utils/api-setup';
import { findFirstPdf, verifySimulationContent } from './utils/simulationSheet';
import type { ConsistFields, Simulation } from './utils/types';

test.describe('Verify stdcm simulation page', () => {
  test.describe.configure({ mode: 'serial' }); // Configure this block to run serially
  test.slow(); // Mark test as slow due to multiple steps
  test.use({ viewport: { width: 1920, height: 1080 } });

  let stdcmPage: STDCMPage;
  let infra: Infra;

  const consistDetails: ConsistFields = {
    tractionEngine: electricRollingStockName,
    tonnage: '950',
    length: '567',
    maxSpeed: '180',
    speedLimitTag: 'HLP',
  };
  const tractionEnginePrefilledValues = {
    tonnage: '900',
    length: '400',
    maxSpeed: '288',
  };

  test.beforeAll('Fetch infrastructure', async () => {
    infra = await getInfra();
  });

  test.beforeEach('Navigate to the STDCM page', async ({ page }) => {
    stdcmPage = new STDCMPage(page);
    await page.goto('/stdcm');
    await page.waitForLoadState('networkidle');
    await stdcmPage.removeViteOverlay();

    // Wait for infra to be in 'CACHED' state before proceeding
    await waitForInfraStateToBeCached(infra.id);
  });

  let downloadDir: string | undefined;

  /** *************** Test 1 **************** */
  test('Verify STDCM stops and simulation sheet', async ({ browserName, context }, testInfo) => {
    // Populate STDCM page with origin, destination, and via details
    await stdcmPage.fillAndVerifyConsistDetails(
      consistDetails,
      tractionEnginePrefilledValues.tonnage,
      tractionEnginePrefilledValues.length,
      tractionEnginePrefilledValues.maxSpeed
    );
    await stdcmPage.fillOriginDetailsLight();
    await stdcmPage.fillDestinationDetailsLight();
    await stdcmPage.fillAndVerifyViaDetails({
      viaNumber: 1,
      ciSearchText: 'mid_west',
    });
    // Verify input map markers in Chromium
    if (browserName === 'chromium') {
      await stdcmPage.mapMarkerVisibility();
    }
    // Launch simulation and verify output data matches expected results
    await stdcmPage.launchSimulation();
    // Verify map results markers in Chromium
    if (browserName === 'chromium') {
      await stdcmPage.mapMarkerResultVisibility();
    }
    await stdcmPage.verifyTableData('./tests/assets/stdcm/stdcmWithoutAllVia.json');
    await stdcmPage.displayAllOperationalPoints();
    await stdcmPage.verifyTableData('./tests/assets/stdcm/stdcmWithAllVia.json');
    await stdcmPage.retainSimulation();
    downloadDir = testInfo.outputDir;
    await stdcmPage.downloadSimulation(downloadDir);
    // Reset and verify empty fields
    const [newPage] = await Promise.all([context.waitForEvent('page'), stdcmPage.startNewQuery()]);
    await newPage.waitForLoadState();
    const newStdcmPage = new STDCMPage(newPage);
    await newStdcmPage.verifyAllDefaultPageFields();
  });

  /** *************** Test 2 **************** */
  test('Verify simulation sheet content', async () => {
    const pdfFilePath = findFirstPdf(downloadDir!);

    if (!pdfFilePath) {
      throw new Error(`No PDF files found in directory: ${downloadDir}`);
    }
    // Read and parse the PDF
    const pdfBuffer = fs.readFileSync(pdfFilePath);
    const pdfData = await pdfParse(pdfBuffer);
    const pdfText = pdfData.text;
    const expectedSimulation: Simulation = simulationSheetDetails();
    verifySimulationContent(pdfText, expectedSimulation);
  });
});