t/t0004-model-save-restore.py

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