added test for storage pickling

This commit is contained in:
niphlod
2014-09-10 18:24:52 +02:00
parent c3cd32cc4f
commit bd87879cde
+8
View File
@@ -37,6 +37,7 @@ def fix_sys_path():
fix_sys_path() fix_sys_path()
from storage import Storage from storage import Storage
import pickle
class TestStorage(unittest.TestCase): class TestStorage(unittest.TestCase):
@@ -97,5 +98,12 @@ class TestStorage(unittest.TestCase):
self.assertEquals(s['a'], None) self.assertEquals(s['a'], None)
self.assertTrue('a' in s) self.assertTrue('a' in s)
def test_pickling(self):
""" Test storage pickling """
s = Storage(a=1)
sd = pickle.dumps(s, pickle.HIGHEST_PROTOCOL)
news = pickle.loads(sd)
self.assertEqual(news.a, 1)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()