updated mocks of Coder.send() to fix tests

This commit is contained in:
Paul Gauthier 2024-04-27 14:08:21 -07:00
parent e71cf03f36
commit 594a6bd934
2 changed files with 20 additions and 10 deletions

View file

@ -312,8 +312,9 @@ new
"""
coder.partial_response_function_call = dict()
return []
coder.send = MagicMock(side_effect=mock_send)
coder.send = mock_send
# Call the run method with a message
coder.run(with_message="hi")
@ -354,8 +355,9 @@ new
"""
coder.partial_response_function_call = dict()
return []
coder.send = MagicMock(side_effect=mock_send)
coder.send = mock_send
# Call the run method with a message
coder.run(with_message="hi")