from lettuce import world, step


@step('I fill in the registration form$')
def i_fill_in_the_registration_form(step):
    register_form = world.browser.find_by_css('form#register_form')
    register_form.find_by_name('email').fill('robot+studio@edx.org')
    register_form.find_by_name('password').fill('test')
    register_form.find_by_name('username').fill('robot-studio')
    register_form.find_by_name('name').fill('Robot Studio')
    register_form.find_by_name('terms_of_service').check()


@step('I press the Create My Account button on the registration form$')
def i_press_the_button_on_the_registration_form(step):
    register_form = world.browser.find_by_css('form#register_form')
    submit_css = 'button#submit'
    register_form.find_by_css(submit_css).click()


@step('I should see be on the studio home page$')
def i_should_see_be_on_the_studio_home_page(step):
    assert world.browser.find_by_css('div.inner-wrapper')


@step(u'I should see the message "([^"]*)"$')
def i_should_see_the_message(step, msg):
    assert world.browser.is_text_present(msg, 5)
