summaryrefslogtreecommitdiffstats
path: root/robotframework-onap
diff options
context:
space:
mode:
authorefiacor <fiachra.corcoran@est.tech>2022-02-17 13:08:00 +0000
committerFiachra Corcoran <fiachra.corcoran@est.tech>2022-03-01 15:27:47 +0000
commit860728b66a792d0944b482b7394d6a04b1a74404 (patch)
treee03a19a91a39e777a283213b658df7d9c2b416c3 /robotframework-onap
parent949f9759c8d4b85790df5da717900737e9b29c88 (diff)
[KAFKA] Add support for scram-sha on kafka client
Signed-off-by: efiacor <fiachra.corcoran@est.tech> Change-Id: I77bbfe97e6cade41b91b5b2c5fd04fd0e5aa4e34 Issue-ID: DMAAP-1705
Diffstat (limited to 'robotframework-onap')
-rw-r--r--robotframework-onap/ONAPLibrary/KafkaKeywords.py17
1 files changed, 9 insertions, 8 deletions
diff --git a/robotframework-onap/ONAPLibrary/KafkaKeywords.py b/robotframework-onap/ONAPLibrary/KafkaKeywords.py
index ba49d68..46e0b3b 100644
--- a/robotframework-onap/ONAPLibrary/KafkaKeywords.py
+++ b/robotframework-onap/ONAPLibrary/KafkaKeywords.py
@@ -1,4 +1,5 @@
# Copyright 2019 AT&T Intellectual Property. All rights reserved.
+# Copyright (C) 2022 Nordix Foundation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
@@ -31,15 +32,15 @@ class KafkaKeywords(object):
self._cache = utils.ConnectionCache('No Kafka Environments created')
@keyword
- def connect(self, alias, kafka_host, sasl_user, sasl_password):
+ def connect(self, alias, kafka_host, sasl_user, sasl_password, sasl_mechanism="SCRAM-SHA-512"):
"""connect to the specified kafka server"""
client = {
"bootstrap_servers": kafka_host,
- "sasl_plain_username": sasl_user,
- "sasl_plain_password": sasl_password,
+ "sasl_username": sasl_user,
+ "sasl_password": sasl_password,
"security_protocol": 'SASL_PLAINTEXT',
"ssl_context": ssl.create_default_context(),
- "sasl_mechanism": 'PLAIN'
+ "sasl_mechanism": sasl_mechanism
}
self._cache.register(client, alias=alias)
@@ -54,8 +55,8 @@ class KafkaKeywords(object):
def _get_producer(self, alias):
cache = self._cache.switch(alias)
prod = KafkaProducer(bootstrap_servers=cache['bootstrap_servers'],
- sasl_plain_username=cache['sasl_plain_username'],
- sasl_plain_password=cache['sasl_plain_password'],
+ sasl_username=cache['sasl_username'],
+ sasl_password=cache['sasl_password'],
security_protocol=cache['security_protocol'],
ssl_context=cache['ssl_context'],
sasl_mechanism=cache['sasl_mechanism'],
@@ -86,8 +87,8 @@ class KafkaKeywords(object):
cache = self._cache.switch(alias)
consumer = KafkaConsumer(bootstrap_servers=cache['bootstrap_servers'],
- sasl_plain_username=cache['sasl_plain_username'],
- sasl_plain_password=cache['sasl_plain_password'],
+ sasl_username=cache['sasl_username'],
+ sasl_password=cache['sasl_password'],
security_protocol=cache['security_protocol'],
ssl_context=cache['ssl_context'],
sasl_mechanism=cache['sasl_mechanism'],