summaryrefslogtreecommitdiff
path: root/tests/test_autogenerate.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_autogenerate.py')
-rw-r--r--tests/test_autogenerate.py13
1 files changed, 7 insertions, 6 deletions
diff --git a/tests/test_autogenerate.py b/tests/test_autogenerate.py
index 1eeeb91..e9ffe8b 100644
--- a/tests/test_autogenerate.py
+++ b/tests/test_autogenerate.py
@@ -11,6 +11,7 @@ from alembic import autogenerate
from alembic.migration import MigrationContext
from alembic.testing import TestBase
from alembic.testing import config
+from alembic.testing import assert_raises_message
from alembic.testing.mock import Mock
from alembic.testing.env import staging_env, clear_staging_env
from alembic.testing import eq_
@@ -1358,12 +1359,12 @@ class CompareMetadataTest(ModelOne, AutogenTest, TestBase):
)
metadata = self.m2
- try:
- autogenerate.compare_metadata(context, metadata)
- except CommandError:
- pass
- else:
- assert False, "unexpected success"
+ assert_raises_message(
+ CommandError,
+ "autogenerate can't use as_sql=True as it prevents "
+ "querying the database for schema information",
+ autogenerate.compare_metadata, context, metadata
+ )
class PGCompareMetaData(ModelOne, AutogenTest, TestBase):