homepage.py

#pylint: disable=C0111
#pylint: disable=W0621

from lettuce import world, step
from nose.tools import assert_in, assert_equals


@step(u'I should see the following Partners in the Partners section')
def i_should_see_partner(step):
    partners = world.browser.find_by_css(".partner .name span")
    names = set(span.text for span in partners)
    for partner in step.hashes:
        assert_in(partner['Partner'], names)


@step(u'I should see the following links and ids')
def should_see_a_link_called(step):
    for link_id_pair in step.hashes:
        link_id = link_id_pair['id']
        text = link_id_pair['Link']
        link = world.browser.find_by_id(link_id)
        assert len(link) > 0
        assert_equals(link.text, text)