Skip to content

Commit 28301ac

Browse files
committed
changed test result handling
1 parent d1a2f88 commit 28301ac

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/example_code/bedrock-agent-runtime/test/test_bedrock_agent_runtime_wrapper.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,8 @@ async def test_invoke_flow(make_stubber, error_code):
8282
stubber.stub_invoke_flow(expected_params, response, error_code=error_code)
8383

8484
if error_code is None:
85-
wrapper.invoke_flow(flow_id, flow_alias_id, inputs, execution_id)
85+
result = wrapper.invoke_flow(flow_id, flow_alias_id, inputs, execution_id)
86+
assert result is not None
8687
else:
8788
with pytest.raises(ClientError) as exc_info:
8889
async for _ in wrapper.invoke_flow(

0 commit comments

Comments
 (0)