test/test_model_serialization.py

   1 #!/usr/bin/env python
   2 import os
   3 import unittest
   4
   5 import testlib
   6 from testlib import InnerModel
   7 from testlib import NestedModel
   8
   9 from cola.model import Model
  10
  11 class TestSaveRestore(unittest.TestCase):
  12     def setUp(self):
  13         testlib.create_test_dir()
  14
  15         self.nested = NestedModel()
  16         path = os.path.join(testlib.get_test_dir(), 'test.data')
  17         # save & reconstitute
  18         self.nested.save(path)
  19         self.clone = Model.instance(path)
  20
  21     def tearDown(self):
  22         testdir = testlib.get_test_dir()
  23         if os.path.exists(testdir):
  24             testlib.remove_test_dir()
  25
  26     def testCloneToClass(self):
  27         self.failUnless( str(NestedModel) ==
  28                          str(self.clone.__class__) )
  29
  30     def testClonedInnerToClass(self):
  31         self.failUnless( str(InnerModel) ==
  32                          str(self.clone.inner.__class__) )
  33
  34     def testClonedListAndParam(self):
  35         self.failUnless( str(self.clone.inner.__class__) ==
  36                          str(self.clone.innerList[0].__class__) )
  37
  38     def testList(self):
  39         self.failUnless( str(InnerModel) ==
  40                          str(self.clone.innerList[1][0].__class__) )
  41
  42     def testListInList(self):
  43         self.failUnless( str(InnerModel) ==
  44                          str(self.clone.innerList[2][0][0].__class__))
  45
  46     def testListInListInList(self):
  47         self.failUnless( str(InnerModel) ==
  48                          str(self.clone.innerList[3][0][0][0].__class__))
  49
  50     def testDictInList(self):
  51         self.failUnless( str(dict) ==
  52                          str(self.clone.innerList[4].__class__))
  53
  54     def testObjectInDictInList(self):
  55         self.failUnless( str(InnerModel) ==
  56                          str(self.clone.innerList[-1]["foo"].__class__))
  57
  58 if __name__ == '__main__':
  59     unittest.main()