11import unittest
2+ from itertools import chain , repeat
23from unittest import TestCase
34from unittest .mock import MagicMock
45from uuid import uuid4
@@ -118,9 +119,7 @@ def test_calls_registered_handler_on_message(self):
118119
119120 with DishReceiver (context ) as receiver :
120121 receiver ._poller = MagicMock (spec = Poller )
121- receiver ._poller .poll .side_effect = [
122- {context .socket .return_value : POLLIN },
123- ]
122+ receiver ._poller .poll .side_effect = chain ([{context .socket .return_value : POLLIN }], repeat ({}))
124123 receiver .register (handler )
125124
126125 # When
@@ -138,9 +137,7 @@ def test_handles_message_receive_error_gracefully(self):
138137
139138 with DishReceiver (context ) as receiver :
140139 receiver ._poller = MagicMock (spec = Poller )
141- receiver ._poller .poll .side_effect = [
142- {context .socket .return_value : POLLIN },
143- ]
140+ receiver ._poller .poll .side_effect = chain ([{context .socket .return_value : POLLIN }], repeat ({}))
144141 receiver .register (handler )
145142
146143 # When
@@ -159,9 +156,7 @@ def test_handles_handler_execution_error_gracefully(self):
159156
160157 with DishReceiver (context ) as receiver :
161158 receiver ._poller = MagicMock (spec = Poller )
162- receiver ._poller .poll .side_effect = [
163- {context .socket .return_value : POLLIN },
164- ]
159+ receiver ._poller .poll .side_effect = chain ([{context .socket .return_value : POLLIN }], repeat ({}))
165160 receiver .register (handler )
166161
167162 # When
0 commit comments