def test_key_error_on_update(self, bad_mock): """ Test KeyError on update """ bad_mock.side_effect = KeyError() promotion = Promotion("A002", "dog", False) promotion.save() promotion.productid = 'Fifi' promotion.update()
def test_update_a_promotion_fail(self): promotion = Promotion( title="test_create", promo_type=PromoType.DISCOUNT, amount=10, start_date=datetime(2020, 10, 17), end_date=datetime(2020, 10, 18), is_site_wide=True, ) try: promotion.update() except: print("Update called with empty ID field")
def test_update_a_promotion(self): """ Update a Promotion """ promotion = Promotion( title="test_create", promo_type=PromoType.DISCOUNT, amount=10, start_date=datetime(2020, 10, 17), end_date=datetime(2020, 10, 18), is_site_wide=True, ) promotion.create() self.assertEqual(promotion.id, 1) # Change it and update it promotion.title = "test_update" promotion.update() self.assertEqual(promotion.id, 1) # Fetch it back and make sure the id hasn't changed # but the data did change promotions = Promotion.all() self.assertEqual(len(promotions), 1) self.assertEqual(promotions[0].title, "test_update")