Use contextmanager to set attrs for object temporarily

temporarily.py

from contextlib import contextmanager

@contextmanager
def temporarily(obj, **kwargs):
   original_values = {k: getattr(obj, k) for k in kwargs}
   for k, v in kwargs.items():
       setattr(obj, k, v)
   obj.save(update_fields=kwargs.keys())
   try:
       yield
   finally:
       for k, v in original_values.items():
           setattr(obj, k, v)
       obj.save(update_fields=original_values.keys())


def test_should_not_send_notification(self):
    with temporarily(
        user_account.feature_set,
        can_pay_with_credit_card=True,
        can_receive_notifications=False,
    ):
        pay_with_credit_card(user_account, 100)
    self.assertEquals(len(outbox), 0)

https://medium.com/@hakibenita/how-we-replaced-dozens-of-test-fixtures-with-one-simple-function-bac73bfc277d