Discussion :

[sebac]

Bonjour,
J'ai une fonction que je veux mocker

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

pwd.getpwnam(USER_NAME).pw_uid

je veux pouvoir lors de mes tests obliger cette fonction à retourner une valeur numérique de mon choix. Je suis en python 2.7 et j'utilise mock et j'arrive à mocker les fonctions écrite par moi même mais les fonctions des librairies python je n'y arrive pas

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
class Permissions():
 
	def pwd():
            USER_NAME="titi"
	    expected_uid=pwd.getpwnam(USER_NAME).pw_uid

dans mon module de test

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

def test_pwd(self):

j'essaye d'utiliser

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

 @patch('pwd.getpwnam')

mais rien n'y fait je n'arrive à rien.
Merci

[wiztricks]

Salut,

Le problème est peut être que vous ne voulez pas mocker une fonction mais l'objet qu'elle retourne.
Partez d'un truc comme:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
import pwd
def f():
    return pwd.getpwnam('user').pw_uid

pwd.getpwnam retourne un objet de type pwd.struct_passwd qui est une sorte de namedtuple, enfin l'instance d'un truc qui a un attribut .pw_uid:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
from collections import namedtuple
fake = namedtuple('fake', 'pw_uid')

Maintenant, remplaçons la fonction pwd.getpwnam par un mock qui retourne un "fake" avec la valeur qu'on veut:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
from unittest.mock import MagicMock
pwd.getpwnam = MagicMock(return_value=fake(1234))

- W

[sebac]

Merci, ça fonctionne bien