squid: rgw-testing: fix unexpected keyword argument 'expected_sizes' for kafka test

Signed-off-by: Hoai-Thu Vuong <thuvh87@gmail.com>

diff --git a/src/test/rgw/bucket_notification/test_bn.py b/src/test/rgw/bucket_notification/test_bn.py
index b9c7c5d6f35b8f02ad0ab092192890b2594b6eac..37f5b4b6cbaa037078f590690bf4593b73394beb 100644 (file)
--- a/src/test/rgw/bucket_notification/test_bn.py
+++ b/src/test/rgw/bucket_notification/test_bn.py
@@ -445,11 +445,15 @@ class KafkaReceiver(object):
         self.topic = topic
         self.stop = False
 
-    def verify_s3_events(self, keys, exact_match=False, deletions=False, etags=[]):
+    def verify_s3_events(self, keys, exact_match=False, deletions=False, etags=[], expected_sizes={}):
         """verify stored s3 records agains a list of keys"""
-        verify_s3_records_by_elements(self.events, keys, exact_match=exact_match, deletions=deletions, etags=etags)
+        verify_s3_records_by_elements(self.events, keys, exact_match=exact_match, deletions=deletions, etags=etags, expected_sizes=expected_sizes)
         self.events = []
 
+    def close(self, task):
+        stop_kafka_receiver(self, task)
+
+
 def kafka_receiver_thread_runner(receiver):
     """main thread function for the kafka receiver"""
     try:
@@ -1432,7 +1436,7 @@ def notification_push(endpoint_type, conn, account=None, cloudevents=False, kafk
         # start amqp receiver
         task, receiver = create_kafka_receiver_thread(topic_name, kafka_brokers=kafka_brokers)
         task.start()
-        endpoint_address = 'kafka://' + kafka_server
+        endpoint_address = 'kafka://' + kafka_server + ':9092'
         # without acks from broker
         endpoint_args = 'push-endpoint='+endpoint_address+'&kafka-ack-level=broker'
         if kafka_brokers is not None: