def test_element_has_class_no_elements_found(): mock_finder = MagicMock() mock_finder.elements.return_value = [] interrogate = Interrogator(None, mock_finder, None) result = interrogate.element_has_class(default_page_element, "test_class") assert_that(result, equal_to(False), "No elements should have been found")
def test_element_has_class_incorrect_class(): mock_element = MagicMock() mock_element.is_displayed.return_value = True mock_element.get_attribute.return_value = "class1 class2" mock_finder = MagicMock() mock_finder.elements.return_value = [mock_element] interrogate = Interrogator(None, mock_finder, None) result = interrogate.element_has_class(default_page_element, "test_class") assert_that(result, equal_to(False), "The class should not match")