MYPY = False
if MYPY:
import typing
from unittest.mock import DEFAULT, Mock
from stone.backend import Backend
def _mock_output(backend):
recorded_output = []
output_buffer_to_string = backend.output_buffer_to_string
def record_output():
recorded_output.append(output_buffer_to_string())
return DEFAULT
backend.output_buffer_to_string = Mock( wraps=output_buffer_to_string,
side_effect=record_output)
def get_output():
backend.output_buffer_to_string = output_buffer_to_string
return recorded_output[0] if recorded_output else ''
return get_output