Merge pull request #492 from niphlod/tests/storage_pickling
added test for storage pickling
This commit is contained in:
@@ -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):
|
||||||
@@ -96,6 +97,13 @@ class TestStorage(unittest.TestCase):
|
|||||||
self.assertEquals(s.a, None)
|
self.assertEquals(s.a, None)
|
||||||
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()
|
||||||
|
|||||||
Reference in New Issue
Block a user