Exemplo n.º 1
0
 def test_capture_page_screenshot_else_case(self):
     mock_desk = MagicMock()
     mock_desk._get_screenshot_paths = MagicMock(
         return_value=['path', 'link'])
     del mock_desk._current_application().get_screenshot_as_file
     DesktopLibrary.capture_page_screenshot(mock_desk, 'filename')
     mock_desk._get_screenshot_paths.assert_called_with('filename')
Exemplo n.º 2
0
 def test_capture_page_screenshot(self):
     mock_desk = MagicMock()
     mock_desk._get_screenshot_paths = MagicMock(
         return_value=['path', 'link'])
     DesktopLibrary.capture_page_screenshot(mock_desk)
     mock_desk._get_screenshot_paths.assert_called_with(None)