pytest-kafka-broker 0.2.0__tar.gz → 0.3.1__tar.gz

{pytest_kafka_broker-0.2.0 → pytest_kafka_broker-0.3.1}/.gitignore

@@ -1,5 +1,6 @@
 __pycache__
 _build
+.coverage
 *.egg-info
 api
 build

{pytest_kafka_broker-0.2.0 → pytest_kafka_broker-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-kafka-broker
-Version: 0.2.0
+Version: 0.3.1
 Summary: Pytest plugin to run a single-broker Kafka cluster
 Author-email: Leo Singer <leo.singer@ligo.org>
 License-Expression: Apache-2.0

{pytest_kafka_broker-0.2.0 → pytest_kafka_broker-0.3.1}/src/pytest_kafka_broker/__init__.py

@@ -12,6 +12,7 @@ import pytest_asyncio
 from astropy.config import get_cache_dir_path  # type: ignore[import-untyped]
 from astropy.utils.data import get_readable_fileobj  # type: ignore[import-untyped]
 from confluent_kafka import Consumer, Producer
+from confluent_kafka.admin import AdminClient
 from confluent_kafka.aio import AIOConsumer, AIOProducer
 from rich.status import Status
 
@@ -87,6 +88,9 @@ class KafkaBrokerContext:
     def config(self, config: dict | None = None) -> dict:
         return {**(config or {}), "bootstrap.servers": self.bootstrap_server}
 
+    def admin(self, config: dict | None = None) -> AdminClient:
+        return AdminClient(self.config(config))
+
     def producer(self, config: dict | None = None) -> Producer:
         return Producer(self.config(config))
 
@@ -100,6 +104,7 @@ class KafkaBrokerContext:
         return AIOConsumer(self.config(config))
 
     config.__doc__ = _doc.format("Get the configuration for a Kafka client.")
+    admin.__doc__ = _doc.format("Create a Kafka admin client connected to the cluster.")
     producer.__doc__ = _doc.format("Create a Kafka producer connected to the cluster.")
     consumer.__doc__ = _doc.format("Create a Kafka consumer connected to the cluster.")
     aio_producer.__doc__ = _doc.format(
@@ -181,5 +186,8 @@ async def kafka_broker(
         yield KafkaBrokerContext(f"127.0.0.1:{plaintext_port}")
     finally:
         with Status("Stopping Kafka broker"):
-            process.terminate()
+            try:
+                process.terminate()
+            except ProcessLookupError:
+                pass  # Process has already terminated
             await exited

{pytest_kafka_broker-0.2.0 → pytest_kafka_broker-0.3.1}/src/pytest_kafka_broker.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-kafka-broker
-Version: 0.2.0
+Version: 0.3.1
 Summary: Pytest plugin to run a single-broker Kafka cluster
 Author-email: Leo Singer <leo.singer@ligo.org>
 License-Expression: Apache-2.0

{pytest_kafka_broker-0.2.0 → pytest_kafka_broker-0.3.1}/tests/test_kafka.py

@@ -1,20 +1,24 @@
 import pytest
 
-topic = "topic"
-payload = b"hello world"
-group_id = "group_id"
+TOPIC = "topic"
+PAYLOAD = b"hello world"
+GROUP_ID = "group_id"
 
 
 def test_sync(kafka_broker):
     """Demonstrate using the kafka_broker fixture in an ordinary test."""
     with kafka_broker.producer() as producer:
-        producer.produce(topic, payload)
+        producer.produce(TOPIC, PAYLOAD)
+
     with kafka_broker.consumer(
-        {"group.id": group_id, "auto.offset.reset": "earliest"}
+        {"group.id": GROUP_ID, "auto.offset.reset": "earliest"}
     ) as consumer:
-        consumer.subscribe([topic])
+        consumer.subscribe([TOPIC])
         (message,) = consumer.consume()
-        assert message.value() == payload
+        assert message.value() == PAYLOAD
+
+    with kafka_broker.admin() as admin:
+        assert TOPIC in admin.list_topics().topics
 
 
 @pytest.mark.asyncio
@@ -22,17 +26,21 @@ async def test_async(kafka_broker):
     """Demonstrate using the kafka_broker fixture in an async test."""
     producer = kafka_broker.aio_producer()
     try:
-        await producer.produce(topic, payload)
+        await producer.produce(TOPIC, PAYLOAD)
     finally:
         # FIXME: use async context manager; see https://github.com/confluentinc/confluent-kafka-python/pull/2180
         await producer.close()
     consumer = kafka_broker.aio_consumer(
-        {"group.id": group_id, "auto.offset.reset": "earliest"}
+        {"group.id": GROUP_ID, "auto.offset.reset": "earliest"}
     )
+
     try:
-        await consumer.subscribe([topic])
+        await consumer.subscribe([TOPIC])
         (message,) = await consumer.consume()
     finally:
         # FIXME: use async context manager; see https://github.com/confluentinc/confluent-kafka-python/pull/2180
         await consumer.close()
-    assert message.value() == payload
+    assert message.value() == PAYLOAD
+
+    with kafka_broker.admin() as admin:
+        assert TOPIC in admin.list_topics().topics