predict-open-position.spec.ts

import { withFixtures } from '../../framework/fixtures/FixtureHelper';
import FixtureBuilder from '../../framework/fixtures/FixtureBuilder';
import { SmokePredictions } from '../../tags';
import { loginToApp } from '../../flows/wallet.flow';
import TabBarComponent from '../../page-objects/wallet/TabBarComponent';
import WalletActionsBottomSheet from '../../page-objects/wallet/WalletActionsBottomSheet';
import PredictMarketList from '../../page-objects/Predict/PredictMarketList';
import PredictDetailsPage from '../../page-objects/Predict/PredictDetailsPage';
import Assertions from '../../framework/Assertions';
import {
  remoteFeatureFlagHomepageSectionsV1Enabled,
  remoteFeatureFlagPredictEnabled,
} from '../../api-mocking/mock-responses/feature-flags-mocks';
import { Mockttp } from 'mockttp';
import { setupRemoteFeatureFlagsMock } from '../../api-mocking/helpers/remoteFeatureFlagsHelper';
import {
  POLYMARKET_COMPLETE_MOCKS,
  POLYMARKET_POSITIONS_WITH_WINNINGS_MOCKS,
  POLYMARKET_POST_OPEN_POSITION_MOCKS,
  POLYMARKET_UPDATE_USDC_BALANCE_MOCKS,
} from '../../api-mocking/mock-responses/polymarket/polymarket-mocks';
import WalletView from '../../page-objects/wallet/WalletView';
import { predictOpenPositionAnalyticsExpectations } from '../../helpers/analytics/expectations/predict-open-position.analytics';

/*
Test Scenario: Open position on Celtics vs. Nets market
  Verifies the open position flow for a predictions market:
  1. Navigate to Predictions tab and open market list
  2. Select Celtics vs. Nets market from sports category
  3. Open a position with $10 investment
  4. Verify position appears in Positions tab
  5. Verify balance updates to $17.76
  6. Verify position appears in Activities tab
*/
const positionDetails = {
  name: 'Celtics vs. Nets',
  positionAmount: '10',
  newBalance: '$17.76',
  category: 'sports' as const,
  marketIndex: 1,
};

const PredictionMarketFeature = async (mockServer: Mockttp) => {
  await setupRemoteFeatureFlagsMock(mockServer, {
    ...remoteFeatureFlagPredictEnabled(true),
    ...remoteFeatureFlagHomepageSectionsV1Enabled(),
    carouselBanners: false,
  });
  await POLYMARKET_COMPLETE_MOCKS(mockServer);
  await POLYMARKET_POSITIONS_WITH_WINNINGS_MOCKS(mockServer, false); // do not include winnings. Claim Button is animated and problematic for e2e
};

describe(SmokePredictions('Predictions'), () => {
  it('opens position on Celtics vs. Nets market', async () => {
    await withFixtures(
      {
        fixture: new FixtureBuilder()
          .withPolygon()
          .withMetaMetricsOptIn()
          .build(),
        restartDevice: true,
        testSpecificMock: PredictionMarketFeature,
        analyticsExpectations: predictOpenPositionAnalyticsExpectations,
      },
      async ({ mockServer }) => {
        await loginToApp();

        await TabBarComponent.tapActions();
        await WalletActionsBottomSheet.tapPredictButton();
        await device.disableSynchronization();

        await Assertions.expectElementToBeVisible(PredictMarketList.container, {
          description: 'Predict market list container should be visible',
        });

        await PredictMarketList.tapCategoryTab(positionDetails.category);
        await PredictMarketList.tapMarketCard(
          positionDetails.category,
          positionDetails.marketIndex,
        );
        await PredictDetailsPage.tapGameBetYesButton();

        await POLYMARKET_POST_OPEN_POSITION_MOCKS(mockServer);

        await PredictDetailsPage.tapPositionAmount(
          positionDetails.positionAmount,
        );
        await PredictDetailsPage.tapDoneButton();

        await PredictDetailsPage.tapOpenPosition();

        await PredictDetailsPage.tapBackButton();
        await Assertions.expectTextDisplayed(positionDetails.newBalance, {
          description: `USDC balance should display ${positionDetails.newBalance} after opening position`,
        });
        await PredictMarketList.tapBackButton();
        await device.enableSynchronization();

        await WalletView.scrollAndTapPredictionsPosition(positionDetails.name);

        /*
        When opening a position, the balance is optimistically updated in PredictController
        with a cache valid for 5 seconds. When getBalance() is called after cache expiration
        it invalidates the NetworkController's block cache and
        makes a fresh RPC balance request. The mock is placed here to
        verify that when the cache expires and a balance refresh request
        is made, it successfully returns the updated balance.
       */
        await POLYMARKET_UPDATE_USDC_BALANCE_MOCKS(mockServer, 'open-position');

        await PredictDetailsPage.tapBackButton();
        await TabBarComponent.tapActions();
        await WalletActionsBottomSheet.tapPredictButton();
        await Assertions.expectTextDisplayed(positionDetails.newBalance);
      },
    );
  });
});