Mock de fonction

Bonjour,
J'ai une fonction que je veux mocker

Code:

---

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:

---

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:

---

def test_pwd(self):

---

j'essaye d'utiliser

Code:

---

@patch('pwd.getpwnam')

---

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

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:

---

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:

---

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:

---

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

---

- W

Merci, ça fonctionne bien