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()