def test_main_menu_add_new(self): """ Tests for menu item 1 selection""" while True: try: with patch('builtins.input', side_effect='1'): self.assertEqual(mainmenu(), main.addnewitem()) except StopIteration as error: return error
def test_main_get_item_info(self): """ Tests for item menu 2 selection""" while True: try: with patch('builtins.input', side_effect='2'): self.assertEqual(mainmenu(), main.addnewitem()) except StopIteration as error: return error
def test_mainmenu_2(self, mocked_input): '''doc string''' mocked_input.side_effect = ['2', '1'] expect = mainmenu()() InventoryDic = { '1': { 'ProductCode': '1', 'Description': 'desc', 'MarketPrice': 24, 'RentalPrice': 20 } } self.assertEqual(expect, InventoryDic)
def test_mainmenu_1_electric(self, mocked_input): '''doc string''' mocked_input.side_effect = ['1', '1', 'desc', 20, 'n', 'y', 'ge', 110] expect = mainmenu()() ElectricDic = { '1': { 'ProductCode': '1', 'Description': 'desc', 'MarketPrice': 24, 'RentalPrice': 20, 'Brand': 'ge', 'Voltage': 110 } } self.assertEqual(expect, ElectricDic)
def test_mainmenu_1_furniture(self, mocked_input): '''doc string''' mocked_input.side_effect = ['1', '1', 'desc', 20, 'y', 'wood', 'M'] expect = mainmenu()() FurnitureDic = { '1': { 'ProductCode': '1', 'Description': 'desc', 'MarketPrice': 24, 'RentalPrice': 20, 'Material': 'wood', 'Size': 'M' } } self.assertEqual(expect, FurnitureDic)
def test_mainmenu_q(self, mocked_input): '''doc string''' mocked_input.side_effect = ['q'] with self.assertRaises(SystemExit): mainmenu()()
def miniGame(): import main main.mainmenu()
def gomain(): import main main.mainmenu()