cake_wallet/integration_test/robots/pin_code_widget_robot.dart

import 'package:cake_wallet/src/screens/pin_code/pin_code_widget.dart';
import 'package:flutter_test/flutter_test.dart';

import '../components/common_test_cases.dart';

class PinCodeWidgetRobot {
  PinCodeWidgetRobot(this.tester) : commonTestCases = CommonTestCases(tester);

  final WidgetTester tester;
  late CommonTestCases commonTestCases;

  void hasPinCodeWidget() {
    final pinCodeWidget = find.bySubtype<PinCodeWidget>();
    expect(pinCodeWidget, findsOneWidget);
  }

  void hasNumberButtonsVisible() {
    // Confirmation for buttons 1-9
    for (var i = 1; i < 10; i++) {
      commonTestCases.hasValueKey('pin_code_button_${i}_key');
    }

    // Confirmation for 0 button
    commonTestCases.hasValueKey('pin_code_button_0_key');
  }

  Future<void> enterPinCode(List<int> pinCode, {int pumpDuration = 100}) async {
    for (int pin in pinCode) {
      await commonTestCases.tapItemByKey(
        'pin_code_button_${pin}_key',
        pumpDuration: pumpDuration,
      );
    }

    await commonTestCases.defaultSleepTime();
  }
}