def test_is_image_visible_by_checking_src_no_src_url(): interrogate = Interrogator(None, None, None) mock_get_attribute = MagicMock(return_value=None) interrogate.get_attribute = mock_get_attribute result = interrogate.is_image_visible_by_checking_src(default_page_element) mock_get_attribute.assert_called_once_with(default_page_element, "src") assert_that( result, equal_to(True), "The image should be considered visible if it has no source URL")
def test_is_image_visible_by_checking_src_success(): mock_driver = MagicMock() mock_driver.execute_script.return_value = "test/url" interrogate = Interrogator(mock_driver, None, None) mock_get_attribute = MagicMock(return_value="/source") interrogate.get_attribute = mock_get_attribute mock_response = MagicMock() mock_response.status_code = 200 mock_get = MagicMock(return_value=mock_response) requests.get = mock_get result = interrogate.is_image_visible_by_checking_src(default_page_element) mock_get_attribute.assert_called_once_with(default_page_element, "src") mock_get.assert_called_once_with("test/url/source") mock_driver.execute_script.assert_called_once_with( "return window.location.origin") assert_that(result, equal_to(True), "The image should have been found")