Back to Black

tests/test_cognitoidp/test_cognitoidp.py

@@ -1272,15 +1272,20 @@ def user_authentication_flow(conn):
     )["UserPoolClient"]["ClientId"]
 
     conn.sign_up(
-        ClientId=client_id, Username=username, Password=password,
+        ClientId=client_id,
+        Username=username,
+        Password=password,
     )
 
     client_secret = conn.describe_user_pool_client(
-        UserPoolId=user_pool_id, ClientId=client_id,
+        UserPoolId=user_pool_id,
+        ClientId=client_id,
     )["UserPoolClient"]["ClientSecret"]
 
     conn.confirm_sign_up(
-        ClientId=client_id, Username=username, ConfirmationCode="123456",
+        ClientId=client_id,
+        Username=username,
+        ConfirmationCode="123456",
     )
 
     # generating secret hash
@@ -1318,18 +1323,25 @@ def user_authentication_flow(conn):
     )
 
     conn.verify_software_token(
-        AccessToken=result["AuthenticationResult"]["AccessToken"], UserCode="123456",
+        AccessToken=result["AuthenticationResult"]["AccessToken"],
+        UserCode="123456",
     )
 
     conn.set_user_mfa_preference(
         AccessToken=result["AuthenticationResult"]["AccessToken"],
-        SoftwareTokenMfaSettings={"Enabled": True, "PreferredMfa": True,},
+        SoftwareTokenMfaSettings={
+            "Enabled": True,
+            "PreferredMfa": True,
+        },
     )
 
     result = conn.initiate_auth(
         ClientId=client_id,
         AuthFlow="REFRESH_TOKEN",
-        AuthParameters={"SECRET_HASH": secret_hash, "REFRESH_TOKEN": refresh_token,},
+        AuthParameters={
+            "SECRET_HASH": secret_hash,
+            "REFRESH_TOKEN": refresh_token,
+        },
     )
 
     result["AuthenticationResult"]["IdToken"].should_not.be.none
@@ -1583,7 +1595,8 @@ def test_sign_up():
     conn = boto3.client("cognito-idp", "us-west-2")
     user_pool_id = conn.create_user_pool(PoolName=str(uuid.uuid4()))["UserPool"]["Id"]
     client_id = conn.create_user_pool_client(
-        UserPoolId=user_pool_id, ClientName=str(uuid.uuid4()),
+        UserPoolId=user_pool_id,
+        ClientName=str(uuid.uuid4()),
     )["UserPoolClient"]["ClientId"]
     username = str(uuid.uuid4())
     password = str(uuid.uuid4())
@@ -1599,12 +1612,16 @@ def test_confirm_sign_up():
     password = str(uuid.uuid4())
     user_pool_id = conn.create_user_pool(PoolName=str(uuid.uuid4()))["UserPool"]["Id"]
     client_id = conn.create_user_pool_client(
-        UserPoolId=user_pool_id, ClientName=str(uuid.uuid4()), GenerateSecret=True,
+        UserPoolId=user_pool_id,
+        ClientName=str(uuid.uuid4()),
+        GenerateSecret=True,
     )["UserPoolClient"]["ClientId"]
     conn.sign_up(ClientId=client_id, Username=username, Password=password)
 
     conn.confirm_sign_up(
-        ClientId=client_id, Username=username, ConfirmationCode="123456",
+        ClientId=client_id,
+        Username=username,
+        ConfirmationCode="123456",
     )
 
     result = conn.admin_get_user(UserPoolId=user_pool_id, Username=username)
@@ -1618,14 +1635,19 @@ def test_initiate_auth_USER_SRP_AUTH():
     password = str(uuid.uuid4())
     user_pool_id = conn.create_user_pool(PoolName=str(uuid.uuid4()))["UserPool"]["Id"]
     client_id = conn.create_user_pool_client(
-        UserPoolId=user_pool_id, ClientName=str(uuid.uuid4()), GenerateSecret=True,
+        UserPoolId=user_pool_id,
+        ClientName=str(uuid.uuid4()),
+        GenerateSecret=True,
     )["UserPoolClient"]["ClientId"]
     conn.sign_up(ClientId=client_id, Username=username, Password=password)
     client_secret = conn.describe_user_pool_client(
-        UserPoolId=user_pool_id, ClientId=client_id,
+        UserPoolId=user_pool_id,
+        ClientId=client_id,
     )["UserPoolClient"]["ClientSecret"]
     conn.confirm_sign_up(
-        ClientId=client_id, Username=username, ConfirmationCode="123456",
+        ClientId=client_id,
+        Username=username,
+        ConfirmationCode="123456",
     )
 
     key = bytes(str(client_secret).encode("latin-1"))
@@ -1669,11 +1691,14 @@ def test_initiate_auth_for_unconfirmed_user():
     password = str(uuid.uuid4())
     user_pool_id = conn.create_user_pool(PoolName=str(uuid.uuid4()))["UserPool"]["Id"]
     client_id = conn.create_user_pool_client(
-        UserPoolId=user_pool_id, ClientName=str(uuid.uuid4()), GenerateSecret=True,
+        UserPoolId=user_pool_id,
+        ClientName=str(uuid.uuid4()),
+        GenerateSecret=True,
     )["UserPoolClient"]["ClientId"]
     conn.sign_up(ClientId=client_id, Username=username, Password=password)
     client_secret = conn.describe_user_pool_client(
-        UserPoolId=user_pool_id, ClientId=client_id,
+        UserPoolId=user_pool_id,
+        ClientId=client_id,
     )["UserPoolClient"]["ClientSecret"]
 
     key = bytes(str(client_secret).encode("latin-1"))
@@ -1705,14 +1730,19 @@ def test_initiate_auth_with_invalid_secret_hash():
     password = str(uuid.uuid4())
     user_pool_id = conn.create_user_pool(PoolName=str(uuid.uuid4()))["UserPool"]["Id"]
     client_id = conn.create_user_pool_client(
-        UserPoolId=user_pool_id, ClientName=str(uuid.uuid4()), GenerateSecret=True,
+        UserPoolId=user_pool_id,
+        ClientName=str(uuid.uuid4()),
+        GenerateSecret=True,
     )["UserPoolClient"]["ClientId"]
     conn.sign_up(ClientId=client_id, Username=username, Password=password)
     client_secret = conn.describe_user_pool_client(
-        UserPoolId=user_pool_id, ClientId=client_id,
+        UserPoolId=user_pool_id,
+        ClientId=client_id,
     )["UserPoolClient"]["ClientSecret"]
     conn.confirm_sign_up(
-        ClientId=client_id, Username=username, ConfirmationCode="123456",
+        ClientId=client_id,
+        Username=username,
+        ConfirmationCode="123456",
     )
 
     invalid_secret_hash = str(uuid.uuid4())