Come si prende in giro un metodo RelatedManager in Django?

Question

Ho un gestore di cliente per un modello di Django, che ridefinisce il metodo create per salvare anche alcuni oggetti correlati:

class CustomManager(models.Manager): 
    def create(self, amount, user, description): 
     txn = self.get_query_set().create(user, description) 
     txn.budget_transactions.create(amount) 
     return txn 

La mia domanda è: come faccio a prendere in giro il chiamata a txn.budget_transactions.create a sollevare un'eccezione?

L'attributo budget_transactions dell'oggetto txn è un'istanza di django.db.models.fields.related.RelatedManager. Usando mock.patch per deridere questa classe non funziona come è dichiarato dinamicamente - non può essere importato direttamente.

Qualcuno sa come fare?

Answer 1

La ragione per cui non è possibile impostare il RelatedManager su un oggetto fittizio è perché django ha sostituito il metodo impostato su sull'oggetto. Quindi, mentre sembra che il mock sia impostato correttamente perché non ci sono lamentele, in realtà sta impostando automaticamente budget_transactions sul RelatedManager. Quindi, se hai davvero bisogno di restituire una simulazione, dovrai sostituire il metodo con il metodo che restituisce il RelatedManager e restituire invece un oggetto fittato.

dovrebbe finire lookig somthing come:

@mock.patch('django.db.models.fields.related.ForeignRelatedObjectsDescriptor.__get__') 
def test_campaign_cancel(self, mock_manager): 
    mock_manager.return_value = mock.MagicMock() 
    mock_manager.return_value.create = Exception('Boom!') 

Detto ci sono molte insidie ​​di questo approccio, poiché sarà override un metodo nucleo django e ora TUTTE RelatedManagers restituirà un oggetto deriso. Da quello che ho vissuto finora è probabilmente più facile esplorare altre opzioni.