Programmation PYTHON - Zenk - Security - Repository

396
Techniques avancées
QUATRIÈME PARTIE
Cette technique consiste à modifier à la volée, au moment du test, la portion de code
qui accède à une ressource externe pour la remplacer par du code qui se contente de
renvoyer un résultat convenable pour la suite des tests. L’objectif est de faire croire au
code appelant que tout s’est bien déroulé, et de lui renvoyer un résultat correct. La
qualité d’imitation du résultat renvoyé dépend de l’utilisation qui en est faite dans le
code et peut parfois être grossière.
Python est un langage suffisamment souple pour permettre de modifier les définitions
de modules, classes et fonctions à la volée, et cette technique peut être appliquée
à tous les étages du code.
Modification de fonctions et méthodes
L’exemple ci-dessous modifie temporairement la fonction urlopen d’urllib2 avant
de démarrer les tests, pour qu’elle renvoie un résultat même si la machine de test ne
peut pas se connecter à l’URL indiquée.
Bouchon pour urllib
#!/usr/bin/python
# -*- coding: utf8 -*import
urllib
import StringIO
# fonction de remplacement
def fakeopen(url, data=None):
res = StringIO.StringIO('Dummy Page')
return res
# monkey patching
original_urllib = urllib.urlopen
urllib.urlopen = fakeopen
# test d'exemple
res = urllib.urlopen('http://google.fr')
assert(res.readlines(),
['Dummy Page'])
...
# retrait du patch
urllib.urlopen = original_urllib
Le test d’exemple ne sert qu’à valider que le patch a bien été appliqué. Dans cette
zone, tout accès à la fonction urlopen, exécutera le patch, sauf si la directive reload
est appelée sur le module unittest.