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

pytest_kafka_broker/__init__.py

@@ -2,7 +2,9 @@ import asyncio
 import subprocess
 from collections.abc import AsyncGenerator
 from dataclasses import dataclass
+from errno import EADDRINUSE
 from pathlib import Path
+from socket import socket
 from tarfile import TarFile
 from tempfile import TemporaryDirectory
 from uuid import uuid4
@@ -75,6 +77,36 @@ config
 """
 
 
+def _unused_tcp_port(default: int = 0) -> int:
+    with socket() as sock:
+        try:
+            sock.bind(("127.0.0.1", default))
+        except OSError as e:
+            if e.errno != EADDRINUSE:
+                raise
+            sock.bind(("127.0.0.1", 0))
+        _, port = sock.getsockname()
+    return port
+
+
+@pytest.fixture
+def find_unused_tcp_port():
+    """Unused TCP port factory.
+
+    This is similar to `unused_tcp_port_factory` from pytest_asyncio, but it
+    supports a default port argument, and is not session-scoped.
+    """
+    used = set()
+
+    def factory(default: int = 0) -> int:
+        while (port := _unused_tcp_port(default)) in used:
+            pass
+        used.add(port)
+        return port
+
+    return factory
+
+
 @dataclass
 class KafkaBrokerContext:
     """Information and convenience methods for a temporary Kafka cluster.
@@ -120,7 +152,7 @@ del _doc
 
 @pytest_asyncio.fixture
 async def kafka_broker(
-    kafka_home, tmp_path, unused_tcp_port_factory
+    kafka_home, tmp_path, find_unused_tcp_port
 ) -> AsyncGenerator[KafkaBrokerContext]:
     """Pytest fixture to run a local, temporary Kafka broker."""
     kafka_storage = kafka_home / "bin" / "kafka-storage.sh"
@@ -131,8 +163,8 @@ async def kafka_broker(
     log_path = tmp_path / "log"
     log_path.mkdir()
     env = {"LOG_DIR": str(log_path)}
-    plaintext_port = unused_tcp_port_factory()
-    controller_port = unused_tcp_port_factory()
+    plaintext_port = find_unused_tcp_port(9092)
+    controller_port = find_unused_tcp_port(9093)
     config_path.write_text(
         f"""
         process.roles=broker,controller
@@ -183,7 +215,9 @@ async def kafka_broker(
             port.cancel()
             raise RuntimeError("Kafka broker terminated unexpectedly")
     try:
-        yield KafkaBrokerContext(f"127.0.0.1:{plaintext_port}")
+        bootstrap_server = f"127.0.0.1:{plaintext_port}"
+        print(f"Kafka broker running at {bootstrap_server}")
+        yield KafkaBrokerContext(bootstrap_server)
     finally:
         with Status("Stopping Kafka broker"):
             try:

{pytest_kafka_broker-0.3.1.dist-info → pytest_kafka_broker-0.5.0.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+pytest_kafka_broker/__init__.py,sha256=N7nTmm22Dp8TCWDImJysySVDs0eq9k-DrW1swiwBNOs,7415
+pytest_kafka_broker/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pytest_kafka_broker-0.5.0.dist-info/METADATA,sha256=UQHkABMoQmBsZptt6rPAHnQpnuO5-sMbF7K_oLBJkpc,683
+pytest_kafka_broker-0.5.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+pytest_kafka_broker-0.5.0.dist-info/entry_points.txt,sha256=mjFsRbu6FOiZkUOlSaXpo4QdLHwniNB-p_NEIgRbDnw,46
+pytest_kafka_broker-0.5.0.dist-info/top_level.txt,sha256=nTrYx9xVeK5hsqbhBRL2bgBV_ea-J66_f4Dk8eD-Ci0,20
+pytest_kafka_broker-0.5.0.dist-info/RECORD,,

pytest_kafka_broker-0.3.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-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,,