def createMenu(self, menu: M.Menu, choice):
if menu.menuType() == "ErrorMenu":
return M.ChoiceMenu(choice,Menufactory.QUESTION_OPTION)
elif menu.menuType() == "ChoiceMenu":
if (choice == "Yes"):
return M.KeywordMenu()
else:
return M.SearchResultsMenu()
elif menu.menuType() == "KeywordMenu":
return M.SearchResultsMenu(choice)
elif menu.menuType() == "SearchResultsMenu":
menu.usersChoice(choice)
return menu
else:
return M.ErrorMenu(choice)
def tearDown(self):
self.menu = M.ErrorMenu(error_menu)
def setUp(self):
self.menu = M.ErrorMenu(error_menu)
self.factory = MF.Menufactory()
def __init__(self, array):
self.previousMenu = []
self.MenuFactory = MF.Menufactory()
self.menu = Menu.ErrorMenu(array)
