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

{pytest_kafka_broker-0.3.1 → pytest_kafka_broker-0.4.0}/.gitlab-ci.yml

@@ -7,6 +7,7 @@ include:
         - "3.12"
         - "3.13"
         - "3.14"
+      pytest_options: --cov=pytest_kafka_broker --cov=tests
   - component: git.ligo.org/computing/gitlab/components/python/type-checking@2.3.2
     inputs:
       fail_on_findings: true

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-kafka-broker
-Version: 0.3.1
+Version: 0.4.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.1 → pytest_kafka_broker-0.4.0}/pyproject.toml

@@ -36,6 +36,13 @@ docs = [
 source = "https://git.ligo.org/ultra-swift/pytest-kafka-broker"
 documentation = "https://pytest-kafka-broker.readthedocs.io/"
 
+[tool.coverage.paths]
+# map paths from installed locations back to project source
+source = [
+  "src/pytest_kafka_broker",  # <-- source path
+  "*/pytest_kafka_broker/",  # <-- any installed path
+]
+
 [tool.mypy]
 exclude = [
     "^docs/conf.py$",

{pytest_kafka_broker-0.3.1 → pytest_kafka_broker-0.4.0}/src/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

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

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