from behave import given, when, then from Navigator import Navigator from getpass import getpass nav = Navigator() nav.clearBrowser() # We will need to clear "br" since br is using mechanize which is a module and can't be instantiated multiple times, only once @given('I am logged in') def impl(context): if (nav.logged_in is False): username = getpass('PID: ') password = getpass() nav.logged_in = nav.login(username, password) assert nav.logged_in is True @given('I enter an incorrect CRN number "{entered_crn}" and my course info is checked') def impl(context, entered_crn): context.response = nav.validCourseInfo(crn=entered_crn, subj="", crse="") context.expected_errors = ["Please enter a valid CRN"] # We are expecting this error @given('I enter a correct CRN number "{entered_crn}" and my course info is checked') def impl(context, entered_crn): context.response = nav.validCourseInfo(crn=entered_crn, subj="", crse="") context.expected_errors = "" @then('I will see "{error_message}"')