def test_find_euh_top_menu_buttons(self): data = ("employee data", "time management", "Payroll Data", "BENEFITS", "WorKFORce administration", "TaLeNt MaNaGeMeNt", "Strategy & Communication","Service Center") for topmenu_name in data: button = LIB._get_euh_top_menu_button(topmenu_name) self.assertEquals("button", button.tag_name) locator = OR["topmenu_%s" % (topmenu_name.lower())][5:] button_id = locator.split(" > ")[0] #print "name = %s button_id = %s button class = %s" % (topmenu_name, button_id, button.get_attribute("class")) self.assertTrue(button.get_attribute("class").find(button_id) > -1)