diff options
Diffstat (limited to 'tests/unit/db/sqlalchemy/test_models.py')
| -rw-r--r-- | tests/unit/db/sqlalchemy/test_models.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/unit/db/sqlalchemy/test_models.py b/tests/unit/db/sqlalchemy/test_models.py index cd1a664..04905a6 100644 --- a/tests/unit/db/sqlalchemy/test_models.py +++ b/tests/unit/db/sqlalchemy/test_models.py @@ -28,6 +28,7 @@ class ModelBaseTest(test_utils.BaseTestCase): 'get', 'next', 'update', + 'save', 'iteritems') for method in dict_methods: self.assertTrue(hasattr(models.ModelBase, method)) @@ -70,3 +71,24 @@ class ModelBaseTest(test_utils.BaseTestCase): found_items += 1 self.assertEqual(min_items, found_items) + + +class TimestampMixinTest(test_utils.BaseTestCase): + + def test_timestampmixin_attr(self): + + class TestModel(models.ModelBase, models.TimestampMixin): + pass + + dict_methods = ('__getitem__', + '__setitem__', + '__iter__', + 'get', + 'next', + 'update', + 'save', + 'iteritems', + 'created_at', + 'updated_at') + for method in dict_methods: + self.assertTrue(hasattr(TestModel, method)) |