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