Add oauth test case

Author: robooo

.github/workflows/main.yml

@@ -40,7 +40,7 @@ jobs:
       - name: Show python version
         run: python3 --version
       - name: Execute tests
-        run: python3 -m robot -d ./docs examples/
+        run: cd examples && python3 -m robot -d ../docs .
       - name: Archive test log
         if: ${{ always() }}
         uses: actions/upload-artifact@v4

examples/docker-compose.yml

@@ -1,4 +1,3 @@
-version: '3.8'
 services:
   broker:
     image: confluentinc/cp-server:7.8.0

examples/oauth2_test.py

@@ -0,0 +1,46 @@
+import time
+import functools
+import json
+from robot.libraries.BuiltIn import BuiltIn
+
+
+def create_test_token():
+    """Create a test OAuth2 token for unsecured testing"""
+    token_payload = {
+        "sub": "test-user",
+        "iss": "test-issuer",
+        "aud": "kafka",
+        "exp": int(time.time()) + 3600,
+        "iat": int(time.time()),
+        "scope": "kafka-producer kafka-consumer"
+    }
+
+    import base64
+    token_json = json.dumps(token_payload)
+    test_token = base64.b64encode(token_json.encode()).decode()
+
+    return test_token
+
+
+def oauth_cb_test_producer(oauth_config):
+    BuiltIn().set_global_variable("${SEEN_RF_OAUTH_CB_PRODUCER}", True)
+    test_token = create_test_token()
+    expiry_time = time.time() + 3600
+    BuiltIn().log(f"Generated test token: {test_token[:50]}...")
+    return test_token, expiry_time
+
+
+def oauth_cb_test_consumer(oauth_config):
+    BuiltIn().set_global_variable("${SEEN_RF_OAUTH_CB_CONSUMER}", True)
+    test_token = create_test_token()
+    expiry_time = time.time() + 3600
+
+    return test_token, expiry_time
+
+
+def get_test_producer_token(oauth_config=None):
+    return functools.partial(oauth_cb_test_producer, oauth_config or {})
+
+
+def get_test_consumer_token(oauth_config=None):
+    return functools.partial(oauth_cb_test_consumer, oauth_config or {})

examples/oauth_example.py

@@ -1,29 +1,32 @@
 *** Settings ***
 Library  ConfluentKafkaLibrary
-Library  oauth_example
+Library  oauth2_test
 Library  Collections
 Library  String
 
 
 *** Variables ***
-${SEEN_RF_OAUTH_CB}  ${False}
+${SEEN_RF_OAUTH_CB_PRODUCER}  ${False}
+${SEEN_RF_OAUTH_CB_CONSUMER}  ${False}
+${KAFKA_BOOTSTRAP_SERVERS}    localhost:9092
+${TEST_TOPIC}                 oauth2-test-topic
 
 
 *** Test Cases ***
-Example Oauth
-    [Documentation]  Example of how to use OAUTH with library and call functools
-    ...              via get_token function. For better handling there could be
-    ...              some global variable which can be set inside of python lib.
-    ...              Not executable right now, needs update env (issue #21).
+Test OAuth2 Token Generation
+    ${test_token}=  oauth2_test.create_test_token
+    Should Not Be Empty  ${test_token}
+    ${producer_token_func}=  oauth2_test.get_test_producer_token
+    ${consumer_token_func}=  oauth2_test.get_test_consumer_token
 
-    Skip
 
+Test OAuth2 Library Integration
     ${string_serializer}=  Get String Serializer
-    ${value_serializer}=  Get String Serializer
-
-    # This returns: functools.partial(<function oauth_cb at 0x7f...>, 'configuration')
-    ${fun}=  oauth_example.get_token  configuration
-
-    ${producer_id}=  Create Producer  key_serializer=${string_serializer}  value_serializer=${value_serializer}  legacy=${False}  security.protocol=sasl_plaintext  sasl.mechanisms=OAUTHBEARER  oauth_cb=${fun}
-
-    #...
+    ${oauth_func}=  oauth2_test.get_test_producer_token
+
+    ${status}  ${error}=  Run Keyword And Ignore Error
+    ...  Create Producer  localhost:9092
+    ...  oauth_cb=${oauth_func}
+    ...  security.protocol=sasl_plaintext
+    ...  sasl.mechanisms=OAUTHBEARER
+    Should Be Equal  ${status}  PASS