Skip to content

Commit 9289105

Browse files
authored
Fix test_queue_return flaky test and other nits (#1118)
* Fix flaky test and other nits * Refactoring small changes within `dispatcher.py`
1 parent eaa1369 commit 9289105

File tree

5 files changed

+7
-13
lines changed

5 files changed

+7
-13
lines changed

azure_functions_worker/dispatcher.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ async def connect(cls, host: str, port: int, worker_id: str,
112112
logger.info('Successfully opened gRPC channel to %s:%s ', host, port)
113113
return disp
114114

115-
async def dispatch_forever(self):
115+
async def dispatch_forever(self): # sourcery skip: swap-if-expression
116116
if DispatcherMeta.__current_dispatcher__ is not None:
117117
raise RuntimeError('there can be only one running dispatcher per '
118118
'process')
@@ -312,10 +312,11 @@ async def _handle__functions_metadata_request(self, request):
312312
status=protos.StatusResult.Success)))
313313

314314
try:
315-
fx_metadata_results = []
316315
indexed_functions = loader.index_function_app(function_path)
317316
logger.info('Indexed function app and found %s functions',
318317
len(indexed_functions))
318+
319+
fx_metadata_results = []
319320
if indexed_functions:
320321
indexed_function_logs: List[str] = []
321322
for func in indexed_functions:
@@ -596,8 +597,7 @@ async def _handle__close_shared_memory_resources_request(self, request):
596597
try:
597598
for map_name in map_names:
598599
try:
599-
to_delete_resources = \
600-
False if self._function_data_cache_enabled else True
600+
to_delete_resources = not self._function_data_cache_enabled
601601
success = self._shmem_mgr.free_mem_map(map_name,
602602
to_delete_resources)
603603
results[map_name] = success

tests/endtoend/queue_functions/queue_functions_stein/generic/function_app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ def queue_trigger_message_return(msg: func.QueueMessage) -> bytes:
203203
@app.function_name(name="queue_trigger_return")
204204
@app.generic_trigger(arg_name="msg",
205205
type="queueTrigger",
206-
queue_name="testqueue-message-return",
206+
queue_name="testqueue-return",
207207
connection="AzureWebJobsStorage")
208208
@app.generic_output_binding(
209209
arg_name="$return",

tests/endtoend/queue_functions/queue_trigger_message_return/function.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
"direction": "in",
88
"name": "msg",
99
"queueName": "testqueue-message-return",
10-
"connection": "AzureWebJobsStorage",
10+
"connection": "AzureWebJobsStorage"
1111
},
1212
{
1313
"type": "blob",

tests/endtoend/queue_functions/queue_trigger_return/function.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
"direction": "in",
88
"name": "msg",
99
"queueName": "testqueue-return",
10-
"connection": "AzureWebJobsStorage",
10+
"connection": "AzureWebJobsStorage"
1111
},
1212
{
1313
"type": "blob",

tests/endtoend/test_queue_functions.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ class TestQueueFunctions(testutils.WebHostTestCase):
1111
def get_script_dir(cls):
1212
return testutils.E2E_TESTS_FOLDER / 'queue_functions'
1313

14-
@testutils.retryable_test(3, 5)
1514
def test_queue_basic(self):
1615
r = self.webhost.request('POST', 'put_queue',
1716
data='test-message')
@@ -33,7 +32,6 @@ def test_queue_basic(self):
3332
'time_next_visible', 'pop_receipt', 'dequeue_count'}:
3433
self.assertIsNotNone(msg.get(attr))
3534

36-
@testutils.retryable_test(3, 5)
3735
def test_queue_return(self):
3836
r = self.webhost.request('POST', 'put_queue_return',
3937
data='test-message-return')
@@ -46,7 +44,6 @@ def test_queue_return(self):
4644
self.assertEqual(r.status_code, 200)
4745
self.assertEqual(r.text, 'test-message-return')
4846

49-
@testutils.retryable_test(3, 5)
5047
def test_queue_message_object_return(self):
5148
r = self.webhost.request('POST', 'put_queue_message_return',
5249
data='test-message-object-return')
@@ -59,7 +56,6 @@ def test_queue_message_object_return(self):
5956
self.assertEqual(r.status_code, 200)
6057
self.assertEqual(r.text, 'test-message-object-return')
6158

62-
@testutils.retryable_test(3, 5)
6359
def test_queue_untyped_return(self):
6460
r = self.webhost.request('POST', 'put_queue_untyped_return',
6561
data='test-untyped-return')
@@ -72,7 +68,6 @@ def test_queue_untyped_return(self):
7268
self.assertEqual(r.status_code, 200)
7369
self.assertEqual(r.text, 'test-untyped-return')
7470

75-
@testutils.retryable_test(3, 5)
7671
def test_queue_return_multiple(self):
7772
r = self.webhost.request('POST', 'put_queue_return_multiple',
7873
data='foo')
@@ -83,7 +78,6 @@ def test_queue_return_multiple(self):
8378
# wait for queue_trigger to process the queue item
8479
time.sleep(1)
8580

86-
@testutils.retryable_test(3, 5)
8781
def test_queue_return_multiple_outparam(self):
8882
r = self.webhost.request('POST', 'put_queue_multiple_out',
8983
data='foo')

0 commit comments

Comments
 (0)