def api_mocking_me(name):
try:
gif = me.via_gif(name)
except Exception:
return 'Error: Unable to generate image'
return send_file(io.BytesIO(gif),
attachment_filename='%s.gif' % name,
mimetype='image/gif')
def test_mocking_me_via_gif_when_resp_200_then_get_gif(self):
# Arrange
name = 'First Last'
# TODO set patch object's return value(s)
# Act
result = via_gif(name)
# Assert
assert result == "gif content"
def test_mocking_me_via_gif_when_resp_200_then_get_gif(self, mock_post):
# Arrange
name = 'First Last'
# TODO set patch object's return value(s)
mock_post.return_value.status_code = 200
mock_post.return_value.content = "gif content"
# Act
result = via_gif(name)
# Assert
assert result == "gif content"