pytest-kafka-broker 0.2.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

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.dist-info → pytest_kafka_broker-0.3.1.dist-info}/METADATA

@@ -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.3.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+pytest_kafka_broker/__init__.py,sha256=1-dEGoWlajumAl7AWsYARbxi9wKlBwTh56sg1KSLC5g,6511
+pytest_kafka_broker/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pytest_kafka_broker-0.3.1.dist-info/METADATA,sha256=_31JrO09YK79CS6ly65rCWZdQEuaFcF_dUBG7lKx-KE,683
+pytest_kafka_broker-0.3.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+pytest_kafka_broker-0.3.1.dist-info/entry_points.txt,sha256=mjFsRbu6FOiZkUOlSaXpo4QdLHwniNB-p_NEIgRbDnw,46
+pytest_kafka_broker-0.3.1.dist-info/top_level.txt,sha256=nTrYx9xVeK5hsqbhBRL2bgBV_ea-J66_f4Dk8eD-Ci0,20
+pytest_kafka_broker-0.3.1.dist-info/RECORD,,

pytest_kafka_broker-0.2.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-pytest_kafka_broker/__init__.py,sha256=H31mMYb-7j90GOaoaOmh12Fj50i7SRTEhX7l8MC6vbM,6148
-pytest_kafka_broker/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pytest_kafka_broker-0.2.0.dist-info/METADATA,sha256=XLPnJeSqqZXMYSuY5GPYZBFsSheAaVmfA1BXSYfZgtw,683
-pytest_kafka_broker-0.2.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-pytest_kafka_broker-0.2.0.dist-info/entry_points.txt,sha256=mjFsRbu6FOiZkUOlSaXpo4QdLHwniNB-p_NEIgRbDnw,46
-pytest_kafka_broker-0.2.0.dist-info/top_level.txt,sha256=nTrYx9xVeK5hsqbhBRL2bgBV_ea-J66_f4Dk8eD-Ci0,20
-pytest_kafka_broker-0.2.0.dist-info/RECORD,,