Programmation PYTHON - Zenk - Security - Repository

test_cPickle.py
#!/usr/bin/python
# -*- coding: utf8 -*import
cPickle
import unittest
# une classe basique pour nos tests
class T:
a = 0
b = 0
class cPickleTestCase(unittest.TestCase):
def _genere_instance(self):
""" renvoi un objet """
o = T()
o.a = 1
o.b = 2
return o
def test_dump_et_load(self):
""" test l'E/S de cPickle """
# objet de test
o = self._genere_instance()
Programmation dirigée par les tests
CHAPITRE 12
# créons un fichier en écriture
fichier = open('/home/tziade/pickled.bin', 'w')
# écrivons l'objet o dans le flux avec la méthode dump de cPickle
cPickle.dump(o, fichier)
# fermons le fichier
fichier.close()
# ouvrons le fichier en lecture
fichier = open('/home/tziade/pickled.bin')
# cPickle.load permet de recharger un objet après une sauvegarde
par dump
o2 = cPickle.load(fic)
# vérifions l'objet renvoyé
assert(isinstance(o2, object))
assert(o2.a == 1)
assert(o2.b == 2)
Tout nouveau scénario de test pour cPickle pourra être implémenté dans une nouvelle
méthode de cette classe, en préfixant son nom par « test ».
405