pytest-kafka-broker 0.0.0__tar.gz

pytest_kafka_broker-0.0.0/.gitignore

@@ -0,0 +1,5 @@
+__pycache__
+*.egg-info
+dist
+_build
+api

pytest_kafka_broker-0.0.0/.gitlab-ci.yml

@@ -0,0 +1,14 @@
+include:
+  - component: git.ligo.org/computing/gitlab/components/python/all@2.3.2
+    inputs:
+      code_quality_analyzer: ruff
+      python_versions:
+        - "3.11"
+        - "3.12"
+        - "3.13"
+        - "3.14"
+
+python_test:
+  before_script:
+    - apt update
+    - apt install -y --no-install-recommends default-jdk-headless

pytest_kafka_broker-0.0.0/.pre-commit-config.yaml

@@ -0,0 +1,48 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v6.0.0
+    hooks:
+      - id: check-added-large-files
+        # Prevent giant files from being committed.
+      - id: check-case-conflict
+        # Check for files with names that would conflict on a case-insensitive
+        # filesystem like MacOS HFS+ or Windows FAT.
+      - id: check-json
+        # Attempts to load all json files to verify syntax.
+      - id: check-merge-conflict
+        # Check for files that contain merge conflict strings.
+      - id: check-symlinks
+        # Checks for symlinks which do not point to anything.
+      - id: check-toml
+        # Attempts to load all TOML files to verify syntax.
+      - id: check-xml
+        # Attempts to load all xml files to verify syntax.
+      - id: check-yaml
+        # Attempts to load all yaml files to verify syntax.
+      - id: detect-private-key
+        # Checks for the existence of private keys.
+      - id: end-of-file-fixer
+        # Makes sure files end in a newline and only a newline.
+      - id: trailing-whitespace
+        # Trims trailing whitespace.
+
+  - repo: https://github.com/codespell-project/codespell
+    rev: v2.4.1
+    hooks:
+      - id: codespell
+        args: ["--write-changes"]
+        additional_dependencies:
+          - tomli
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.14.14
+    hooks:
+      - id: ruff-check
+        args: ["--extend-select", "I", "--fix"]
+      - id: ruff-format
+
+  - repo: https://github.com/tofuutils/pre-commit-opentofu
+    rev: v2.2.2
+    hooks:
+      - id: tofu_fmt
+      - id: tofu_validate

pytest_kafka_broker-0.0.0/.readthedocs.yml

@@ -0,0 +1,18 @@
+version: 2
+
+build:
+  apt_packages:
+    - graphviz
+  os: ubuntu-24.04
+  tools:
+    python: '3.11'
+
+python:
+  install:
+    - method: pip
+      path: .
+      extra_requirements:
+        - docs
+
+sphinx:
+  configuration: docs/conf.py

pytest_kafka_broker-0.0.0/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: pytest-kafka-broker
+Version: 0.0.0
+Summary: Pytest plugin to run a single-broker Kafka cluster
+Author-email: Leo Singer <leo.singer@ligo.org>
+License-Expression: Apache-2.0
+Classifier: Framework :: Pytest
+Classifier: Topic :: System :: Networking
+Requires-Python: >=3.11
+Requires-Dist: astropy
+Requires-Dist: confluent-kafka
+Requires-Dist: rich
+Requires-Dist: pytest-asyncio
+Provides-Extra: docs
+Requires-Dist: sphinx-astropy[confv2]; extra == "docs"
+Requires-Dist: sphinx-automodapi>=0.20.0; extra == "docs"

pytest_kafka_broker-0.0.0/README.md

@@ -0,0 +1,3 @@
+# pytest-kafka-broker
+
+A Pytest plugin to run a single-broker Kafka cluster.

pytest_kafka_broker-0.0.0/docs/Makefile

@@ -0,0 +1,20 @@
+# Minimal makefile for Sphinx documentation
+#
+
+# You can set these variables from the command line, and also
+# from the environment for the first two.
+SPHINXOPTS    ?=
+SPHINXBUILD   ?= sphinx-build
+SOURCEDIR     = .
+BUILDDIR      = _build
+
+# Put it first so that "make" without argument is like "make help".
+help:
+	@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
+
+.PHONY: help Makefile
+
+# Catch-all target: route all unknown targets to Sphinx using the new
+# "make mode" option.  $(O) is meant as a shortcut for $(SPHINXOPTS).
+%: Makefile
+	@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)

pytest_kafka_broker-0.0.0/docs/conf.py

@@ -0,0 +1,210 @@
+# -*- coding: utf-8 -*-
+# Licensed under a 3-clause BSD style license - see LICENSE.rst
+#
+# Astropy documentation build configuration file.
+#
+# This file is execfile()d with the current directory set to its containing dir.
+#
+# Note that not all possible configuration values are present in this file.
+#
+# All configuration values have a default. Some values are defined in
+# the global Astropy configuration which is loaded here before anything else.
+# See astropy.sphinx.conf for which values are set there.
+
+# If extensions (or modules to document with autodoc) are in another directory,
+# add these directories to sys.path here. If the directory is relative to the
+# documentation root, use os.path.abspath to make it absolute, like shown here.
+# sys.path.insert(0, os.path.abspath('..'))
+# IMPORTANT: the above commented section was generated by sphinx-quickstart, but
+# is *NOT* appropriate for astropy or Astropy affiliated packages. It is left
+# commented out with this explanation to make it clear why this should not be
+# done. If the sys.path entry above is added, when the astropy.sphinx.conf
+# import occurs, it will import the *source* version of astropy instead of the
+# version installed (if invoked as "make html" or directly with sphinx), or the
+# version in the build directory (if "python setup.py build_sphinx" is used).
+# Thus, any C-extensions that are needed to build the documentation will *not*
+# be accessible, and the documentation will not build correctly.
+
+import datetime
+import os
+import sys
+import tomllib
+
+sys.path.insert(0, os.path.realpath(".."))
+
+try:
+    from sphinx_astropy.conf.v2 import *  # noqa
+except ImportError:
+    print(
+        "ERROR: the documentation requires the sphinx-astropy package to be installed"
+    )
+    sys.exit(1)
+
+# Get configuration information from pyproject.toml
+
+with open(os.path.join(os.path.dirname(__file__), "..", "pyproject.toml"), "rb") as f:
+    project_metadata = tomllib.load(f)["project"]
+
+# -- General configuration ----------------------------------------------------
+
+# By default, highlight as Python 3.
+highlight_language = "python3"
+
+# If your documentation needs a minimal Sphinx version, state it here.
+# needs_sphinx = '1.2'
+
+# To perform a Sphinx version check that needs to be more specific than
+# major.minor, call `check_sphinx_version("X.Y.Z")` here.
+# check_sphinx_version("1.2.1")
+
+# List of patterns, relative to source directory, that match files and
+# directories to ignore when looking for source files.
+exclude_patterns.append("_templates")  # noqa: F405
+
+# This is added to the end of RST files - a good place to put substitutions to
+# be used globally.
+# rst_epilog += r"""
+# """  # noqa: F405
+
+# -- Project information ------------------------------------------------------
+
+# This does not *have* to match the package name, but typically does
+project = project_metadata["name"]
+author = project_metadata["authors"][0]["name"]
+copyright = "{0}, {1}".format(datetime.datetime.now().year, author)
+
+# The version info for the project you're documenting, acts as replacement for
+# |version| and |release|, also used in various other places throughout the
+# built documents.
+
+# import_module(project_metadata["name"])
+# package = sys.modules[project_metadata["name"]]
+
+# The short X.Y version.
+# version = package.__version__.split("-", 1)[0]
+# The full version, including alpha/beta/rc tags.
+# release = package.__version__
+
+# Only include dev docs in dev version.
+# dev = "dev" in release
+
+
+# -- Options for HTML output --------------------------------------------------
+
+# A NOTE ON HTML THEMES
+# The global astropy configuration uses a custom theme, 'bootstrap-astropy',
+# which is installed along with astropy. A different theme can be used or
+# the options for this theme can be modified by overriding some of the
+# variables set in the global configuration. The variables set in the
+# global configuration are listed below, commented out.
+
+
+# Add any paths that contain custom themes here, relative to this directory.
+# To use a different custom theme, add the directory containing the theme.
+# html_theme_path = []
+
+# The theme to use for HTML and HTML Help pages.  See the documentation for
+# a list of builtin themes. To override the custom theme, set this to the
+# name of a builtin theme or the name of a custom theme in html_theme_path.
+# html_theme = None
+
+html_static_path = ["_static"]
+# html_css_files = []
+
+
+# Custom sidebar templates, maps document names to template names.
+# html_sidebars = {}
+
+# The name of an image file (relative to this directory) to place at the top
+# of the sidebar.
+# html_logo = ''
+
+# The name of an image file (within the static path) to use as favicon of the
+# docs.  This file should be a Windows icon file (.ico) being 16x16 or 32x32
+# pixels large.
+# html_favicon = ''
+
+# If not '', a 'Last updated on:' timestamp is inserted at every page bottom,
+# using the given strftime format.
+# html_last_updated_fmt = ''
+
+# The name for this set of Sphinx documents.  If None, it defaults to
+# "<project> v<release> documentation".
+# html_title = ""
+
+# Output file base name for HTML help builder.
+htmlhelp_basename = project + "doc"
+
+# Prefixes that are ignored for sorting the Python module index
+modindex_common_prefix = ["shigawire."]
+
+html_context = {
+    "default_mode": "light",
+    "to_be_indexed": ["stable", "latest"],
+    "is_development": True,
+    "doc_path": "docs",
+}
+
+# -- Options for LaTeX output -------------------------------------------------
+
+# Grouping the document tree into LaTeX files. List of tuples
+# (source start file, target name, title, author, documentclass [howto/manual]).
+latex_documents = [
+    ("index", project + ".tex", project + " Documentation", author, "manual")
+]
+
+
+# -- Options for manual page output -------------------------------------------
+
+# One entry per manual page. List of tuples
+# (source start file, name, description, authors, manual section).
+man_pages = [("index", project.lower(), project + " Documentation", [author], 1)]
+
+
+# -- Options for linkcheck output -------------------------------------------
+linkcheck_retry = 5
+linkcheck_ignore = []
+linkcheck_timeout = 180
+linkcheck_anchors = False
+
+# -- Turn on nitpicky mode for sphinx (to warn about references not found) ----
+#
+# nitpicky = True
+# nitpick_ignore = []
+#
+# Some warnings are impossible to suppress, and you can list specific references
+# that should be ignored in a nitpick-exceptions file which should be inside
+# the docs/ directory. The format of the file should be:
+#
+# <type> <class>
+#
+# for example:
+#
+# py:class astropy.io.votable.tree.Element
+# py:class astropy.io.votable.tree.SimpleElement
+# py:class astropy.io.votable.tree.SimpleElementWithContent
+#
+# Uncomment the following lines to enable the exceptions:
+#
+# for line in open('nitpick-exceptions'):
+#     if line.strip() == "" or line.startswith("#"):
+#         continue
+#     dtype, target = line.split(None, 1)
+#     target = target.strip()
+#     nitpick_ignore.append((dtype, six.u(target)))
+
+
+# -- Options for intersphinx --------------------------------------------------
+intersphinx_mapping.update(  # noqa: F405
+    {
+        "confluent-kafka": (
+            "https://docs.confluent.io/platform/current/clients/confluent-kafka-python/html/",
+            None,
+        ),
+    }
+)
+
+# -- Merge type annotations with numpydoc parameter docstrings ----------------
+extensions.remove("numpydoc")  # noqa: F405
+extensions.append("sphinx.ext.napoleon")  # noqa: F405
+autodoc_typehints = "description"

pytest_kafka_broker-0.0.0/docs/index.rst

@@ -0,0 +1,16 @@
+pytest-kafka-broker documentation
+=================================
+
+This is a pytest plugin to run a temporary, local, single-broker Kafka cluster.
+
+*******
+Example
+*******
+
+.. literalinclude:: ../tests/test_kafka.py
+
+*********
+Reference
+*********
+
+.. automodapi:: pytest_kafka_broker

pytest_kafka_broker-0.0.0/docs/make.bat

@@ -0,0 +1,35 @@
+@ECHO OFF
+
+pushd %~dp0
+
+REM Command file for Sphinx documentation
+
+if "%SPHINXBUILD%" == "" (
+	set SPHINXBUILD=sphinx-build
+)
+set SOURCEDIR=.
+set BUILDDIR=_build
+
+%SPHINXBUILD% >NUL 2>NUL
+if errorlevel 9009 (
+	echo.
+	echo.The 'sphinx-build' command was not found. Make sure you have Sphinx
+	echo.installed, then set the SPHINXBUILD environment variable to point
+	echo.to the full path of the 'sphinx-build' executable. Alternatively you
+	echo.may add the Sphinx directory to PATH.
+	echo.
+	echo.If you don't have Sphinx installed, grab it from
+	echo.https://www.sphinx-doc.org/
+	exit /b 1
+)
+
+if "%1" == "" goto help
+
+%SPHINXBUILD% -M %1 %SOURCEDIR% %BUILDDIR% %SPHINXOPTS% %O%
+goto end
+
+:help
+%SPHINXBUILD% -M help %SOURCEDIR% %BUILDDIR% %SPHINXOPTS% %O%
+
+:end
+popd

pytest_kafka_broker-0.0.0/pyproject.toml

@@ -0,0 +1,33 @@
+[build-system]
+requires = [
+    "setuptools",
+    "setuptools-scm>=8",
+]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "pytest-kafka-broker"
+authors = [{name = "Leo Singer", email = "leo.singer@ligo.org"}]
+description = "Pytest plugin to run a single-broker Kafka cluster"
+license = "Apache-2.0"
+dynamic = ["version"]
+requires-python = ">=3.11"
+dependencies = [
+    "astropy",
+    "confluent-kafka",
+    "rich",
+    "pytest-asyncio",
+]
+classifiers = [
+    "Framework :: Pytest",
+    "Topic :: System :: Networking",
+]
+
+[project.entry-points.pytest11]
+kafka_broker = "pytest_kafka_broker"
+
+[project.optional-dependencies]
+docs = [
+    "sphinx-astropy[confv2]",
+    "sphinx-automodapi>=0.20.0",
+]

pytest_kafka_broker-0.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

pytest_kafka_broker-0.0.0/src/pytest_kafka_broker.egg-info/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: pytest-kafka-broker
+Version: 0.0.0
+Summary: Pytest plugin to run a single-broker Kafka cluster
+Author-email: Leo Singer <leo.singer@ligo.org>
+License-Expression: Apache-2.0
+Classifier: Framework :: Pytest
+Classifier: Topic :: System :: Networking
+Requires-Python: >=3.11
+Requires-Dist: astropy
+Requires-Dist: confluent-kafka
+Requires-Dist: rich
+Requires-Dist: pytest-asyncio
+Provides-Extra: docs
+Requires-Dist: sphinx-astropy[confv2]; extra == "docs"
+Requires-Dist: sphinx-automodapi>=0.20.0; extra == "docs"

pytest_kafka_broker-0.0.0/src/pytest_kafka_broker.egg-info/SOURCES.txt

@@ -0,0 +1,19 @@
+.gitignore
+.gitlab-ci.yml
+.pre-commit-config.yaml
+.readthedocs.yml
+README.md
+pyproject.toml
+docs/Makefile
+docs/conf.py
+docs/index.rst
+docs/make.bat
+src/pytest_kafka_broker.py
+src/pytest_kafka_broker.egg-info/PKG-INFO
+src/pytest_kafka_broker.egg-info/SOURCES.txt
+src/pytest_kafka_broker.egg-info/dependency_links.txt
+src/pytest_kafka_broker.egg-info/entry_points.txt
+src/pytest_kafka_broker.egg-info/requires.txt
+src/pytest_kafka_broker.egg-info/top_level.txt
+tests/__init__.py
+tests/test_kafka.py

pytest_kafka_broker-0.0.0/src/pytest_kafka_broker.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

pytest_kafka_broker-0.0.0/src/pytest_kafka_broker.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[pytest11]
+kafka_broker = pytest_kafka_broker

pytest_kafka_broker-0.0.0/src/pytest_kafka_broker.egg-info/requires.txt

@@ -0,0 +1,8 @@
+astropy
+confluent-kafka
+rich
+pytest-asyncio
+
+[docs]
+sphinx-astropy[confv2]
+sphinx-automodapi>=0.20.0

pytest_kafka_broker-0.0.0/src/pytest_kafka_broker.egg-info/top_level.txt

@@ -0,0 +1 @@
+pytest_kafka_broker

pytest_kafka_broker-0.0.0/src/pytest_kafka_broker.py

@@ -0,0 +1,187 @@
+import asyncio
+import subprocess
+from dataclasses import dataclass
+from pathlib import Path
+from tarfile import TarFile
+from tempfile import TemporaryDirectory
+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 confluent_kafka import Consumer, Producer
+from confluent_kafka.aio import AIOConsumer, AIOProducer
+from rich.status import Status
+
+__all__ = (
+    "kafka_broker",
+    "KafkaBrokerContext",
+)
+
+
+SCALA_VERSION = "2.13"
+KAFKA_VERSION = "4.1.1"
+
+
+async def wait_port(port: int, timeout: float = 0.25):
+    """Wait until a connection is detected listening on the given port."""
+    while True:
+        try:
+            _, writer = await asyncio.open_connection("localhost", port)
+        except OSError:
+            await asyncio.sleep(timeout)
+        else:
+            writer.close()
+            await writer.wait_closed()
+            return
+
+
+@pytest.fixture(scope="session")
+def kafka_home() -> Path:
+    """Download and install Kafka into a cached directory.
+
+    Returns the path where Kafka is installed.
+    """
+    dirname = f"kafka_{SCALA_VERSION}-{KAFKA_VERSION}"
+    cache_path = get_cache_dir_path() / __package__
+    dest_path = cache_path / dirname
+    if not dest_path.exists():
+        dest_path.mkdir(parents=True, exist_ok=True)
+        with (
+            Status("Downloading Kafka"),
+            get_readable_fileobj(
+                f"https://dlcdn.apache.org/kafka/{KAFKA_VERSION}/{dirname}.tgz",
+                encoding="binary",
+                cache=True,
+            ) as download,
+            TarFile(fileobj=download) as tarfile,
+            TemporaryDirectory(dir=cache_path) as temp_dir,
+        ):
+            tarfile.extractall(temp_dir)
+            (Path(temp_dir) / dirname).rename(dest_path)
+    return dest_path
+
+
+_doc = """{}
+
+Parameters
+----------
+config
+    Extra Kafka client configuration properties. See list in the
+    `librdkafka documentation <https://github.com/confluentinc/librdkafka/blob/master/CONFIGURATION.md>`_.
+"""
+
+
+@dataclass
+class KafkaBrokerContext:
+    """Information and convenience methods for a temporary Kafka cluster.
+
+    This object is returned by :func:`kafka_broker`.
+    """
+
+    bootstrap_server: str
+    """Kafka bootstrap server in the form :samp:`{host}:{port}`."""
+
+    def config(self, config: dict | None = None):
+        return {**(config or {}), "bootstrap.servers": self.bootstrap_server}
+
+    def producer(self, config: dict | None = None) -> Producer:
+        return Producer(self.config(config))
+
+    def consumer(self, config: dict | None = None) -> Consumer:
+        return Consumer(self.config(config))
+
+    def aio_producer(self, config: dict | None = None) -> AIOProducer:
+        return AIOProducer(self.config(config))
+
+    def aio_consumer(self, config: dict | None = None) -> AIOConsumer:
+        return AIOConsumer(self.config(config))
+
+    config.__doc__ = _doc.format("Get the configuration for a Kafka client.")
+    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(
+        "Create an asynchronous Kafka producer connected to the cluster."
+    )
+    aio_consumer.__doc__ = _doc.format(
+        "Create an asynchronous Kafka consumer connected to the cluster."
+    )
+
+
+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
+    """
+    kafka_storage = kafka_home / "bin" / "kafka-storage.sh"
+    kafka_server_start = kafka_home / "bin" / "kafka-server-start.sh"
+    config_path = tmp_path / "server.properties"
+    data_path = tmp_path / "run"
+    data_path.mkdir()
+    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()
+    config_path.write_text(
+        f"""
+        process.roles=broker,controller
+        node.id=1
+        controller.quorum.bootstrap.servers=127.0.0.1:{controller_port}
+        listeners=PLAINTEXT://127.0.0.1:{plaintext_port},CONTROLLER://127.0.0.1:{controller_port}
+        controller.listener.names=CONTROLLER
+        listener.security.protocol.map=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT
+        log.dirs={data_path}
+        num.recovery.threads.per.data.dir=1
+        offsets.topic.replication.factor=1
+        share.coordinator.state.topic.replication.factor=1
+        share.coordinator.state.topic.min.isr=1
+        transaction.state.log.replication.factor=1
+        transaction.state.log.min.isr=1
+        """
+    )
+    with Status("Starting Kafka broker"):
+        subprocess.run(
+            [
+                kafka_storage,
+                "format",
+                "--standalone",
+                "-t",
+                str(uuid4()),
+                "-c",
+                config_path,
+            ],
+            env=env,
+            check=True,
+            stdout=subprocess.DEVNULL,
+        )
+        process = await asyncio.create_subprocess_exec(
+            kafka_server_start,
+            config_path,
+            env=env,
+            stdin=None,
+            stdout=subprocess.DEVNULL,
+            stderr=None,
+        )
+    with Status(f"Waiting for connection on port {plaintext_port}"):
+        exited = asyncio.create_task(process.wait())
+        port = asyncio.create_task(wait_port(plaintext_port))
+        done, _ = await asyncio.wait(
+            (exited, port), return_when=asyncio.FIRST_COMPLETED
+        )
+        if exited in done:
+            port.cancel()
+            raise RuntimeError("Kafka broker terminated unexpectedly")
+    try:
+        yield KafkaBrokerContext(f"127.0.0.1:{plaintext_port}")
+    finally:
+        with Status("Stopping Kafka broker"):
+            process.terminate()
+            await exited

pytest_kafka_broker-0.0.0/tests/test_kafka.py

@@ -0,0 +1,38 @@
+import pytest
+
+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)
+    with kafka_broker.consumer(
+        {"group.id": group_id, "auto.offset.reset": "earliest"}
+    ) as consumer:
+        consumer.subscribe([topic])
+        (message,) = consumer.consume()
+        assert message.value() == payload
+
+
+@pytest.mark.asyncio
+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)
+    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"}
+    )
+    try:
+        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