test/test_cola_model.py

   1 #!/usr/bin/env python
   2 import unittest
   3 import helper
   4
   5 from cola.models.base import BaseModel
   6
   7 class ExampleModel(BaseModel):
   8     """An example model for use by these tests"""
   9     def pass_through(self, value):
  10         """Passes values through unmodified"""
  11         return value
  12
  13 class ModelTest(unittest.TestCase):
  14     """Tests the cola.model.Model class"""
  15
  16     def test_create_attribute(self):
  17         """Test that arbitrary attributes provide set_* methods"""
  18         model = ExampleModel()
  19         model.set_foo('bar')
  20         self.assertEqual(model.foo, 'bar')
  21
  22     def test_param(self):
  23         """Test attribute access in case we f*** up __getattr__"""
  24         model = ExampleModel()
  25         model.attribute = 'value'
  26         self.assertEqual(model.param('attribute'), 'value')
  27
  28     # basic tests to ensure __getattr__ is sane
  29     def test_method(self):
  30         """Test calling a concrete method"""
  31         model = ExampleModel()
  32         self.assertEqual(model.pass_through('value'), 'value')
  33
  34     def test_dict_attribute(self):
  35         """Test setting dictionaries/dictionary values."""
  36         model = ExampleModel()
  37         model.thedict = { 'hello': 'world' }
  38         self.assertEqual(model.thedict['hello'], 'world' )
  39
  40
  41 if __name__ == '__main__':
  42     unittest.main()