Merge pull request #28 from stax-labs/feat/auth_errors

feat (auth) fix error messages

Author: ZXYmania

staxapp/auth.py

@@ -62,11 +62,11 @@ def id_token_from_cognito(self, username=None, password=None, client=None):
             try:
                 tokens = aws.authenticate_user()
             except ClientError as e:
-                if e.response["Error"]["Code"] == "UserNotFoundException":
+                if e.response["Error"]["Code"] == "NotAuthorizedException":
                     raise InvalidCredentialsException(
                         message=str(e), detail="Please check your Secret Key is correct"
                     )
-                elif e.response["Error"]["Code"] == "NotAuthorizedException":
+                elif e.response["Error"]["Code"] == "UserNotFoundException":
                     raise InvalidCredentialsException(
                         message=str(e),
                         detail="Please check your Access Key, that you have created your Api Token and that you are using the right STAX REGION",

tests/test_auth.py

@@ -67,7 +67,7 @@ def testCredentialErrors(self):
 
         sa = StaxAuth("ApiAuth")
         # Test with invalid username password
-        self.stub_aws_srp(sa, "bad_password", "UserNotFoundException")
+        self.stub_aws_srp(sa, "bad_password", "NotAuthorizedException")
         user_not_found_success = False
         try:
             sa.id_token_from_cognito(
@@ -79,7 +79,7 @@ def testCredentialErrors(self):
         self.assertTrue(user_not_found_success)
 
         # Test with no access
-        self.stub_aws_srp(sa, "no_access", "NotAuthorizedException")
+        self.stub_aws_srp(sa, "no_access", "UserNotFoundException")
         no_access_success = False
         try:
             sa.id_token_from_cognito(