stack_wallet/test/widget_tests/managed_favorite_test.dart

import 'package:decimal/decimal.dart';
import 'package:flutter/material.dart';
import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:flutter_test/flutter_test.dart';
import 'package:mockito/annotations.dart';
import 'package:mockito/mockito.dart';
import 'package:stackwallet/models/balance.dart';
import 'package:stackwallet/providers/providers.dart';
import 'package:stackwallet/services/coins/bitcoin/bitcoin_wallet.dart';
import 'package:stackwallet/services/coins/coin_service.dart';
import 'package:stackwallet/services/coins/manager.dart';
import 'package:stackwallet/services/locale_service.dart';
import 'package:stackwallet/services/node_service.dart';
import 'package:stackwallet/services/wallets.dart';
import 'package:stackwallet/services/wallets_service.dart';
import 'package:stackwallet/utilities/amount/amount.dart';
import 'package:stackwallet/utilities/enums/coin_enum.dart';
import 'package:stackwallet/utilities/listenable_list.dart';
import 'package:stackwallet/utilities/theme/light_colors.dart';
import 'package:stackwallet/utilities/theme/stack_colors.dart';
import 'package:stackwallet/widgets/managed_favorite.dart';

import 'managed_favorite_test.mocks.dart';

/// quick amount constructor wrapper. Using an int is bad practice but for
/// testing with small amounts this should be fine
Amount _a(int i) => Amount.fromDecimal(
      Decimal.fromInt(i),
      fractionDigits: 8,
    );

@GenerateMocks([
  Wallets,
  WalletsService,
  BitcoinWallet,
  LocaleService
], customMocks: [
  MockSpec<NodeService>(returnNullOnMissingStub: true),
  MockSpec<Manager>(returnNullOnMissingStub: true),
  MockSpec<CoinServiceAPI>(returnNullOnMissingStub: true),
])
void main() {
  testWidgets("Test wallet info row displays correctly", (widgetTester) async {
    final wallets = MockWallets();
    final CoinServiceAPI wallet = MockBitcoinWallet();

    when(wallet.coin).thenAnswer((_) => Coin.bitcoin);
    when(wallet.walletName).thenAnswer((_) => "some wallet");
    when(wallet.walletId).thenAnswer((_) => "some wallet id");

    final manager = Manager(wallet);
    when(wallets.getManager("some wallet id"))
        .thenAnswer((realInvocation) => manager);
    when(manager.balance).thenAnswer(
      (realInvocation) => Balance(
        coin: Coin.bitcoin,
        total: _a(10),
        spendable: _a(10),
        blockedTotal: _a(0),
        pendingSpendable: _a(0),
      ),
    );

    when(manager.isFavorite).thenAnswer((realInvocation) => false);
    final key = UniqueKey();
    // const managedFavorite = ManagedFavorite(walletId: "some wallet id", key: key,);
    await widgetTester.pumpWidget(
      ProviderScope(
        overrides: [
          walletsChangeNotifierProvider.overrideWithValue(wallets),
        ],
        child: MaterialApp(
          theme: ThemeData(
            extensions: [
              StackColors.fromStackColorTheme(
                LightColors(),
              ),
            ],
          ),
          home: Material(
            child: ManagedFavorite(
              walletId: "some wallet id",
              key: key,
            ),
          ),
        ),
      ),
    );

    expect(find.byType(ManagedFavorite), findsOneWidget);
  });

  testWidgets("Button Pressed - wallet unfavorite", (widgetTester) async {
    final wallets = MockWallets();
    final CoinServiceAPI wallet = MockBitcoinWallet();
    final mockLocaleService = MockLocaleService();
    final mockWalletsService = MockWalletsService();

    when(wallet.coin).thenAnswer((_) => Coin.bitcoin);
    when(wallet.walletName).thenAnswer((_) => "some wallet");
    when(wallet.walletId).thenAnswer((_) => "some wallet id");

    final manager = Manager(wallet);

    when(wallets.getManager("some wallet id"))
        .thenAnswer((realInvocation) => manager);
    when(manager.balance).thenAnswer(
      (realInvocation) => Balance(
        coin: Coin.bitcoin,
        total: _a(10),
        spendable: _a(10),
        blockedTotal: _a(0),
        pendingSpendable: _a(0),
      ),
    );

    when(manager.isFavorite).thenAnswer((realInvocation) => false);

    when(mockLocaleService.locale).thenAnswer((_) => "en_US");

    when(wallets.getManagerProvider("some wallet id")).thenAnswer(
        (realInvocation) => ChangeNotifierProvider((ref) => manager));

    const managedFavorite = ManagedFavorite(walletId: "some wallet id");

    final ListenableList<ChangeNotifierProvider<Manager>> favorites =
        ListenableList();

    final ListenableList<ChangeNotifierProvider<Manager>> nonfavorites =
        ListenableList();
    await widgetTester.pumpWidget(
      ProviderScope(
        overrides: [
          walletsChangeNotifierProvider.overrideWithValue(wallets),
          localeServiceChangeNotifierProvider
              .overrideWithValue(mockLocaleService),
          favoritesProvider.overrideWithValue(favorites),
          nonFavoritesProvider.overrideWithValue(nonfavorites),
          walletsServiceChangeNotifierProvider
              .overrideWithValue(mockWalletsService)
        ],
        child: MaterialApp(
          theme: ThemeData(
            extensions: [
              StackColors.fromStackColorTheme(
                LightColors(),
              ),
            ],
          ),
          home: const Material(
            child: managedFavorite,
          ),
        ),
      ),
    );

    expect(find.byType(RawMaterialButton), findsOneWidget);
    await widgetTester.tap(find.byType(RawMaterialButton));
    await widgetTester.pump();
  });

  testWidgets("Button Pressed - wallet is favorite", (widgetTester) async {
    final wallets = MockWallets();
    final CoinServiceAPI wallet = MockBitcoinWallet();
    final mockLocaleService = MockLocaleService();
    final mockWalletsService = MockWalletsService();

    when(wallet.coin).thenAnswer((_) => Coin.bitcoin);
    when(wallet.walletName).thenAnswer((_) => "some wallet");
    when(wallet.walletId).thenAnswer((_) => "some wallet id");

    final manager = Manager(wallet);

    when(wallets.getManager("some wallet id"))
        .thenAnswer((realInvocation) => manager);

    when(manager.isFavorite).thenAnswer((realInvocation) => true);
    when(manager.balance).thenAnswer(
      (realInvocation) => Balance(
        coin: Coin.bitcoin,
        total: _a(10),
        spendable: _a(10),
        blockedTotal: _a(0),
        pendingSpendable: _a(0),
      ),
    );

    when(mockLocaleService.locale).thenAnswer((_) => "en_US");

    when(wallets.getManagerProvider("some wallet id")).thenAnswer(
        (realInvocation) => ChangeNotifierProvider((ref) => manager));

    const managedFavorite = ManagedFavorite(walletId: "some wallet id");

    final ListenableList<ChangeNotifierProvider<Manager>> favorites =
        ListenableList();

    final ListenableList<ChangeNotifierProvider<Manager>> nonfavorites =
        ListenableList();
    await widgetTester.pumpWidget(
      ProviderScope(
        overrides: [
          walletsChangeNotifierProvider.overrideWithValue(wallets),
          localeServiceChangeNotifierProvider
              .overrideWithValue(mockLocaleService),
          favoritesProvider.overrideWithValue(favorites),
          nonFavoritesProvider.overrideWithValue(nonfavorites),
          walletsServiceChangeNotifierProvider
              .overrideWithValue(mockWalletsService)
        ],
        child: MaterialApp(
          theme: ThemeData(
            extensions: [
              StackColors.fromStackColorTheme(
                LightColors(),
              ),
            ],
          ),
          home: const Material(
            child: managedFavorite,
          ),
        ),
      ),
    );

    expect(find.byType(RawMaterialButton), findsOneWidget);
    await widgetTester.tap(find.byType(RawMaterialButton));
    await widgetTester.pump();
  });
}