def test_add_new_item_furniture(self):
text_trap = io.StringIO()
sys.stdout = text_trap
item_dict = {}
item_dict["item_code"] = "1"
item_dict["item_description"] = "a"
item_dict["item_rental_price"] = "1.0"
item_dict["item_price"] = "1.0"
item_dict["item_material"] = "a"
item_dict["size"] = "S"
item_dict["style"] = "furniture"
user_input = ["1", "a", "1.0", "y", "a", "S"]
f_inventory = {}
f = Furniture(**item_dict)
f_inventory = {}
with patch('builtins.input', side_effect=user_input):
f_inventory = add_new_item(f_inventory)
new_dict = f_inventory["1"]
f = Furniture(**item_dict)
new_dict1 = f.return_as_dictionary()
self.assertEqual(new_dict["product_code"], new_dict1["product_code"])
self.assertEqual(new_dict["description"], new_dict1["description"])
self.assertEqual(new_dict["rental_price"], new_dict1["rental_price"])
self.assertEqual(new_dict["material"], new_dict1["material"])
self.assertEqual(new_dict["size"], new_dict1["size"])
sys.stdout = sys.__stdout__
def add_new_item_furniture(self,f_inventory):
text_trap = io.StringIO()
sys.stdout = text_trap
user_input = ["1", "a", "1.0", "y", "a", "S"]
with patch('builtins.input', side_effect=user_input):
f_inventory = add_new_item(f_inventory)
sys.stdout = sys.__stdout__
return f_inventory
def test_add_new_item_inventory(self):
text_trap = io.StringIO()
sys.stdout = text_trap
user_input = ["3", "c", "3.0", "n", "n"]
f_inventory = {}
with patch('builtins.input', side_effect=user_input):
f_inventory = add_new_item(f_inventory)
new_dict = f_inventory["3"]
item_dict = {}
item_dict["item_code"] = "3"
item_dict["item_description"] = "c"
item_dict["item_rental_price"] = "3.0"
item_dict["item_price"] = get_latest_price()
i = Inventory(**item_dict)
new_dict1 = i.return_as_dictionary()
self.assertEqual(new_dict["product_code"], new_dict1["product_code"])
self.assertEqual(new_dict["description"], new_dict1["description"])
self.assertEqual(new_dict["rental_price"], new_dict1["rental_price"])
self.assertEqual(new_dict["market_price"], new_dict1["market_price"])
sys.stdout = sys.__stdout__
def test_add_new_item_electrical(self):
text_trap = io.StringIO()
sys.stdout = text_trap
user_input = ["2", "b", "2.0", "n", "y", "b", "2.0"]
f_inventory = {}
with patch('builtins.input', side_effect=user_input):
f_inventory = add_new_item(f_inventory)
new_dict = f_inventory["2"]
item_dict = {}
item_dict["item_code"] = "2"
item_dict["item_description"] = "b"
item_dict["item_rental_price"] = "2.0"
item_dict["item_price"] = "2.0"
item_dict["brand"] = "b"
item_dict["voltage"] = "2.0"
e = ElectricAppliances(**item_dict)
new_dict1 = e.return_as_dictionary()
self.assertEqual(new_dict["product_code"], new_dict1["product_code"])
self.assertEqual(new_dict["description"], new_dict1["description"])
self.assertEqual(new_dict["brand"], new_dict1["brand"])
self.assertEqual(new_dict["voltage"], new_dict1["voltage"])
sys.stdout = sys.__stdout__
