pytest-kafka-broker 0.1.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

pytest_kafka_broker.py → pytest_kafka_broker/__init__.py

@@ -1,5 +1,6 @@
 import asyncio
 import subprocess
+from collections.abc import AsyncGenerator
 from dataclasses import dataclass
 from pathlib import Path
 from tarfile import TarFile
@@ -8,9 +9,10 @@ from uuid import uuid4
 
 import pytest
 import pytest_asyncio
-from astropy.config import get_cache_dir_path
-from astropy.utils.data import get_readable_fileobj
+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
 
@@ -24,7 +26,7 @@ SCALA_VERSION = "2.13"
 KAFKA_VERSION = "4.1.1"
 
 
-async def wait_port(port: int, timeout: float = 0.25):
+async def wait_port(port: int, timeout: float = 0.25) -> None:
     """Wait until a connection is detected listening on the given port."""
     while True:
         try:
@@ -83,9 +85,12 @@ class KafkaBrokerContext:
     bootstrap_server: str
     """Kafka bootstrap server in the form :samp:`{host}:{port}`."""
 
-    def config(self, config: dict | None = None):
+    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))
 
@@ -99,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(
@@ -113,13 +119,10 @@ del _doc
 
 
 @pytest_asyncio.fixture
-async def kafka_broker(kafka_home, tmp_path, unused_tcp_port_factory):
-    """Pytest fixture to run a local, temporary Kafka broker.
-
-    Returns
-    -------
-    : KafkaBrokerContext
-    """
+async def kafka_broker(
+    kafka_home, tmp_path, unused_tcp_port_factory
+) -> AsyncGenerator[KafkaBrokerContext]:
+    """Pytest fixture to run a local, temporary Kafka broker."""
     kafka_storage = kafka_home / "bin" / "kafka-storage.sh"
     kafka_server_start = kafka_home / "bin" / "kafka-server-start.sh"
     config_path = tmp_path / "server.properties"

{pytest_kafka_broker-0.1.1.dist-info → pytest_kafka_broker-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-kafka-broker
-Version: 0.1.1
+Version: 0.3.0
 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.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+pytest_kafka_broker/__init__.py,sha256=3mODu1laOwp9moAXLnrMPZo6Zb7CWe-L7g7pVUVvQPU,6396
+pytest_kafka_broker/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pytest_kafka_broker-0.3.0.dist-info/METADATA,sha256=KI9mVBcRN688rjCpdLfNx99-LhOpq4pj_D4u7WTlwYI,683
+pytest_kafka_broker-0.3.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+pytest_kafka_broker-0.3.0.dist-info/entry_points.txt,sha256=mjFsRbu6FOiZkUOlSaXpo4QdLHwniNB-p_NEIgRbDnw,46
+pytest_kafka_broker-0.3.0.dist-info/top_level.txt,sha256=nTrYx9xVeK5hsqbhBRL2bgBV_ea-J66_f4Dk8eD-Ci0,20
+pytest_kafka_broker-0.3.0.dist-info/RECORD,,

pytest_kafka_broker-0.1.1.dist-info/RECORD

@@ -1,6 +0,0 @@
-pytest_kafka_broker.py,sha256=h0D80ZYRw_cxB7wdwKvhctSiGyB_GNX4G75w8M_BQF0,6036
-pytest_kafka_broker-0.1.1.dist-info/METADATA,sha256=sihNIutx9OFvjaVwEhO4oP-wiDe6qJCumcYgNovf-3g,683
-pytest_kafka_broker-0.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-pytest_kafka_broker-0.1.1.dist-info/entry_points.txt,sha256=mjFsRbu6FOiZkUOlSaXpo4QdLHwniNB-p_NEIgRbDnw,46
-pytest_kafka_broker-0.1.1.dist-info/top_level.txt,sha256=nTrYx9xVeK5hsqbhBRL2bgBV_ea-J66_f4Dk8eD-Ci0,20
-pytest_kafka_broker-0.1.1.dist-info/RECORD,,