From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Sat, 1 Mar 2014 18:55:30 +0000 (-0500)
Subject: - add new tests for individual type reflection
X-Git-Tag: rel_0_9_4~92
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=fa23570a338b53c813b6342fba0cacd4f5e61384;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

- add new tests for individual type reflection
---

diff --git a/lib/sqlalchemy/testing/suite/test_reflection.py b/lib/sqlalchemy/testing/suite/test_reflection.py
index 7d805e8eeb..2b2ba52fef 100644
--- a/lib/sqlalchemy/testing/suite/test_reflection.py
+++ b/lib/sqlalchemy/testing/suite/test_reflection.py
@@ -238,6 +238,52 @@ class ComponentReflectionTest(fixtures.TablesTest):
     def test_get_columns(self):
         self._test_get_columns()
 
+    @testing.provide_metadata
+    def _type_round_trip(self, *types):
+        t = Table('t', self.metadata,
+                    *[
+                        Column('t%d' % i, type_)
+                        for i, type_ in enumerate(types)
+                    ]
+                )
+        t.create()
+
+        return [
+            c['type'] for c in
+            inspect(self.metadata.bind).get_columns('t')
+        ]
+
+    @testing.requires.table_reflection
+    def test_numeric_reflection(self):
+        for typ in self._type_round_trip(
+                            sql_types.Numeric(18, 5),
+                        ):
+            assert isinstance(typ, sql_types.Numeric)
+            eq_(typ.precision, 18)
+            eq_(typ.scale, 5)
+
+    @testing.requires.table_reflection
+    def test_varchar_reflection(self):
+        typ = self._type_round_trip(sql_types.String(52))[0]
+        assert isinstance(typ, sql_types.String)
+        eq_(typ.length, 52)
+
+    @testing.requires.table_reflection
+    @testing.provide_metadata
+    def test_nullable_reflection(self):
+        t = Table('t', self.metadata,
+                        Column('a', Integer, nullable=True),
+                        Column('b', Integer, nullable=False))
+        t.create()
+        eq_(
+            dict(
+                (col['name'], col['nullable'])
+                for col in inspect(self.metadata.bind).get_columns('t')
+            ),
+            {"a": True, "b": False}
+        )
+
+
     @testing.requires.table_reflection
     @testing.requires.schemas
     def test_get_columns_with_schema(self):