Mocking una variabile globale

Question

Ho cercato di implementare alcuni test unitari per un modulo. Un modulo di esempio di nome alphabet.py è la seguente:

import database 

def length_letters(): 
    return len(letters) 

def contains_letter(letter): 
    return True if letter in letters else False 


letters = database.get('letters') # returns a list of letters 

mi piacerebbe prendere in giro la risposta da un database con alcuni valori della mia scelta, ma il codice qui sotto non sembra funzionare.

import unittests 
import alphabet 
from unittest.mock import patch 


class TestAlphabet(unittest.TestCase): 
    @patch('alphabet.letters') 
    def setUp(self, mock_letters): 
     mock_letters.return_value = ['a', 'b', 'c'] 

    def test_length_letters(self): 
     self.assertEqual(3, alphabet.length_letters()) 

    def test_contains_letter(self): 
     self.assertTrue(alphabet.contains_letter('a')) 

Ho visto molti esempi in cui 'patch' è applicata a metodi e classi, ma non a variabili. Preferisco non applicare la patch al metodo database.get perché potrei utilizzarlo di nuovo con parametri diversi in seguito, quindi avrei bisogno di una risposta diversa.

Cosa sto facendo di sbagliato qui?

Answer 1

Non è necessario utilizzare il mock. Basta importare il modulo e modificare il valore del globale all'interno setUp():

import alphabet 

class TestAlphabet(unittest.TestCase): 
    def setUp(self): 
     alphabet.letters = ['a', 'b', 'c'] 

Answer 2

Prova questa:

import unittests 
import alphabet 
from unittest.mock import patch 


class TestAlphabet(unittest.TestCase): 
    def setUp(self): 
     self.mock_letters = mock.patch.object(
      alphabet, 'letters', return_value=['a', 'b', 'c'] 
     ) 

    def test_length_letters(self): 
     with self.mock_letters: 
      self.assertEqual(3, alphabet.length_letters()) 

    def test_contains_letter(self): 
     with self.mock_letters: 
      self.assertTrue(alphabet.contains_letter('a')) 

È necessario applicare il mock mentre i singoli test sono effettivamente in esecuzione, non solo in setUp(). Possiamo creare il simulato in setUp() e applicarlo in seguito con un gestore di contesto with ....

Answer 3

Mi sono imbattuto in un problema in cui stavo cercando di prendere in giro le variabili utilizzate al di fuori di qualsiasi funzione o classe, il che è problematico perché vengono utilizzate nel momento in cui si tenta di prendere in giro la classe, prima di poter prendere in giro i valori.

Ho finito per utilizzare una variabile di ambiente. Se la variabile di ambiente esiste, utilizzare tale valore, altrimenti utilizzare l'impostazione predefinita dell'applicazione. In questo modo ho potuto impostare il valore della variabile di ambiente nei miei test.

Nella mia prova, ho avuto questo codice prima della classe è stato importato

os.environ["PROFILER_LOG_PATH"] = "./" 

Nella mia classe:

log_path = os.environ.get("PROFILER_LOG_PATH",config.LOG_PATH) 

Per impostazione predefinita il mio config.LOG_PATH è /var/log/<my app name>, ma ora quando l'esecuzione del test, il percorso del log è impostato sulla directory corrente. In questo modo non è necessario l'accesso root per eseguire i test.