@@ -16,8 +16,8 @@ def setup_mocks():
1616 config .maximum_runtime_milliseconds = "500"
1717 config .ssm_prefix = "/dl/test/mesh"
1818
19- client_lookup = Mock ()
20- client_lookup .is_valid_sender .return_value = True # Default to valid sender
19+ sender_lookup = Mock ()
20+ sender_lookup .is_valid_sender .return_value = True # Default to valid sender
2121
2222 mesh_client = Mock (spec = MeshClient )
2323
@@ -27,7 +27,7 @@ def setup_mocks():
2727
2828 return (
2929 config ,
30- client_lookup ,
30+ sender_lookup ,
3131 mesh_client ,
3232 log ,
3333 polling_metric
@@ -67,7 +67,7 @@ class TestMeshMessageProcessor:
6767
6868 def test_process_messages_iterates_through_inbox (self , mock_event_publisher_class , mock_getenv ):
6969 """Test that processor iterates through all messages in MESH inbox"""
70- (config , client_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
70+ (config , sender_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
7171 message1 = setup_message_data ("1" )
7272 message2 = setup_message_data ("2" )
7373
@@ -79,7 +79,7 @@ def test_process_messages_iterates_through_inbox(self, mock_event_publisher_clas
7979
8080 processor = MeshMessageProcessor (
8181 config = config ,
82- client_lookup = client_lookup ,
82+ sender_lookup = sender_lookup ,
8383 mesh_client = mesh_client ,
8484 get_remaining_time_in_millis = get_remaining_time_in_millis ,
8585 log = log ,
@@ -88,7 +88,7 @@ def test_process_messages_iterates_through_inbox(self, mock_event_publisher_clas
8888
8989 mesh_client .iterate_all_messages .side_effect = [
9090 [message1 , message2 ], []]
91- client_lookup .is_valid_sender .return_value = True
91+ sender_lookup .is_valid_sender .return_value = True
9292
9393 processor .process_messages ()
9494
@@ -98,7 +98,7 @@ def test_process_messages_iterates_through_inbox(self, mock_event_publisher_clas
9898
9999 def test_process_messages_stops_near_timeout (self , mock_event_publisher_class , mock_getenv ):
100100 """Test that processor stops processing when near timeout"""
101- (config , client_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
101+ (config , sender_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
102102 message1 = setup_message_data ("1" )
103103
104104 mock_getenv .side_effect = lambda key , default = '' : {
@@ -108,7 +108,7 @@ def test_process_messages_stops_near_timeout(self, mock_event_publisher_class, m
108108
109109 processor = MeshMessageProcessor (
110110 config = config ,
111- client_lookup = client_lookup ,
111+ sender_lookup = sender_lookup ,
112112 mesh_client = mesh_client ,
113113 get_remaining_time_in_millis = get_remaining_time_in_millis_near_timeout ,
114114 log = log ,
@@ -119,12 +119,12 @@ def test_process_messages_stops_near_timeout(self, mock_event_publisher_class, m
119119
120120 processor .process_messages ()
121121
122- client_lookup .is_valid_sender .assert_not_called ()
122+ sender_lookup .is_valid_sender .assert_not_called ()
123123 polling_metric .record .assert_called_once ()
124124
125125 def test_process_message_with_valid_sender (self , mock_event_publisher_class , mock_getenv ):
126126 """Test processing a single message from valid sender"""
127- (config , client_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
127+ (config , sender_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
128128 message = setup_message_data ("1" )
129129
130130 mock_getenv .side_effect = lambda key , default = '' : {
@@ -138,25 +138,25 @@ def test_process_message_with_valid_sender(self, mock_event_publisher_class, moc
138138
139139 processor = MeshMessageProcessor (
140140 config = config ,
141- client_lookup = client_lookup ,
141+ sender_lookup = sender_lookup ,
142142 mesh_client = mesh_client ,
143143 get_remaining_time_in_millis = get_remaining_time_in_millis ,
144144 log = log ,
145145 polling_metric = polling_metric
146146 )
147147
148- client_lookup .is_valid_sender .return_value = True
148+ sender_lookup .is_valid_sender .return_value = True
149149
150150 processor .process_message (message )
151151
152152 mesh_client .handshake .assert_called_once ()
153- client_lookup .is_valid_sender .assert_called_once_with (message .sender )
153+ sender_lookup .is_valid_sender .assert_called_once_with (message .sender )
154154 mock_event_publisher .send_events .assert_called_once ()
155155 message .acknowledge .assert_not_called () # Only acknowledged on auth error
156156
157157 def test_process_message_with_unknown_sender (self , mock_event_publisher_class , mock_getenv ):
158158 """Test that messages from unknown senders are rejected silently"""
159- (config , client_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
159+ (config , sender_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
160160 message = setup_message_data ("1" )
161161
162162 mock_getenv .side_effect = lambda key , default = '' : {
@@ -166,24 +166,24 @@ def test_process_message_with_unknown_sender(self, mock_event_publisher_class, m
166166
167167 processor = MeshMessageProcessor (
168168 config = config ,
169- client_lookup = client_lookup ,
169+ sender_lookup = sender_lookup ,
170170 mesh_client = mesh_client ,
171171 get_remaining_time_in_millis = get_remaining_time_in_millis ,
172172 log = log ,
173173 polling_metric = polling_metric
174174 )
175175
176176 # Invalid sender
177- client_lookup .is_valid_sender .return_value = False
177+ sender_lookup .is_valid_sender .return_value = False
178178
179179 processor .process_message (message )
180180
181- client_lookup .is_valid_sender .assert_called_once_with (message .sender )
181+ sender_lookup .is_valid_sender .assert_called_once_with (message .sender )
182182 message .acknowledge .assert_called_once ()
183183
184184 def test_process_messages_across_multiple_iterations (self , mock_event_publisher_class , mock_getenv ):
185185 """Test that processor continues polling until no messages remain"""
186- (config , client_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
186+ (config , sender_lookup , mesh_client , log , polling_metric ) = setup_mocks ()
187187 message1 = setup_message_data ("1" )
188188 message2 = setup_message_data ("2" )
189189 message3 = setup_message_data ("3" )
@@ -195,7 +195,7 @@ def test_process_messages_across_multiple_iterations(self, mock_event_publisher_
195195
196196 processor = MeshMessageProcessor (
197197 config = config ,
198- client_lookup = client_lookup ,
198+ sender_lookup = sender_lookup ,
199199 mesh_client = mesh_client ,
200200 get_remaining_time_in_millis = get_remaining_time_in_millis ,
201201 log = log ,
@@ -207,11 +207,11 @@ def test_process_messages_across_multiple_iterations(self, mock_event_publisher_
207207 [message3 ], # Second iteration
208208 [] # Third iteration - empty, stops
209209 ]
210- client_lookup .is_valid_sender .return_value = True
210+ sender_lookup .is_valid_sender .return_value = True
211211
212212 processor .process_messages ()
213213
214214 mesh_client .handshake .assert_called_once ()
215215 assert mesh_client .iterate_all_messages .call_count == 3
216- assert client_lookup .is_valid_sender .call_count == 3
216+ assert sender_lookup .is_valid_sender .call_count == 3
217217 polling_metric .record .assert_called_once ()
0 commit comments