From fa84da27458847d90075c9f8d77333b532d7882a Mon Sep 17 00:00:00 2001
From: Zhongyue Luo <zhongyue.nah@intel.com>
Date: Wed, 10 Apr 2013 10:31:03 +0800
Subject: Removes created_at, updated_at from ModelBase

Created a 'TimestampMixin' class.
Next step is to update Nova which is the only project using oslo.db

Fixes bug #1119702

Change-Id: I1ffcc09a2971e6e6102da7ecb855a2837a159baf
---
 tests/unit/db/sqlalchemy/test_models.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'tests/unit/db/sqlalchemy/test_models.py')

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))
-- 
cgit