onchain.spec.ts

import { Effect } from 'effect';
import {
  afterEach,
  beforeEach,
  describe,
  expect,
  it,
  vi,
} from '@effect/vitest';

import { AggregatedDataFeedStoreConsumer } from '@blocksense/contracts/viem';

import type { FeedsValueAndRound } from './onchain';
import { getDataFeedsInfoFromNetwork } from './onchain';

describe('getDataFeedsInfoFromNetwork', () => {
  const mockConsumer = {
    getLatestSingleDataAndIndex: vi.fn(),
    getSingleDataAtIndex: vi.fn(),
  };

  const data =
    '0x0000000000000000000000000000000000000a3549cbad30000001986c265bf0';

  beforeEach(() => {
    vi.spyOn(AggregatedDataFeedStoreConsumer, 'create').mockReturnValue(
      mockConsumer as any,
    );
  });

  afterEach(() => {
    vi.clearAllMocks();
  });

  it.effect('fetches latest data when roundsInfo is not provided', () =>
    Effect.gen(function* () {
      mockConsumer.getLatestSingleDataAndIndex.mockResolvedValue({
        data,
        index: 42n,
      });

      const feedId = 1n;

      const result: FeedsValueAndRound = yield* getDataFeedsInfoFromNetwork(
        [feedId],
        '0xabc123',
        'ethereum-mainnet',
      );

      const expected: FeedsValueAndRound = {
        '1': { value: data, round: 42 },
      };

      expect(result).toEqual(expected);
      expect(mockConsumer.getLatestSingleDataAndIndex).toHaveBeenCalledWith(
        feedId,
      );
    }),
  );

  it.effect('fetches specific round data when roundsInfo is provided', () =>
    Effect.gen(function* () {
      mockConsumer.getSingleDataAtIndex.mockResolvedValue(data);

      const feedId = 2n;
      const round = 99;

      const result: FeedsValueAndRound = yield* getDataFeedsInfoFromNetwork(
        [feedId],
        '0xabc123',
        'ethereum-mainnet',
        { '2': round },
      );

      const expected: FeedsValueAndRound = {
        '2': { value: data, round: 99 },
      };

      expect(result).toEqual(expected);
      expect(mockConsumer.getSingleDataAtIndex).toHaveBeenCalledWith(
        feedId,
        round,
      );
    }),
  );
});