From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Wed, 8 Aug 2012 23:45:13 +0000 (-0400)
Subject: - add a test for Enum repr() - requires SQLA 0.7.9 or 0.8
X-Git-Tag: rel_0_3_6~2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=1a6bba1caf2856ec235ac46773149ae4a98d0d03;p=thirdparty%2Fsqlalchemy%2Falembic.git

- add a test for Enum repr() - requires SQLA 0.7.9 or 0.8
---

diff --git a/tests/test_autogenerate.py b/tests/test_autogenerate.py
index 2edd8693..ecf2c3ed 100644
--- a/tests/test_autogenerate.py
+++ b/tests/test_autogenerate.py
@@ -1,6 +1,6 @@
 from sqlalchemy import MetaData, Column, Table, Integer, String, Text, \
     Numeric, CHAR, ForeignKey, DATETIME, \
-    TypeDecorator, CheckConstraint, Unicode,\
+    TypeDecorator, CheckConstraint, Unicode, Enum,\
     UniqueConstraint, Boolean
 from sqlalchemy.types import NULLTYPE
 from sqlalchemy.dialects import mysql
@@ -566,4 +566,20 @@ class AutogenRenderTest(TestCase):
             "existing_server_default='5')"
         )
 
+    def test_render_enum(self):
+        eq_ignore_whitespace(
+            autogenerate._repr_type(
+                        "sa.",
+                        Enum("one", "two", "three", name="myenum"),
+                        self.autogen_context),
+            "sa.Enum('one', 'two', 'three', name='myenum')"
+        )
+        eq_ignore_whitespace(
+            autogenerate._repr_type(
+                        "sa.",
+                        Enum("one", "two", "three"),
+                        self.autogen_context),
+            "sa.Enum('one', 'two', 'three')"
+        )
+
 # TODO: tests for dialect-specific type rendering + imports