import {itBench} from "@dapplion/benchmark";
import {processInactivityUpdates} from "../../../src/epoch/processInactivityUpdates.js";
import {StateAltairEpoch} from "../types.js";
import {generatePerfTestCachedStateAltair, numValidators} from "../util.js";
import {mutateInactivityScores} from "./util.js";
import {FlagFactors, generateBalanceDeltasEpochProcess} from "./utilPhase0.js";

// PERF: Cost = iterate over an array of size $VALIDATOR_COUNT + 'proportional' to how many validtors are inactive or
// have been inactive in the past, i.e. that require an update to their inactivityScore. Worst case = all validators
// need to update their non-zero `inactivityScore`.
//
// On normal mainnet conditions
//   - prevTargetAttester: 96%
//   - unslashed:          100%
//   - eligibleAttester:   98%
//  TODO: Compute from altair testnet inactivityScores updates on average
//
// On worst case:
//   - status = 0xff
//   - all inactivityScores > 0

describe("altair processInactivityUpdates", () => {
  const vc = numValidators;

  const testCases: {id: string; isInInactivityLeak: boolean; flagFactors: FlagFactors; factorWithPositive: number}[] = [
    {
      id: "normalcase",
      isInInactivityLeak: false,
      flagFactors: 0xff,
      factorWithPositive: 0.04,
    },
    {
      id: "worstcase",
      isInInactivityLeak: true,
      flagFactors: 0xff,
      factorWithPositive: 0.04,
    },
  ];

  for (const {id, isInInactivityLeak, flagFactors, factorWithPositive} of testCases) {
    itBench<StateAltairEpoch, StateAltairEpoch>({
      id: `altair processInactivityUpdates - ${vc} ${id}`,
      yieldEventLoopAfterEach: true, // So SubTree(s)'s WeakRef can be garbage collected https://github.com/nodejs/node/issues/39902
      before: () => {
        const state = generatePerfTestCachedStateAltair({goBackOneSlot: true});
        const epochProcess = generateBalanceDeltasEpochProcess(state, isInInactivityLeak, flagFactors);
        mutateInactivityScores(state, factorWithPositive);
        return {state, epochProcess};
      },
      beforeEach: ({state, epochProcess}) => ({state: state.clone(), epochProcess}),
      fn: ({state, epochProcess}) => processInactivityUpdates(state, epochProcess),
    });
  }
});