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)