def test_take_screenshot(mocks, result_out):
from phoneauto.scriptgenerator.scriptgenerator_ui import get_filedialog
with patch.object(get_filedialog(), 'asksaveasfilename', return_value='abc'):
click_sidebar_button(mocks, 'screenshot_button')
assert mocks.device.screenshot.called
mocks.dummy_img.save.assert_called_with('abc')
def test_take_screenshot_cancel(mocks):
ui = create_scriptgenerator_ui()
with patch.object(
scriptgenerator_ui.get_filedialog(),
'asksaveasfilename', return_value=''):
ui._take_screenshot()
def test_take_screenshot_cancel(mocks):
ui = create_scriptgenerator_ui()
with patch.object(scriptgenerator_ui.get_filedialog(),
'asksaveasfilename',
return_value=''):
ui._take_screenshot()
