override funzione python locale-variabile in unittest

Question

Ho un metodo in python (2.7) che fa pippo, e si arresta dopo 5 minuti se foo non ha funzionato.

def keep_trying(self): 
    timeout = 300 #empirically derived, appropriate timeout 
    end_time = time.time() + timeout 
    while (time.time() < end_time): 
     result = self.foo() 
     if (result == 'success'): 
      break 
     time.sleep(2) 
    else: 
     raise MyException('useful msg here') 

So che alcuni possibili risultati di foo(), quindi sto usando finta di falsi quei valori di ritorno. Il problema è che non voglio che il test venga eseguito 5 minuti prima che venga visualizzata l'eccezione.

C'è un modo per ignorare quel valore locale di timeout? Mi piacerebbe che fossero solo pochi secondi in modo che io possa vedere il ciclo provare un paio di volte, poi rinunciare e rilanciare.

Quanto segue non funziona:

@patch.object(myClass.keep_trying, 'timeout') 
@patch.object(myClass, 'foo') 
def test_keep_trying(self, mock_foo, mock_timeout): 
    mock_foo.return_value = 'failed' 
    mock_timeout.return_value = 10 # raises AttributeError 
    mock_timeout = 10 # raises AttributeError 
    ... 

Answer 1

Piuttosto che cercare di deridere il valore se timeout, si vorrà prendere in giro il valore di ritorno di time.time().

ad es.

@patch.object(time, 'time') 
def test_keep_trying(self, mock_time): 
    mock_time.side_effect = iter([100, 200, 300, 400, 500, 600, 700, 800]) 
    ... 

Ora la prima volta time.time() si chiama, si otterrà il valore di 100, quindi dovrebbe timeout quando dopo un paio di giri del vostro ciclo while. Puoi anche prendere in giro time.sleep e contare quante volte viene chiamato per assicurarti che parte del codice funzioni correttamente.

---

Un altro approccio (che non è del tutto ortogonale a quella di cui sopra) è quello di permettere all'utente di passare una parola chiave timeout opzionale alla funzione:

def keep_trying(self, timeout=300): 
    ... 

Ciò consente di specificare qualunque timeout vuoi nei test (e nel futuro codice che non vuole aspettare 5 minuti ;-).

Answer 2

Non è possibile prendere in giro la variabile locale di una funzione. Per rendere il codice più facile da testare, cambiare a, per esempio:

def keep_trying(self, timeout=300): 
    end_time = time.time() + timeout 
    # etc, as above 

così diventa banale per i test da eseguire con un timeout più breve!