// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

import 'package:amplify_auth_cognito_dart/amplify_auth_cognito_dart.dart';
import 'package:amplify_auth_cognito_dart/src/sdk/cognito_identity_provider.dart'
    hide SignUpRequest;
import 'package:amplify_auth_cognito_dart/src/sdk/sdk_bridge.dart';
import 'package:amplify_auth_cognito_dart/src/state/cognito_state_machine.dart';
import 'package:amplify_auth_cognito_test/common/mock_clients.dart';
import 'package:amplify_auth_cognito_test/common/mock_config.dart';
import 'package:amplify_auth_cognito_test/common/mock_secure_storage.dart';
import 'package:amplify_core/amplify_core.dart';
import 'package:amplify_secure_storage_dart/amplify_secure_storage_dart.dart';
import 'package:test/test.dart';

// Follows signUp test cases:
// https://github.com/aws-amplify/amplify-android/tree/main/aws-auth-cognito/src/test/resources/feature-test/testsuites/signUp
void main() {
  late AmplifyAuthCognitoDart plugin;
  late CognitoAuthStateMachine stateMachine;
  late MockSecureStorage secureStorage;

  final testAuthRepo = AmplifyAuthProviderRepository();

  group('AmplifyAuthCognitoDart', () {
    setUp(() async {
      secureStorage = MockSecureStorage();
      SecureStorageInterface storageFactory(scope) => secureStorage;
      stateMachine = CognitoAuthStateMachine();

      plugin = AmplifyAuthCognitoDart(secureStorageFactory: storageFactory)
        ..stateMachine = stateMachine;
    });

    tearDown(Amplify.reset);

    group('signUp', () {
      test('finishes when confirmed in the first step', () async {
        await plugin.configure(
          config: mockConfig,
          authProviderRepo: testAuthRepo,
        );

        final mockIdp = MockCognitoIdentityProviderClient(
          signUp: () async => SignUpResponse(
            userConfirmed: true,
            userSub: userSub,
          ),
        );
        stateMachine.addInstance<CognitoIdentityProviderClient>(mockIdp);

        await expectLater(
          plugin.signUp(
            username: username,
            password: 'password',
            options: SignUpOptions(
              userAttributes: {
                CognitoUserAttributeKey.email: 'user@domain.com',
              },
            ),
          ),
          completion(
            isA<CognitoSignUpResult>()
                .having(
                  (res) => res.isSignUpComplete,
                  'isSignUpComplete',
                  isTrue,
                )
                .having(
                  (res) => res.nextStep,
                  'nextStep',
                  const AuthNextSignUpStep(signUpStep: AuthSignUpStep.done),
                ),
          ),
        );
      });

      test('prompts confirmation', () async {
        await plugin.configure(
          config: mockConfig,
          authProviderRepo: testAuthRepo,
        );

        const destination = 'user@domain.com';
        const deliveryMedium = DeliveryMediumType.email;
        const attributeName = 'attributeName';
        final codeDeliveryDetails = CodeDeliveryDetailsType(
          destination: destination,
          deliveryMedium: deliveryMedium,
          attributeName: attributeName,
        );
        final mockIdp = MockCognitoIdentityProviderClient(
          signUp: () async => SignUpResponse(
            userConfirmed: false,
            userSub: userSub,
            codeDeliveryDetails: codeDeliveryDetails,
          ),
        );
        stateMachine.addInstance<CognitoIdentityProviderClient>(mockIdp);

        await expectLater(
          plugin.signUp(
            username: username,
            password: 'password',
            options: SignUpOptions(
              userAttributes: {
                CognitoUserAttributeKey.email: 'user@domain.com',
              },
            ),
          ),
          completion(
            isA<CognitoSignUpResult>()
                .having(
                  (res) => res.isSignUpComplete,
                  'isSignUpComplete',
                  isFalse,
                )
                .having(
                  (res) => res.nextStep,
                  'nextStep',
                  AuthNextSignUpStep(
                    signUpStep: AuthSignUpStep.confirmSignUp,
                    codeDeliveryDetails:
                        codeDeliveryDetails.asAuthCodeDeliveryDetails,
                  ),
                ),
          ),
        );
      });
    });
  });
}