tests: fix Confluent TestClient tests

airtai · May 3, 2024 · f33768b · f33768b
1 parent 870c057
commit f33768b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -76,9 +76,7 @@ redis = ["redis>=5.0.0,<6.0.0"]
 telemetry = ["opentelemetry-sdk>=1.24.0,<2.0.0"]
 
 # dev dependencies
-optionals = [
-    "faststream[rabbit,kafka,confluent,nats,redis,telemetry]",
-]
+optionals = ["faststream[rabbit,kafka,confluent,nats,redis,telemetry]"]
 
 devdocs = [
     "mkdocs-material==9.5.18",
@@ -341,4 +339,3 @@ omit = [
 [tool.codespell]
 skip = "./venv,./docs/site/*"
 ignore-words = ".codespell-whitelist.txt"
-
diff --git a/tests/brokers/confluent/test_test_client.py b/tests/brokers/confluent/test_test_client.py
@@ -11,6 +11,8 @@
 class TestTestclient(BrokerTestclientTestcase):
     """A class to represent a test Kafka broker."""
 
+    test_class = TestKafkaBroker
+
     def get_broker(self, apply_types: bool = False):
         return KafkaBroker(apply_types=apply_types)