Added new fixure for single broker and using this fixure for test_serializer tests

Author: pranavrth

tests/common/__init__.py

@@ -25,7 +25,7 @@
 
 
 def _trivup_cluster_type_kraft():
-    return _TRIVUP_CLUSTER_TYPE_ENV in os.environ and os.environ[_TRIVUP_CLUSTER_TYPE_ENV] == 'kratf'
+    return _TRIVUP_CLUSTER_TYPE_ENV in os.environ and os.environ[_TRIVUP_CLUSTER_TYPE_ENV] == 'kraft'
 
 
 def use_group_protocol_consumer():

tests/integration/conftest.py

@@ -120,6 +120,13 @@ def kafka_cluster():
         yield fixture
 
 
+@pytest.fixture(scope="package")
+def kafka_single_broker_cluster():
+    for fixture in kafka_cluster_fixture(
+            trivup_cluster_conf={'broker_cnt': 1}):
+        yield fixture
+
+
 @pytest.fixture(scope="package")
 def sasl_cluster(request):
     for fixture in sasl_cluster_fixture(request.param):

tests/integration/serialization/test_serializers.py

@@ -31,12 +31,12 @@
                           (IntegerSerializer(), IntegerDeserializer(), 4124),
                           (DoubleSerializer(), DoubleDeserializer(), None),
                           (IntegerSerializer(), IntegerDeserializer(), None)])
-def test_numeric_serialization(kafka_cluster, serializer, deserializer, data):
+def test_numeric_serialization(kafka_single_broker_cluster, serializer, deserializer, data):
     """
     Tests basic serialization/deserialization of numeric types.
 
     Args:
-        kafka_cluster (KafkaClusterFixture): cluster fixture
+        kafka_single_broker_cluster (KafkaClusterFixture): cluster fixture
 
         serializer (Serializer): serializer to test
 
@@ -45,13 +45,13 @@ def test_numeric_serialization(kafka_cluster, serializer, deserializer, data):
         data(object): input data
 
     """
-    topic = kafka_cluster.create_topic("serialization-numeric")
+    topic = kafka_single_broker_cluster.create_topic("serialization-numeric")
 
-    producer = kafka_cluster.producer(value_serializer=serializer)
+    producer = kafka_single_broker_cluster.producer(value_serializer=serializer)
     producer.produce(topic, value=data)
     producer.flush()
 
-    consumer = kafka_cluster.consumer(value_deserializer=deserializer)
+    consumer = kafka_single_broker_cluster.consumer(value_deserializer=deserializer)
     consumer.subscribe([topic])
 
     msg = consumer.poll()
@@ -65,26 +65,26 @@ def test_numeric_serialization(kafka_cluster, serializer, deserializer, data):
                          [(u'Jämtland', 'utf_8'),
                           (u'Härjedalen', 'utf_16'),
                           (None, 'utf_32')])
-def test_string_serialization(kafka_cluster, data, codec):
+def test_string_serialization(kafka_single_broker_cluster, data, codec):
     """
     Tests basic unicode serialization/deserialization functionality
 
     Args:
-        kafka_cluster (KafkaClusterFixture): cluster fixture
+        kafka_single_broker_cluster (KafkaClusterFixture): cluster fixture
 
         data (unicode): input data
 
         codec (str): encoding type
 
     """
-    topic = kafka_cluster.create_topic("serialization-string")
+    topic = kafka_single_broker_cluster.create_topic("serialization-string")
 
-    producer = kafka_cluster.producer(value_serializer=StringSerializer(codec))
+    producer = kafka_single_broker_cluster.producer(value_serializer=StringSerializer(codec))
 
     producer.produce(topic, value=data)
     producer.flush()
 
-    consumer = kafka_cluster.consumer(value_deserializer=StringDeserializer(codec))
+    consumer = kafka_single_broker_cluster.consumer(value_deserializer=StringDeserializer(codec))
 
     consumer.subscribe([topic])
 
@@ -102,13 +102,13 @@ def test_string_serialization(kafka_cluster, data, codec):
                           (StringSerializer('utf_16'), DoubleSerializer(),
                            StringDeserializer('utf_16'), DoubleDeserializer(),
                            u'Härjedalen', 1.2168215450814477)])
-def test_mixed_serialization(kafka_cluster, key_serializer, value_serializer,
+def test_mixed_serialization(kafka_single_broker_cluster, key_serializer, value_serializer,
                              key_deserializer, value_deserializer, key, value):
     """
     Tests basic mixed serializer/deserializer functionality.
 
     Args:
-        kafka_cluster (KafkaClusterFixture): cluster fixture
+        kafka_single_broker_cluster (KafkaClusterFixture): cluster fixture
 
         key_serializer (Serializer): serializer to test
 
@@ -119,14 +119,14 @@ def test_mixed_serialization(kafka_cluster, key_serializer, value_serializer,
         value (object): value data
 
     """
-    topic = kafka_cluster.create_topic("serialization-numeric")
+    topic = kafka_single_broker_cluster.create_topic("serialization-numeric")
 
-    producer = kafka_cluster.producer(key_serializer=key_serializer,
+    producer = kafka_single_broker_cluster.producer(key_serializer=key_serializer,
                                       value_serializer=value_serializer)
     producer.produce(topic, key=key, value=value)
     producer.flush()
 
-    consumer = kafka_cluster.consumer(key_deserializer=key_deserializer,
+    consumer = kafka_single_broker_cluster.consumer(key_deserializer=key_deserializer,
                                       value_deserializer=value_deserializer)
     consumer.subscribe([topic])