def test_get_text_current_element_set(self):
mock_desk = MagicMock()
mock_desk.current_element = MagicMock()
DesktopLibrary._get_text(mock_desk, mock_desk.current_element)
mock_desk._element_find.assert_not_called()
def test_get_text_element_none(self):
mock_desk = MagicMock()
mock_desk._check_for_cached_element = MagicMock(return_value=None)
result = DesktopLibrary._get_text(mock_desk, mock_desk.current_element)
self.assertEqual(result, None)
def test_get_text(self):
mock_desk = MagicMock()
mock_desk.current_element = None
DesktopLibrary._get_text(mock_desk, 1)
mock_desk._check_for_cached_element.assert_called_with(1)
