def test_update_should_raise_not_found_when_cant_find_product(self): repository = MongoKitRepository(self.mongo_db) kit_products = [ KitProduct(product_sku='FASD-498', quantity=2, discount_percentage=10.5), KitProduct(product_sku='FASD-1489', quantity=1, discount_percentage=10.5) ] kit = Kit(id='5f566e9c1022bd08188d674b', name='Sony Gaming Pack', sku='FASD-789', kit_products=kit_products) with self.assertRaises(NotFound): repository.update(kit)
def test_update(self): repository = MongoKitRepository(self.mongo_db) kit_products = [ KitProduct(product_sku='FASD-498', quantity=2, discount_percentage=10.5), KitProduct(product_sku='FASD-1489', quantity=1, discount_percentage=10.5) ] kit = Kit(name='Sony Gaming Pack', sku='FASD-789', kit_products=kit_products) kit_id = repository.add(kit) kit.define_id(kit_id) repository.add( Kit(name='Sony Gaming Pack II', sku='FASD-7894', kit_products=[ KitProduct(product_sku='FASD-4988', quantity=9, discount_percentage=10.5), KitProduct(product_sku='FASD-1489', quantity=1, discount_percentage=10.5) ])) kit.update_infos(name='Sony Gaming Pack I', kit_products=[ KitProduct(product_sku='FASD-498', quantity=7, discount_percentage=80.5), KitProduct(product_sku='FASD-1429', quantity=5, discount_percentage=72.5) ]) repository.update(kit) kit = repository.get_by_id(kit_id) self.assertEqual(kit.name, 'Sony Gaming Pack I') self.assertEqual(kit.kit_products[0], kit.kit_products[0]) self.assertEqual(kit.kit_products[1], kit.kit_products[1])