test_keyword_arguments_element.py

import pytest
from mockito import mock, unstub, when

from SeleniumLibrary.keywords import ElementKeywords


@pytest.fixture(scope="function")
def element():
    ctx = mock()
    ctx._browser = mock()
    return ElementKeywords(ctx)


def teardown_function():
    unstub()


def test_element_text_should_be(element):
    locator = "//div"
    webelement = mock()
    webelement.text = "text"
    when(element).find_element(locator).thenReturn(webelement)
    with pytest.raises(AssertionError) as error:
        element.element_text_should_be(locator, "not text")
    assert "should have been" in str(error.value)

    with pytest.raises(AssertionError) as error:
        element.element_text_should_be(locator, "not text", "foobar")
    assert "foobar" in str(error.value)

    webelement.text = "text "
    when(element).find_element(locator).thenReturn(webelement)
    with pytest.raises(AssertionError) as error:
        element.element_text_should_be(locator, "text", strip_spaces=False)
    assert "should have been" in str(error.value)

    with pytest.raises(AssertionError) as error:
        element.element_text_should_be(locator, "text", strip_spaces="LEADING")
    assert "should have been" in str(error.value)

    webelement.text = " text"
    when(element).find_element(locator).thenReturn(webelement)
    with pytest.raises(AssertionError) as error:
        element.element_text_should_be(locator, "text", strip_spaces="TRAILING")
    assert "should have been" in str(error.value)

    webelement.text = "testing  is cool"
    when(element).find_element(locator).thenReturn(webelement)
    with pytest.raises(AssertionError) as error:
        element.element_text_should_be(locator, "testing is cool", collapse_spaces=False)
    assert "should have been" in str(error.value)