t/t0002-model-object.py

   1 #!/usr/bin/env python
   2 import unittest
   3 from testmodel import TestModel
   4
   5 class ModelTest(unittest.TestCase):
   6         def setUp(self):
   7                 self.model = TestModel()
   8         def tearDown(self):
   9                 del self.model
  10
  11         def testCreate(self):
  12                 self.model.create(foo='bar')
  13                 self.failUnless(self.model.get_foo()=='bar')
  14
  15         def testRegularAttributes(self):
  16                 '''Test accessing attribute via model.attribute.'''
  17                 self.failUnless( self.model.attribute == 'value' )
  18
  19         def testMixedcaseAttributes(self):
  20                 '''Test accessing attribute via model.Attribute.'''
  21                 self.failUnless( self.model.AtTrIbUte == 'value' )
  22
  23         def testExplicitAttr(self):
  24                 '''Test accessing an explicit attribute. Just in case we f** up getattr'''
  25                 self.failUnless( self.model.hello == 'world' )
  26
  27         def testRealMethod(self):
  28                 '''Test calling a concrete model method.'''
  29                 self.failUnless( self.model.testMethod() == 'test' )
  30
  31         def testGetter(self):
  32                 '''Test calling using the get* method.'''
  33                 self.failUnless( self.model.getAttribute() == 'value' )
  34
  35         def testSetter(self):
  36                 '''Test using the set* method.'''
  37                 self.model.set_param('newAttribute','baz')
  38                 self.failUnless( self.model.newattribute == 'baz' )
  39                 self.failUnless( self.model.getNewAttribute() == 'baz' )
  40
  41         def testAddArray(self):
  42                 '''Test using the add* array method.'''
  43                 self.model.array = []
  44                 self.model.addArray('bar')
  45                 self.failUnless( self.model.array[0] == 'bar' )
  46
  47                 self.model.addArray('baz','quux')
  48                 self.failUnless( self.model.array[1] == 'baz' )
  49                 self.failUnless( self.model.array[2] == 'quux' )
  50
  51         def testAppendArray(self):
  52                 '''Test using the append* array method.'''
  53                 self.model.array = []
  54                 self.model.appendArray('bar')
  55                 self.failUnless( self.model.array[0] == 'bar' )
  56
  57                 self.model.appendArray('baz','quux')
  58                 self.failUnless( self.model.array[1] == 'baz' )
  59                 self.failUnless( self.model.array[2] == 'quux' )
  60
  61         def testDict(self):
  62                 '''Test setting dictionary/dictionary values.'''
  63                 self.model.dict = { 'hello': 'world' }
  64                 self.failUnless( self.model.dict['hello'] == 'world' )
  65                 self.failUnless( self.model.getDict()['hello'] == 'world' )
  66
  67         def testFromDict(self):
  68                 '''Test reconstituting a model from a dictionary.'''
  69                 self.model.from_dict({
  70                         'test_dict': { 'hello':'world' },
  71                         'test_list': [ 'foo', 'bar' ],
  72                         'test_str': 'foo',
  73                 })
  74                 self.failUnless( self.model.get_test_dict()['hello'] == 'world' )
  75                 self.failUnless( self.model.get_test_list()[1] == 'bar' )
  76
  77 if __name__ == '__main__':
  78         unittest.main()