From ba1bfef0f281dc131eb296eb84cec915638cd012 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Tue, 2 Feb 2016 16:59:56 -0500 Subject: [PATCH] - test fixes for metadata.enums value now being a list - modernize those tests as well --- test/dialect/postgresql/test_reflection.py | 2 +- test/dialect/postgresql/test_types.py | 42 ++++++++++------------ 2 files changed, 20 insertions(+), 24 deletions(-) diff --git a/test/dialect/postgresql/test_reflection.py b/test/dialect/postgresql/test_reflection.py index 851facd2a7..8da18108f9 100644 --- a/test/dialect/postgresql/test_reflection.py +++ b/test/dialect/postgresql/test_reflection.py @@ -202,7 +202,7 @@ class DomainReflectionTest(fixtures.TestBase, AssertsExecutionResults): table = Table('enum_test', metadata, autoload=True) eq_( table.c.data.type.enums, - ('test', ) + ['test'] ) def test_table_is_reflected_test_schema(self): diff --git a/test/dialect/postgresql/test_types.py b/test/dialect/postgresql/test_types.py index c53c67cefd..8818a99412 100644 --- a/test/dialect/postgresql/test_types.py +++ b/test/dialect/postgresql/test_types.py @@ -170,7 +170,7 @@ class EnumTest(fixtures.TestBase, AssertsExecutionResults): t2 = Table('table', m2, autoload=True) eq_( t2.c.value.type.enums, - (util.u('réveillé'), util.u('drôle'), util.u('S’il')) + [util.u('réveillé'), util.u('drôle'), util.u('S’il')] ) @testing.provide_metadata @@ -445,8 +445,9 @@ class EnumTest(fixtures.TestBase, AssertsExecutionResults): e.connect() assert not dialect.supports_native_enum + @testing.provide_metadata def test_reflection(self): - metadata = MetaData(testing.db) + metadata = self.metadata etype = Enum('four', 'five', 'six', name='fourfivesixtype', metadata=metadata) t1 = Table( @@ -458,18 +459,16 @@ class EnumTest(fixtures.TestBase, AssertsExecutionResults): 'one', 'two', 'three', name='onetwothreetype')), Column('value2', etype)) metadata.create_all() - try: - m2 = MetaData(testing.db) - t2 = Table('table', m2, autoload=True) - assert t2.c.value.type.enums == ('one', 'two', 'three') - assert t2.c.value.type.name == 'onetwothreetype' - assert t2.c.value2.type.enums == ('four', 'five', 'six') - assert t2.c.value2.type.name == 'fourfivesixtype' - finally: - metadata.drop_all() + m2 = MetaData(testing.db) + t2 = Table('table', m2, autoload=True) + eq_(t2.c.value.type.enums, ['one', 'two', 'three']) + eq_(t2.c.value.type.name, 'onetwothreetype') + eq_(t2.c.value2.type.enums, ['four', 'five', 'six']) + eq_(t2.c.value2.type.name, 'fourfivesixtype') + @testing.provide_metadata def test_schema_reflection(self): - metadata = MetaData(testing.db) + metadata = self.metadata etype = Enum( 'four', 'five', @@ -478,7 +477,7 @@ class EnumTest(fixtures.TestBase, AssertsExecutionResults): schema='test_schema', metadata=metadata, ) - t1 = Table( + Table( 'table', metadata, Column( 'id', Integer, primary_key=True), @@ -488,16 +487,13 @@ class EnumTest(fixtures.TestBase, AssertsExecutionResults): name='onetwothreetype', schema='test_schema')), Column('value2', etype)) metadata.create_all() - try: - m2 = MetaData(testing.db) - t2 = Table('table', m2, autoload=True) - assert t2.c.value.type.enums == ('one', 'two', 'three') - assert t2.c.value.type.name == 'onetwothreetype' - assert t2.c.value2.type.enums == ('four', 'five', 'six') - assert t2.c.value2.type.name == 'fourfivesixtype' - assert t2.c.value2.type.schema == 'test_schema' - finally: - metadata.drop_all() + m2 = MetaData(testing.db) + t2 = Table('table', m2, autoload=True) + eq_(t2.c.value.type.enums, ['one', 'two', 'three']) + eq_(t2.c.value.type.name, 'onetwothreetype') + eq_(t2.c.value2.type.enums, ['four', 'five', 'six']) + eq_(t2.c.value2.type.name, 'fourfivesixtype') + eq_(t2.c.value2.type.schema, 'test_schema') @testing.provide_metadata def test_custom_subclass(self): -- 2.47.2