Skip to content

Commit 8e255e3

Browse files
1 parent 82f9586 commit 8e255e3

File tree

3 files changed

+40
-17
lines changed

3 files changed

+40
-17
lines changed

‎tests/unit/pubsub_v1/conftest.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ def span_exporter():
4545
provider.add_span_processor(processor)
4646
yield exporter
4747

48+
4849
@pytest.fixture()
4950
def modify_google_logger_propagation():
5051
"""

‎tests/unit/pubsub_v1/subscriber/test_heartbeater.py‎

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,10 @@
2727

2828
import pytest
2929

30-
def test_heartbeat_inactive_manager_active_rpc(caplog, modify_google_logger_propagation):
30+
31+
def test_heartbeat_inactive_manager_active_rpc(
32+
caplog, modify_google_logger_propagation
33+
):
3134
caplog.set_level(logging.DEBUG)
3235

3336
manager = mock.create_autospec(
@@ -45,7 +48,10 @@ def test_heartbeat_inactive_manager_active_rpc(caplog, modify_google_logger_prop
4548
assert "exiting" in caplog.text
4649

4750

48-
def test_heartbeat_inactive_manager_inactive_rpc(caplog, modify_google_logger_propagation,):
51+
def test_heartbeat_inactive_manager_inactive_rpc(
52+
caplog,
53+
modify_google_logger_propagation,
54+
):
4955
caplog.set_level(logging.DEBUG)
5056

5157
manager = mock.create_autospec(

‎tests/unit/pubsub_v1/subscriber/test_streaming_pull_manager.py‎

Lines changed: 31 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -603,7 +603,9 @@ def test__maybe_release_messages_below_overload():
603603
assert call_args[1].ack_id in ("ack_foo", "ack_bar")
604604

605605

606-
def test__maybe_release_messages_negative_on_hold_bytes_warning(caplog, modify_google_logger_propagation):
606+
def test__maybe_release_messages_negative_on_hold_bytes_warning(
607+
caplog, modify_google_logger_propagation
608+
):
607609
manager = make_manager(
608610
flow_control=types.FlowControl(max_messages=10, max_bytes=1000)
609611
)
@@ -978,7 +980,9 @@ def test_send_unary_modack_api_call_error(caplog, modify_google_logger_propagati
978980
assert "The front fell off" in caplog.text
979981

980982

981-
def test_send_unary_ack_retry_error_exactly_once_disabled_no_futures(caplog, modify_google_logger_propagation):
983+
def test_send_unary_ack_retry_error_exactly_once_disabled_no_futures(
984+
caplog, modify_google_logger_propagation
985+
):
982986
caplog.set_level(logging.DEBUG)
983987

984988
manager, _, _, _, _, _ = make_running_manager()
@@ -1014,7 +1018,9 @@ def test_send_unary_ack_retry_error_exactly_once_disabled_no_futures(caplog, mod
10141018
assert "signaled streaming pull manager shutdown" in caplog.text
10151019

10161020

1017-
def test_send_unary_ack_retry_error_exactly_once_disabled_with_futures(caplog, modify_google_logger_propagation):
1021+
def test_send_unary_ack_retry_error_exactly_once_disabled_with_futures(
1022+
caplog, modify_google_logger_propagation
1023+
):
10181024
caplog.set_level(logging.DEBUG)
10191025

10201026
manager, _, _, _, _, _ = make_running_manager()
@@ -1054,7 +1060,9 @@ def test_send_unary_ack_retry_error_exactly_once_disabled_with_futures(caplog, m
10541060
assert future2.result() == subscriber_exceptions.AcknowledgeStatus.SUCCESS
10551061

10561062

1057-
def test_send_unary_ack_retry_error_exactly_once_enabled_no_futures(caplog, modify_google_logger_propagation):
1063+
def test_send_unary_ack_retry_error_exactly_once_enabled_no_futures(
1064+
caplog, modify_google_logger_propagation
1065+
):
10581066
caplog.set_level(logging.DEBUG)
10591067

10601068
manager, _, _, _, _, _ = make_running_manager()
@@ -1090,7 +1098,9 @@ def test_send_unary_ack_retry_error_exactly_once_enabled_no_futures(caplog, modi
10901098
assert "signaled streaming pull manager shutdown" in caplog.text
10911099

10921100

1093-
def test_send_unary_ack_retry_error_exactly_once_enabled_with_futures(caplog, modify_google_logger_propagation):
1101+
def test_send_unary_ack_retry_error_exactly_once_enabled_with_futures(
1102+
caplog, modify_google_logger_propagation
1103+
):
10941104
caplog.set_level(logging.DEBUG)
10951105

10961106
manager, _, _, _, _, _ = make_running_manager()
@@ -1136,7 +1146,9 @@ def test_send_unary_ack_retry_error_exactly_once_enabled_with_futures(caplog, mo
11361146
)
11371147

11381148

1139-
def test_send_unary_modack_retry_error_exactly_once_disabled_no_future(caplog, modify_google_logger_propagation):
1149+
def test_send_unary_modack_retry_error_exactly_once_disabled_no_future(
1150+
caplog, modify_google_logger_propagation
1151+
):
11401152
caplog.set_level(logging.DEBUG)
11411153

11421154
manager, _, _, _, _, _ = make_running_manager()
@@ -1162,8 +1174,7 @@ def test_send_unary_modack_retry_error_exactly_once_disabled_no_future(caplog, m
11621174

11631175

11641176
def test_send_unary_modack_retry_error_exactly_once_disabled_with_futures(
1165-
caplog,
1166-
modify_google_logger_propagation
1177+
caplog, modify_google_logger_propagation
11671178
):
11681179
caplog.set_level(logging.DEBUG)
11691180

@@ -1192,8 +1203,7 @@ def test_send_unary_modack_retry_error_exactly_once_disabled_with_futures(
11921203

11931204

11941205
def test_send_unary_modack_retry_error_exactly_once_enabled_no_futures(
1195-
caplog,
1196-
modify_google_logger_propagation
1206+
caplog, modify_google_logger_propagation
11971207
):
11981208
caplog.set_level(logging.DEBUG)
11991209

@@ -1220,8 +1230,7 @@ def test_send_unary_modack_retry_error_exactly_once_enabled_no_futures(
12201230

12211231

12221232
def test_send_unary_modack_retry_error_exactly_once_enabled_with_futures(
1223-
caplog,
1224-
modify_google_logger_propagation
1233+
caplog, modify_google_logger_propagation
12251234
):
12261235
caplog.set_level(logging.DEBUG)
12271236

@@ -1274,7 +1283,9 @@ def test_heartbeat_inactive():
12741283
assert not result
12751284

12761285

1277-
def test_heartbeat_stream_ack_deadline_seconds(caplog, modify_google_logger_propagation):
1286+
def test_heartbeat_stream_ack_deadline_seconds(
1287+
caplog, modify_google_logger_propagation
1288+
):
12781289
caplog.set_level(logging.DEBUG)
12791290
manager = make_manager()
12801291
manager._rpc = mock.create_autospec(bidi.BidiRpc, instance=True)
@@ -2090,7 +2101,10 @@ def test__on_response_disable_exactly_once():
20902101
assert manager._stream_ack_deadline == 60
20912102

20922103

2093-
def test__on_response_exactly_once_immediate_modacks_fail(caplog, modify_google_logger_propagation,):
2104+
def test__on_response_exactly_once_immediate_modacks_fail(
2105+
caplog,
2106+
modify_google_logger_propagation,
2107+
):
20942108
manager, _, dispatcher, leaser, _, scheduler = make_running_manager()
20952109
manager._callback = mock.sentinel.callback
20962110

@@ -2162,7 +2176,9 @@ def complete_futures_with_error(*args, **kwargs):
21622176
assert manager.load == 0.001
21632177

21642178

2165-
def test__on_response_exactly_once_immediate_modacks_fail_non_invalid(caplog, modify_google_logger_propagation):
2179+
def test__on_response_exactly_once_immediate_modacks_fail_non_invalid(
2180+
caplog, modify_google_logger_propagation
2181+
):
21662182
manager, _, dispatcher, leaser, _, scheduler = make_running_manager()
21672183
manager._callback = mock.sentinel.callback
21682184

0 commit comments

Comments
 (0)