hash-wallet/integration_test/robots/wallet_keys_robot.dart

import "package:hash_wallet/reactions/wallet_connect.dart";
import "package:hash_wallet/src/screens/wallet_keys/wallet_keys_page.dart";
import "package:hash_wallet/store/app_store.dart";
import "package:cw_core/monero_wallet_keys.dart";
import "package:cw_core/wallet_type.dart";
import "package:cw_monero/monero_wallet.dart";
import "package:flutter_test/flutter_test.dart";
import "package:polyseed/polyseed.dart";

import "../core/base_robot.dart";

class WalletKeysAndSeedPageRobot extends BaseRobot {
  WalletKeysAndSeedPageRobot(super.tester);

  @override
  Future<void> isDisplayed() async {
    await isSpecificPage<WalletKeysPage>();
  }

  int _verifiedCredentials = 0;

  Future<void> confirmWalletCredentials(WalletType walletType) async {
    _verifiedCredentials = 0;

    final walletKeysPage = tester.widget<WalletKeysPage>(find.byType(WalletKeysPage));
    final walletKeysViewModel = walletKeysPage.walletKeysViewModel;

    final appStore = walletKeysViewModel.appStore;
    final walletName = walletType.name;
    final hasSeed = appStore.wallet!.seed != null;
    final hasHexSeed = appStore.wallet!.hexSeed != null;
    final hasPrivateKey = appStore.wallet!.privateKey != null;

    if (walletType == WalletType.monero) {
      final moneroWallet = appStore.wallet! as MoneroWalletBase;
      final lang = PolyseedLang.getByPhrase(moneroWallet.seed);
      final legacySeed = moneroWallet.seedLegacy(lang.nameEnglish);

      await _confirmMoneroWalletCredentials(
        appStore,
        walletName,
        moneroWallet.seed,
        legacySeed,
      );
    }

    if (walletType == WalletType.bitcoin ||
        walletType == WalletType.litecoin ||
        walletType == WalletType.bitcoinCash) {
      _showsSeedWords(appStore.wallet!.seed!);
      _verifiedCredentials++;
      tester.printToConsole("$walletName wallet has seeds properly displayed");
    }

    // Hash Bags does not ship Solana or Tron; those WalletType members were
    // removed with the chains, so only the EVM chains remain here.
    if (isEVMCompatibleChain(walletType)) {
      if (hasSeed) {
        _showsSeedWords(appStore.wallet!.seed!);
        _verifiedCredentials++;
        tester.printToConsole("$walletName wallet has seeds properly displayed");
      }
      if (hasPrivateKey) {
        await _openKeysTab();
        _showsCredentialOnce(appStore.wallet!.privateKey!, "private key");
        _verifiedCredentials++;
        tester.printToConsole("$walletName wallet has private key properly displayed");
      }
    }

    if (walletType == WalletType.nano || walletType == WalletType.banano) {
      if (hasSeed) {
        _showsSeedWords(appStore.wallet!.seed!);
        _verifiedCredentials++;
        tester.printToConsole("$walletName wallet has seeds properly displayed");
      }
      if (hasHexSeed) {
        await _openKeysTab();
        _showsCredentialOnce(appStore.wallet!.hexSeed!, "hex seed");
        _verifiedCredentials++;
        tester.printToConsole("$walletName wallet has hexSeed properly displayed");
      }
      if (hasPrivateKey) {
        await _openKeysTab();
        _showsCredentialOnce(appStore.wallet!.privateKey!, "private key");
        _verifiedCredentials++;
        tester.printToConsole("$walletName wallet has private key properly displayed");
      }
    }

    _expectSomethingWasVerified(walletType);
  }

  void _expectSomethingWasVerified(WalletType walletType) {
    expect(
      _verifiedCredentials,
      greaterThan(0),
      reason: "Nothing was verified for ${walletType.name}, this suite covers no credential "
          "for that type so it cannot tell a working keys page from a broken one",
    );
  }

  Future<void> _openKeysTab() async {
    await tapByKey("wallet_keys_page_keys");

    await settle();
  }

  void _showsCredential(String value, String label) {
    expect(
      tester.any(find.text(value, skipOffstage: false)),
      true,
      reason: "The keys page did not show the wallet's $label",
    );
  }

  void _showsCredentialOnce(String value, String label) {
    final rows = tester.widgetList(find.text(value, skipOffstage: false)).length;

    expect(
      rows,
      1,
      reason: "The keys page showed $rows rows for the wallet's $label, expected one",
    );
  }

  void _showsSeedWords(String seed) {
    final words = seed.split(" ");

    for (var index = 0; index < words.length; index++) {
      _showsCredential(words[index], "seed word ${index + 1}");
    }
  }

  Future<void> _confirmMoneroWalletCredentials(
    AppStore appStore,
    String walletName,
    String seed,
    String legacySeed,
  ) async {
    final keys = appStore.wallet!.keys as MoneroWalletKeys;
    final hasSeeds = seed.isNotEmpty;
    final hasSeedLegacy = Polyseed.isValidSeed(seed);

    await _openKeysTab();

    _showsCredentialOnce(keys.publicSpendKey, "public spend key");
    _verifiedCredentials++;
    tester.printToConsole("$walletName wallet has public spend key properly displayed");

    _showsCredentialOnce(keys.privateSpendKey, "private spend key");
    _verifiedCredentials++;
    tester.printToConsole("$walletName wallet has private spend key properly displayed");

    _showsCredentialOnce(keys.publicViewKey, "public view key");
    _verifiedCredentials++;
    tester.printToConsole("$walletName wallet has public view key properly displayed");

    _showsCredentialOnce(keys.privateViewKey, "private view key");
    _verifiedCredentials++;
    tester.printToConsole("$walletName wallet has private view key properly displayed");

    if (hasSeeds) {
      await tapByKey("wallet_keys_page_seed");
      await settle();

      _showsSeedWords(seed);
      _verifiedCredentials++;
      tester.printToConsole("$walletName wallet has seeds properly displayed");
    }
    if (hasSeedLegacy) {
      await tapByKey("wallet_keys_page_seed_legacy");
      await settle();

      _showsSeedWords(legacySeed);
      _verifiedCredentials++;
      tester.printToConsole("$walletName wallet has legacy seeds properly displayed");
    }
  }
}