Пример #1
0
 def test_add_new_inventory(self):
     Inventory = MagicMock(return_value=0)
     new_item = Inventory(**item_info)
     Inventory.assert_called_with(product_code='12345',
                                  description='refrigerator',
                                  market_price='2100.00',
                                  rental_price='140.00')
Пример #2
0
    def test_add_new_items(self):
        Inventory = MagicMock(return_value=0)
        new_item = Inventory(**item_info)
        Inventory.assert_called_with(product_code='400',
                                     description='refrigerator',
                                     market_price='800.00',
                                     rental_price='300.00')

        with patch('market_prices.get_latest_price', return_value=40.0):
            with patch('builtins.input', side_effect=e_obj):
                e_dict = {}
                main.FULLINVENTORY = {}
                main.add_new_item()
                e_dict['100'] = {
                    'product_code': '100',
                    'description': 'lamp',
                    'market_price': 40.0,
                    'rental_price': 10.0,
                    'brand': 'Decor Therapy',
                    'voltage': 120
                }
                self.assertEqual(e_dict, main.FULLINVENTORY)

        with patch('market_prices.get_latest_price', return_value=600.0):
            with patch('builtins.input', side_effect=f_obj):
                f_dict = {}
                main.FULLINVENTORY = {}
                main.add_new_item()
                f_dict['200'] = {
                    'product_code': '200',
                    'description': 'beanbag',
                    'market_price': 600.0,
                    'rental_price': 100.0,
                    'material': 'faux fur',
                    'size': 'L'
                }
                self.assertEqual(f_dict, main.FULLINVENTORY)

        with patch('market_prices.get_latest_price', return_value=60.0):
            with patch('builtins.input', side_effect=i_obj):
                i_dict = {}
                main.FULLINVENTORY = {}
                main.add_new_item()
                i_dict['300'] = {
                    'product_code': '300',
                    'description': 'recliner',
                    'market_price': 60.0,
                    'rental_price': 20.0
                }
                self.assertEqual(i_dict, main.FULLINVENTORY)