feat: added test_gssapi in test_security.py

airtai · Aug 1, 2024 · 7ede80b · 7ede80b
1 parent 609601f
commit 7ede80b
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 38 deletions.
diff --git a/tests/brokers/kafka/test_security.py b/tests/brokers/kafka/test_security.py
diff --git a/tests/docs/kafka/test_security.py b/tests/docs/kafka/test_security.py
@@ -98,3 +98,24 @@ async def test_plaintext():
             assert call_kwargs.items() <= producer_call_kwargs.items()
 
             assert type(producer_call_kwargs["ssl_context"]) is ssl.SSLContext
+
+
+@pytest.mark.kafka()
+@pytest.mark.asyncio()
+async def test_gssapi():
+    from docs.docs_src.kafka.security.sasl_gssapi import (
+        broker as gssapi_broker,
+    )
+
+    with patch_aio_consumer_and_producer() as producer:
+        async with gssapi_broker:
+            producer_call_kwargs = producer.call_args.kwargs
+
+            call_kwargs = {
+                "sasl_mechanism": "GSSAPI",
+                "security_protocol": "SASL_SSL",
+            }
+
+            assert call_kwargs.items() <= producer_call_kwargs.items()
+
+            assert type(producer_call_kwargs["ssl_context"]) is ssl.SSLContext