chore: remove mock state machine from test

aws-amplify · Aug 31, 2023 · eb246bf · eb246bf
1 parent 05c5ff9
commit eb246bf
Showing 1 changed file with 15 additions and 8 deletions.
diff --git a/packages/auth/amplify_auth_cognito_test/test/plugin/fetch_user_attributes_test.dart b/packages/auth/amplify_auth_cognito_test/test/plugin/fetch_user_attributes_test.dart
@@ -93,7 +93,11 @@ void main() {
   late AmplifyAuthCognitoDart plugin;
 
   group('fetchUserAttributes', () {
-    setUp(() {
+    tearDown(() async {
+      await plugin.close();
+    });
+
+    test('converts user attributes correctly', () async {
       stateMachine = MockCognitoAuthStateMachine()
         ..addInstance<CognitoIdentityProviderClient>(
           MockCognitoIdentityProviderClient(
@@ -107,13 +111,6 @@ void main() {
           ),
         );
       plugin = AmplifyAuthCognitoDart()..stateMachine = stateMachine;
-    });
-
-    tearDown(() async {
-      await plugin.close();
-    });
-
-    test('converts user attributes correctly', () async {
       final res = await plugin.fetchUserAttributes();
       final expected = [
         AuthUserAttribute(
@@ -202,6 +199,16 @@ void main() {
     });
 
     test('refreshes token before calling Cognito', () async {
+      stateMachine = CognitoAuthStateMachine()
+        ..addInstance<CognitoIdentityProviderClient>(
+          MockCognitoIdentityProviderClient(
+            getUser: () async => GetUserResponse(
+              userAttributes: [],
+              username: username,
+            ),
+          ),
+        );
+
       final secureStorage = MockSecureStorage();
       SecureStorageInterface storageFactory(scope) => secureStorage;
       seedStorage(