From 4d4f675cea7abe32ed1505150a35dcf209bb6f4a Mon Sep 17 00:00:00 2001
From: Erik Forsberg <erik.forsberg@ferroamp.se>
Date: Fri, 13 Nov 2020 11:27:35 +0100
Subject: [PATCH] Make SCRAM-SHA-256/512 SASL Mechanisms available.

Fixes robinhood/faust#601.
---
 faust/types/auth.py     | 2 ++
 tests/unit/test_auth.py | 9 +++++++++
 2 files changed, 11 insertions(+)

diff --git a/faust/types/auth.py b/faust/types/auth.py
index 8006dd8a2..4a78f5bdd 100644
--- a/faust/types/auth.py
+++ b/faust/types/auth.py
@@ -23,6 +23,8 @@ class AuthProtocol(Enum):
 class SASLMechanism(Enum):
     PLAIN = "PLAIN"
     GSSAPI = "GSSAPI"
+    SCRAM_SHA_256 = "SCRAM-SHA-256"
+    SCRAM_SHA_512 = "SCRAM-SHA-512"
 
 
 AUTH_PROTOCOLS_SSL = {AuthProtocol.SSL, AuthProtocol.SASL_SSL}
diff --git a/tests/unit/test_auth.py b/tests/unit/test_auth.py
index cbcb195a3..616bef779 100644
--- a/tests/unit/test_auth.py
+++ b/tests/unit/test_auth.py
@@ -45,6 +45,15 @@ class test_SASLCredentials:
                     "protocol": AuthProtocol.SASL_SSL,
                 },
             ),
+            pytest.param(
+                "SCRAM-SHA-512 supported",
+                SASLCredentials(
+                    username="george",
+                    password="pw1",
+                    mechanism=SASLMechanism.SCRAM_SHA_512,
+                ),
+                {"mechanism": SASLMechanism.SCRAM_SHA_512},
+            ),
         ],
     )
     def test_constructor(self, credentials, expected_fields, reason):