Add test (fails w/o changes)

Author: dpkp

test/sasl/test_gssapi.py

@@ -0,0 +1,41 @@
+from __future__ import absolute_import
+
+try:
+    from unittest import mock
+except ImportError:
+    import mock
+
+from kafka.sasl import get_sasl_mechanism
+from kafka.sasl.gssapi import SaslMechanismGSSAPI
+
+
+def test_gssapi():
+    config = {
+        'sasl_kerberos_domain_name': 'foo',
+        'sasl_kerberos_service_name': 'bar',
+    }
+    client_ctx = mock.Mock()
+    client_ctx.step.side_effect = [b'init', b'exchange', b'complete', b'final', b'xxxx']
+    client_ctx.complete = False
+    def mocked_message_wrapper(msg, *args):
+        wrapped = mock.Mock()
+        type(wrapped).message = mock.PropertyMock(return_value=msg)
+        return wrapped
+    client_ctx.unwrap.side_effect = mocked_message_wrapper
+    client_ctx.wrap.side_effect = mocked_message_wrapper
+    with mock.patch('kafka.sasl.gssapi.gssapi.SecurityContext', return_value=client_ctx) as patched:
+        gssapi = get_sasl_mechanism('GSSAPI')(**config)
+        assert isinstance(gssapi, SaslMechanismGSSAPI)
+        client_ctx.step.assert_called_with(None)
+
+        while not gssapi.is_done():
+            send_token = gssapi.auth_bytes()
+            receive_token = send_token # not realistic, but enough for testing
+            if send_token == b'\x00cbar@foo': # final wrapped message
+                receive_token = b'' # final message gets an empty response
+            gssapi.receive(receive_token)
+            if send_token == b'exchange':
+                client_ctx.complete = True
+
+        assert gssapi.is_done()
+        assert gssapi.is_authenticated()