1 files changed, 52 insertions, 0 deletions
diff --git a/keystone-moon/keystone/tests/unit/common/test_sql_core.py b/keystone-moon/keystone/tests/unit/common/test_sql_core.py
new file mode 100644
index 00000000..1f33cfc3
--- /dev/null
+++ b/keystone-moon/keystone/tests/unit/common/test_sql_core.py
@@ -0,0 +1,52 @@
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+
+from sqlalchemy.ext import declarative
+
+from keystone.common import sql
+from keystone.tests import unit as tests
+from keystone.tests.unit import utils
+
+
+ModelBase = declarative.declarative_base()
+
+
+class TestModel(ModelBase, sql.ModelDictMixin):
+    __tablename__ = 'testmodel'
+    id = sql.Column(sql.String(64), primary_key=True)
+    text = sql.Column(sql.String(64), nullable=False)
+
+
+class TestModelDictMixin(tests.BaseTestCase):
+
+    def test_creating_a_model_instance_from_a_dict(self):
+        d = {'id': utils.new_uuid(), 'text': utils.new_uuid()}
+        m = TestModel.from_dict(d)
+        self.assertEqual(m.id, d['id'])
+        self.assertEqual(m.text, d['text'])
+
+    def test_creating_a_dict_from_a_model_instance(self):
+        m = TestModel(id=utils.new_uuid(), text=utils.new_uuid())
+        d = m.to_dict()
+        self.assertEqual(m.id, d['id'])
+        self.assertEqual(m.text, d['text'])
+
+    def test_creating_a_model_instance_from_an_invalid_dict(self):
+        d = {'id': utils.new_uuid(), 'text': utils.new_uuid(), 'extra': None}
+        self.assertRaises(TypeError, TestModel.from_dict, d)
+
+    def test_creating_a_dict_from_a_model_instance_that_has_extra_attrs(self):
+        expected = {'id': utils.new_uuid(), 'text': utils.new_uuid()}
+        m = TestModel(id=expected['id'], text=expected['text'])
+        m.extra = 'this should not be in the dictionary'
+        self.assertEqual(m.to_dict(), expected)