4
4
from botocore .config import Config
5
5
from botocore .stub import Stubber
6
6
7
- from aws_lambda_powertools .utilities .batch import PartialSQSProcessor , partial_sqs_processor
7
+ from aws_lambda_powertools .utilities .batch import PartialSQSProcessor , batch_processor
8
8
9
9
10
10
@pytest .fixture (scope = "module" )
@@ -50,7 +50,6 @@ def test_partial_sqs_processor_context_with_failure(sqs_event_factory, record_ha
50
50
"""
51
51
Test processor with one failing record
52
52
"""
53
-
54
53
fail_record = sqs_event_factory ("fail" )
55
54
success_record = sqs_event_factory ("success" )
56
55
@@ -76,7 +75,6 @@ def test_partial_sqs_processor_context_only_success(sqs_event_factory, record_ha
76
75
"""
77
76
Test processor without failure
78
77
"""
79
-
80
78
first_record = sqs_event_factory ("success" )
81
79
second_record = sqs_event_factory ("success" )
82
80
@@ -95,7 +93,6 @@ def test_partial_sqs_processor_context_multiple_calls(sqs_event_factory, record_
95
93
"""
96
94
Test processor without failure
97
95
"""
98
-
99
96
first_record = sqs_event_factory ("success" )
100
97
second_record = sqs_event_factory ("success" )
101
98
@@ -110,12 +107,12 @@ def test_partial_sqs_processor_context_multiple_calls(sqs_event_factory, record_
110
107
assert partial_processor .success_messages == [first_record ]
111
108
112
109
113
- def test_partial_sqs_processor_middleware_with_default (sqs_event_factory , record_handler , partial_processor ):
110
+ def test_batch_processor_middleware_with_partial_sqs_processor (sqs_event_factory , record_handler , partial_processor ):
114
111
"""
115
- Test middleware with default partial processor
112
+ Test middleware's integration with PartialSQSProcessor
116
113
"""
117
114
118
- @partial_sqs_processor (record_handler = record_handler , processor = partial_processor )
115
+ @batch_processor (record_handler = record_handler , processor = partial_processor )
119
116
def lambda_handler (event , context ):
120
117
return True
121
118
@@ -134,9 +131,9 @@ def lambda_handler(event, context):
134
131
assert result is True
135
132
136
133
137
- def test_partial_sqs_processor_middleware_with_custom (capsys , sqs_event_factory , record_handler , config ):
134
+ def test_batch_processor_middleware_with_custom_processor (capsys , sqs_event_factory , record_handler , config ):
138
135
"""
139
- Test middle with custom partial processor
136
+ Test middlewares' integration with custom batch processor
140
137
"""
141
138
142
139
class CustomProcessor (PartialSQSProcessor ):
@@ -146,7 +143,7 @@ def failure_handler(self, record, exception):
146
143
147
144
processor = CustomProcessor (config = config )
148
145
149
- @partial_sqs_processor (record_handler = record_handler , processor = processor )
146
+ @batch_processor (record_handler = record_handler , processor = processor )
150
147
def lambda_handler (event , context ):
151
148
return True
152
149
0 commit comments