Skip to content

Commit 642555f

Browse files
committed
Updated code for Asphalt 5
1 parent d522aa7 commit 642555f

File tree

5 files changed

+40
-25
lines changed

5 files changed

+40
-25
lines changed

.pre-commit-config.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
# * Run "pre-commit install".
66
repos:
77
- repo: https://github.com/pre-commit/pre-commit-hooks
8-
rev: v4.5.0
8+
rev: v4.6.0
99
hooks:
1010
- id: check-toml
1111
- id: check-yaml
@@ -16,14 +16,14 @@ repos:
1616
- id: trailing-whitespace
1717

1818
- repo: https://github.com/astral-sh/ruff-pre-commit
19-
rev: v0.3.5
19+
rev: v0.4.8
2020
hooks:
2121
- id: ruff
2222
args: [--fix, --show-fixes]
2323
- id: ruff-format
2424

2525
- repo: https://github.com/pre-commit/mirrors-mypy
26-
rev: v1.9.0
26+
rev: v1.10.0
2727
hooks:
2828
- id: mypy
2929
additional_dependencies:

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,13 +60,13 @@ local_scheme = "dirty-tag"
6060
[tool.ruff.lint]
6161
extend-select = [
6262
"ASYNC", # flake8-async
63-
"W", # pycodestyle warnings
6463
"G", # flake8-logging-format
6564
"I", # isort
6665
"ISC", # flake8-implicit-str-concat
6766
"PGH", # pygrep-hooks
6867
"RUF100", # unused noqa (yesqa)
6968
"UP", # pyupgrade
69+
"W", # pycodestyle warnings
7070
]
7171

7272
[tool.ruff.lint.isort]

tests/test_admin_client.py

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import pytest
66
from aiokafka.admin import AIOKafkaAdminClient
7-
from asphalt.core import Context, add_resource, get_resource_nowait
7+
from asphalt.core import Context, add_resource, get_resource_nowait, start_component
88

99
from asphalt.kafka import KafkaAdminComponent
1010

@@ -23,14 +23,17 @@ async def test_default_settings() -> None:
2323
)
2424
async def test_existing_resource(lookup: bool) -> None:
2525
client = AIOKafkaAdminClient()
26-
component = KafkaAdminComponent(
27-
existing_resource="default" if lookup else client, resource_name="alt"
28-
)
2926
async with Context():
3027
if lookup:
3128
add_resource(client)
3229

33-
await component.start()
30+
await start_component(
31+
KafkaAdminComponent,
32+
{
33+
"existing_resource": "default" if lookup else client,
34+
"resource_name": "alt",
35+
},
36+
)
3437
assert get_resource_nowait(AIOKafkaAdminClient, "alt") is client
3538

3639

@@ -47,11 +50,13 @@ async def test_existing_resource_conflict() -> None:
4750
)
4851
async def test_ssl_context(lookup: bool) -> None:
4952
ssl_context = ssl.create_default_context()
50-
component = KafkaAdminComponent(ssl_context="default" if lookup else ssl_context)
5153
async with Context():
5254
if lookup:
5355
add_resource(ssl_context)
5456

55-
await component.start()
57+
await start_component(
58+
KafkaAdminComponent,
59+
{"ssl_context": "default" if lookup else ssl_context},
60+
)
5661
client = get_resource_nowait(AIOKafkaAdminClient)
5762
assert client._client._ssl_context is ssl_context

tests/test_consumer.py

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import pytest
66
from aiokafka import AIOKafkaConsumer
7-
from asphalt.core import Context, add_resource, get_resource_nowait
7+
from asphalt.core import Context, add_resource, get_resource_nowait, start_component
88

99
from asphalt.kafka import KafkaConsumerComponent
1010

@@ -23,14 +23,17 @@ async def test_default_settings() -> None:
2323
)
2424
async def test_existing_resource(lookup: bool) -> None:
2525
consumer = AIOKafkaConsumer()
26-
component = KafkaConsumerComponent(
27-
existing_resource="default" if lookup else consumer, resource_name="alt"
28-
)
2926
async with Context():
3027
if lookup:
3128
add_resource(consumer)
3229

33-
await component.start()
30+
await start_component(
31+
KafkaConsumerComponent,
32+
{
33+
"existing_resource": "default" if lookup else consumer,
34+
"resource_name": "alt",
35+
},
36+
)
3437
assert get_resource_nowait(AIOKafkaConsumer, "alt") is consumer
3538

3639

@@ -47,11 +50,13 @@ async def test_existing_resource_conflict() -> None:
4750
)
4851
async def test_ssl_context(lookup: bool) -> None:
4952
ssl_context = ssl.create_default_context()
50-
component = KafkaConsumerComponent(ssl_context="default" if lookup else ssl_context)
5153
async with Context():
5254
if lookup:
5355
add_resource(ssl_context)
5456

55-
await component.start()
57+
await start_component(
58+
KafkaConsumerComponent,
59+
{"ssl_context": "default" if lookup else ssl_context},
60+
)
5661
consumer = get_resource_nowait(AIOKafkaConsumer)
5762
assert consumer._client._ssl_context is ssl_context

tests/test_producer.py

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import pytest
66
from aiokafka import AIOKafkaProducer
7-
from asphalt.core import Context, add_resource, get_resource_nowait
7+
from asphalt.core import Context, add_resource, get_resource_nowait, start_component
88

99
from asphalt.kafka import KafkaProducerComponent
1010

@@ -23,14 +23,17 @@ async def test_default_settings() -> None:
2323
)
2424
async def test_existing_resource(lookup: bool) -> None:
2525
producer = AIOKafkaProducer()
26-
component = KafkaProducerComponent(
27-
existing_resource="default" if lookup else producer, resource_name="alt"
28-
)
2926
async with Context():
3027
if lookup:
3128
add_resource(producer)
3229

33-
await component.start()
30+
await start_component(
31+
KafkaProducerComponent,
32+
{
33+
"existing_resource": "default" if lookup else producer,
34+
"resource_name": "alt",
35+
},
36+
)
3437
assert get_resource_nowait(AIOKafkaProducer, "alt") is producer
3538

3639

@@ -47,11 +50,13 @@ async def test_existing_resource_conflict() -> None:
4750
)
4851
async def test_ssl_context(lookup: bool) -> None:
4952
ssl_context = ssl.create_default_context()
50-
component = KafkaProducerComponent(ssl_context="default" if lookup else ssl_context)
5153
async with Context():
5254
if lookup:
5355
add_resource(ssl_context)
5456

55-
await component.start()
57+
await start_component(
58+
KafkaProducerComponent,
59+
{"ssl_context": "default" if lookup else ssl_context},
60+
)
5661
producer = get_resource_nowait(AIOKafkaProducer)
5762
assert producer.client._ssl_context is ssl_context

0 commit comments

Comments
 (0)