const test = require('node:test');
const assert = require('node:assert/strict');

const Valve = require('../../src/specificClass');
const supplierCurve = require('../../../generalFunctions/datasets/assetData/curves/ECDV.json');

function buildValve({ asset = {}, runtimeOptions = {} } = {}) {
  return new Valve(
    {
      general: {
        name: 'valve-test',
        logging: { enabled: false, logLevel: 'error' },
      },
      asset: {
        supplier: 'binder',
        category: 'valve',
        type: 'control',
        model: 'ECDV',
        unit: 'm3/h',
        ...asset,
      },
      functionality: {
        positionVsParent: 'atEquipment',
      },
    },
    {
      general: {
        logging: { enabled: false, logLevel: 'error' },
      },
      movement: { speed: 1 },
      time: { starting: 0, warmingup: 0, stopping: 0, coolingdown: 0 },
    },
    runtimeOptions
  );
}

test('valve selects supplier curve and predicts Kv from supplier data', () => {
  const valve = buildValve();

  valve.updatePressure('measured', 500, 'downstream', 'mbar');
  valve.updateFlow('predicted', 100, 'downstream', 'm3/h');
  valve.state.movementManager.currentPosition = 50;
  valve.updatePosition();

  const expectedKv = supplierCurve['1.204']['125'].y[5];
  assert.equal(valve.curveSelection.densityKey, 1.204);
  assert.equal(valve.curveSelection.diameterKey, 125);
  assert.ok(Math.abs(valve.kv - expectedKv) < 0.01, `expected Kv ${expectedKv}, got ${valve.kv}`);

  valve.destroy();
});

test('valve deltaP math uses converted flow units in formula path', () => {
  const valve = buildValve();

  valve.kv = 10;
  valve.rho = 1.204;
  valve.T = 293.15;
  valve.updatePressure('measured', 500, 'downstream', 'mbar');

  // 10 l/s equals 36 m3/h; formula path should use 36 m3/h.
  valve.updateFlow('predicted', 10, 'downstream', 'l/s');

  const downstreamAbsBar = (500 / 1000) + 1.01325;
  const qM3h = 36;
  const expectedDeltaPMbar = ((qM3h ** 2 * valve.rho * valve.T) / (514 ** 2 * valve.kv ** 2 * downstreamAbsBar)) * 1000;
  const actualDeltaP = valve.measurements
    .type('pressure')
    .variant('predicted')
    .position('delta')
    .getCurrentValue('mbar');

  assert.ok(Number.isFinite(actualDeltaP), 'deltaP should be finite');
  assert.ok(Math.abs(actualDeltaP - expectedDeltaPMbar) < 0.05, `expected ${expectedDeltaPMbar}, got ${actualDeltaP}`);

  valve.destroy();
});

test('valve liquid mode uses liquid Kv equation through update loop', () => {
  const valve = buildValve({ runtimeOptions: { serviceType: 'liquid', fluidDensity: 998 } });

  valve.kv = 50;
  valve.updatePressure('measured', 500, 'downstream', 'mbar');
  valve.updateFlow('predicted', 100, 'downstream', 'm3/h');

  const expectedDeltaPMbar = (((100 / 50) ** 2) * (998 / 1000)) * 1000;
  const actualDeltaP = valve.measurements
    .type('pressure')
    .variant('predicted')
    .position('delta')
    .getCurrentValue('mbar');

  assert.ok(Math.abs(actualDeltaP - expectedDeltaPMbar) < 0.01, `expected ${expectedDeltaPMbar}, got ${actualDeltaP}`);
  valve.destroy();
});

test('valve gas mode applies choked cap in update loop', () => {
  const valve = buildValve({ runtimeOptions: { serviceType: 'gas', gasChokedRatioLimit: 0.2 } });

  valve.kv = 1;
  valve.rho = 1.204;
  valve.T = 293.15;
  valve.updatePressure('measured', 500, 'downstream', 'mbar');
  valve.updateFlow('predicted', 1000, 'downstream', 'm3/h');

  const downstreamAbsBar = (500 / 1000) + 1.01325;
  const expectedDeltaPMbar = downstreamAbsBar * 0.2 * 1000;
  const actualDeltaP = valve.measurements
    .type('pressure')
    .variant('predicted')
    .position('delta')
    .getCurrentValue('mbar');

  assert.ok(Math.abs(actualDeltaP - expectedDeltaPMbar) < 0.0001, `expected ${expectedDeltaPMbar}, got ${actualDeltaP}`);
  assert.equal(valve.hydraulicDiagnostics?.isChoked, true);
  valve.destroy();
});